#501 优化，还需要做一些优化。

Author: xlorne

starter-txlcn-tc/src/main/resources/META-INF/spring.factories

@@ -10,13 +10,11 @@ com.codingapi.txlcn.tc.jdbc.JdbcConfiguration,\
 com.codingapi.txlcn.tc.jdbc.event.JdbcEventConfiguration,\
 com.codingapi.txlcn.tc.jdbc.log.JdbcLogConfiguration,\
 com.codingapi.txlcn.tc.jdbc.sql.JdbcSqlConfiguration,\
+com.codingapi.txlcn.tc.jdbc.sql.analyse.AnalyseConfiguration,\
+com.codingapi.txlcn.tc.jdbc.sql.strategy.StrategyConfiguration,\
 com.codingapi.txlcn.tc.reporter.ReporterConfiguration,\
 com.codingapi.txlcn.tc.resolver.ResolverConfiguration,\
 com.codingapi.txlcn.tc.runner.TcRunnerConfiguration,\
 com.codingapi.txlcn.tc.rpc.RpcTransactionConfiguration,\
 com.codingapi.txlcn.tc.TCAutoConfiguration,\
-com.codingapi.txlcn.tc.jdbc.sql.strategy.MysqlInsertAnalyseStrategy,\
-com.codingapi.txlcn.tc.jdbc.sql.strategy.MysqlSqlDeleteAnalyseStrategy,\
-com.codingapi.txlcn.tc.jdbc.sql.strategy.MysqlUpdateAnalyseStrategy,\
-com.codingapi.txlcn.tc.jdbc.sql.strategy.AnalyseStrategryFactory,\
- com.codingapi.txlcn.tc.jdbc.sql.analyse.MysqlAnalyse
+

txlcn-tc/src/main/java/com/codingapi/txlcn/tc/jdbc/sql/JdbcSqlConfiguration.java

@@ -1,6 +1,7 @@
 package com.codingapi.txlcn.tc.jdbc.sql;
 
 import com.codingapi.txlcn.tc.config.TxConfig;
+import com.codingapi.txlcn.tc.jdbc.sql.strategy.AnalyseStrategryFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -21,7 +22,7 @@ public SqlAnalyseStrategy sqlParserContext(@Autowired(required = false) List<Sql
  }
 
  @Bean
- public SqlAnalyse mysqlSqlParser(){
- return new MysqlSqlAnalyse();
+ public SqlAnalyse mysqlSqlParser(AnalyseStrategryFactory analyseStrategryFactory){
+ return new MysqlSqlAnalyse(analyseStrategryFactory);
  }
 }

txlcn-tc/src/main/java/com/codingapi/txlcn/tc/jdbc/sql/MysqlSqlAnalyse.java

@@ -2,14 +2,11 @@
 
 import com.codingapi.txlcn.p6spy.common.StatementInformation;
 import com.codingapi.txlcn.tc.jdbc.sql.strategy.AnalyseStrategryFactory;
-import com.codingapi.txlcn.tc.jdbc.sql.strategy.MysqlAnalyseEnum;
+import com.codingapi.txlcn.tc.jdbc.sql.strategy.SqlSqlAnalyseHandler;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
 import net.sf.jsqlparser.parser.CCJSqlParserManager;
 import net.sf.jsqlparser.statement.Statement;
-import net.sf.jsqlparser.statement.delete.Delete;
-import net.sf.jsqlparser.statement.insert.Insert;
-import net.sf.jsqlparser.statement.update.Update;
 
 import java.io.StringReader;
 import java.sql.Connection;
@@ -23,6 +20,12 @@
 @Slf4j
 public class MysqlSqlAnalyse implements SqlAnalyse {
 
+ private AnalyseStrategryFactory analyseStrategryFactory;
+
+ public MysqlSqlAnalyse(AnalyseStrategryFactory analyseStrategryFactory) {
+ this.analyseStrategryFactory = analyseStrategryFactory;
+ }
+
  @Override
  public String sqlType() {
  return "mysql";
@@ -37,15 +40,13 @@ public String analyse(String sql,StatementInformation statementInformation) thr
  // if else 实现并不是很优雅
  CCJSqlParserManager parser = new CCJSqlParserManager();
  Statement stmt = parser.parse(new StringReader(sql));
- if (stmt instanceof Insert) {
- return AnalyseStrategryFactory.getInvokeStrategy(MysqlAnalyseEnum.INSERT.name()).mysqlAnalyseStrategy(sql,connection,stmt);
- } else if (stmt instanceof Update) {
- return AnalyseStrategryFactory.getInvokeStrategy(MysqlAnalyseEnum.UPDATE.name()).mysqlAnalyseStrategy(sql,connection,stmt);
- } else if (stmt instanceof Delete) {
- return AnalyseStrategryFactory.getInvokeStrategy(MysqlAnalyseEnum.DELETE.name()).mysqlAnalyseStrategy(sql,connection,stmt);
+ SqlSqlAnalyseHandler sqlSqlAnalyseHandler = analyseStrategryFactory.getInvokeStrategy(sqlType(),stmt);
+ if(sqlSqlAnalyseHandler==null){
+ return sql;
  }
- return sql;
+ return sqlSqlAnalyseHandler.analyse(sql,connection,stmt);
  }
+
  @Override
  public boolean preAnalyse(String sql) {
  // SQL类型检查，只有对CUD(CURD)操作做处理

txlcn-tc/src/main/java/com/codingapi/txlcn/tc/jdbc/sql/analyse/AnalyseConfiguration.java

@@ -0,0 +1,23 @@
+package com.codingapi.txlcn.tc.jdbc.sql.analyse;
+
+import org.springframework.beans.factory.annotation.Configurable;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
+import org.springframework.context.annotation.Bean;
+
+/**
+ * @author lorne
+ * @date 2020/9/17
+ * @description
+ */
+@Configurable
+public class AnalyseConfiguration {
+
+ @Bean
+ @ConditionalOnMissingBean
+ public SqlDetailAnalyse mysqlSqlDetailAnalyse(){
+ return new MysqlSqlDetailAnalyse();
+ }
+
+
+
+}

txlcn-tc/src/main/java/com/codingapi/txlcn/tc/jdbc/sql/analyse/MysqlAnalyse.java renamed to txlcn-tc/src/main/java/com/codingapi/txlcn/tc/jdbc/sql/analyse/MysqlSqlDetailAnalyse.java

@@ -23,7 +23,7 @@
 import java.util.Map;
 
 @Slf4j
-public class MysqlAnalyse  implements SqlDetailAnalyse{
+public class MysqlSqlDetailAnalyse implements SqlDetailAnalyse{
 
  @Override
  public boolean singleInsertAnalyse(String sql, Connection connection, Insert statement, ItemsList itemsList, String pk, int pkIndex) throws SQLException {

txlcn-tc/src/main/java/com/codingapi/txlcn/tc/jdbc/sql/analyse/SqlDetailAnalyse.java

@@ -13,6 +13,9 @@
 import java.util.List;
 import java.util.Map;
 
+/**
+ * //todo 若作为各种数据库的适配的话应该还需要做其他的数据库的适配，这样的话还不能满足需求。
+ */
 public interface SqlDetailAnalyse {
 
  /**

txlcn-tc/src/main/java/com/codingapi/txlcn/tc/jdbc/sql/strategy/AnalyseStrategryFactory.java

@@ -1,11 +1,8 @@
 package com.codingapi.txlcn.tc.jdbc.sql.strategy;
 
-import com.google.common.collect.Maps;
-import org.springframework.context.annotation.Configuration;
-import org.springframework.util.StringUtils;
+import net.sf.jsqlparser.statement.Statement;
 
-import javax.annotation.PostConstruct;
-import java.util.Map;
+import java.util.List;
 
 /**
  * @author Gz.
@@ -15,17 +12,19 @@
 
 public class AnalyseStrategryFactory {
 
- private static Map<String, SqlSqlAnalyseHandler> strategyMap = Maps.newHashMap();
+ private List<SqlSqlAnalyseHandler> analyseHandlers;
 
- public static SqlSqlAnalyseHandler getInvokeStrategy(String name) {
- return strategyMap.get(name);
+ public AnalyseStrategryFactory(List<SqlSqlAnalyseHandler> analyseHandlers) {
+ this.analyseHandlers = analyseHandlers;
  }
 
- public static void register(String name, SqlSqlAnalyseHandler handler) {
- if (StringUtils.isEmpty(name) || null == handler) {
- return;
+ public SqlSqlAnalyseHandler getInvokeStrategy(String sqlType,Statement statement) {
+ for(SqlSqlAnalyseHandler handler : analyseHandlers){
+ if(handler.preAnalyse(sqlType,statement)){
+ return handler;
+ }
  }
- strategyMap.put(name, handler);
+ return null;
  }
 
 }

txlcn-tc/src/main/java/com/codingapi/txlcn/tc/jdbc/sql/strategy/MysqlAnalyseEnum.java

@@ -1,9 +1,7 @@
 package com.codingapi.txlcn.tc.jdbc.sql.strategy;
 
-import com.codingapi.txlcn.p6spy.common.StatementInformation;
 import net.sf.jsqlparser.JSQLParserException;
 import net.sf.jsqlparser.statement.Statement;
-import org.springframework.beans.factory.InitializingBean;
 
 import java.sql.Connection;
 import java.sql.SQLException;
@@ -13,7 +11,12 @@
  * @description: SQL分析则略接口
  * @date 2020-08-13 23:08:26
  */
-public interface SqlSqlAnalyseHandler extends InitializingBean {
+public interface SqlSqlAnalyseHandler {
+
+
+ String analyse(String sql, Connection connection, Statement stmt) throws JSQLParserException, SQLException;
+
+
+ boolean preAnalyse(String sqlType,Statement stmt);
 
- String mysqlAnalyseStrategy(String sql,Connection connection,Statement stmt) throws JSQLParserException, SQLException;
 }

txlcn-tc/src/main/java/com/codingapi/txlcn/tc/jdbc/sql/strategy/SqlSqlAnalyseHandler.java

@@ -0,0 +1,46 @@
+package com.codingapi.txlcn.tc.jdbc.sql.strategy;
+
+import com.codingapi.txlcn.tc.jdbc.sql.analyse.SqlDetailAnalyse;
+import com.codingapi.txlcn.tc.jdbc.sql.strategy.mysql.MysqlDeleteAnalyseStrategy;
+import com.codingapi.txlcn.tc.jdbc.sql.strategy.mysql.MysqlInsertAnalyseStrategy;
+import com.codingapi.txlcn.tc.jdbc.sql.strategy.mysql.MysqlUpdateAnalyseStrategy;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Configurable;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
+import org.springframework.context.annotation.Bean;
+
+import java.util.List;
+
+/**
+ * @author lorne
+ * @date 2020/9/17
+ * @description
+ */
+@Configurable
+public class StrategyConfiguration {
+
+ @Bean
+ public AnalyseStrategryFactory analyseStrategryFactory(@Autowired(required = false)List<SqlSqlAnalyseHandler> analyseHandlers){
+ return new AnalyseStrategryFactory(analyseHandlers);
+ }
+
+ @Bean
+ @ConditionalOnMissingBean
+ public SqlSqlAnalyseHandler mysqlInsertAnalyseStrategy(SqlDetailAnalyse sqlDetailAnalyse){
+ return new MysqlInsertAnalyseStrategy(sqlDetailAnalyse);
+ }
+
+ @Bean
+ @ConditionalOnMissingBean
+ public SqlSqlAnalyseHandler mysqlDeleteAnalyseStrategy(SqlDetailAnalyse sqlDetailAnalyse){
+ return new MysqlDeleteAnalyseStrategy(sqlDetailAnalyse);
+ }
+
+ @Bean
+ @ConditionalOnMissingBean
+ public SqlSqlAnalyseHandler mysqlUpdateAnalyseStrategy(SqlDetailAnalyse sqlDetailAnalyse){
+ return new MysqlUpdateAnalyseStrategy(sqlDetailAnalyse);
+ }
+
+
+}