拆分策略模式

dbsyncer-parser/src/main/java/org/dbsyncer/parser/ddl/alter/AddStrategy.java

@@ -0,0 +1,48 @@
+package org.dbsyncer.parser.ddl.alter;
+
+import java.util.LinkedList;
+import java.util.List;
+import net.sf.jsqlparser.statement.alter.AlterExpression;
+import org.dbsyncer.common.util.StringUtil;
+import org.dbsyncer.connector.config.DDLConfig;
+import org.dbsyncer.connector.model.Field;
+import org.dbsyncer.parser.model.FieldMapping;
+
+/**
+ * 解析add的属性
+ * exampleSql: ALTER TABLE cost ADD duan INT after(before) `tmp`;
+ * @author life
+ */
+public class AddStrategy implements ExpressionStrategy{
+
+    @Override
+    public void parse(AlterExpression expression, DDLConfig ddlConfig,
+            List<FieldMapping> fieldMappingList) {
+        for (AlterExpression.ColumnDataType columnDataType : expression.getColDataTypeList()) {
+            boolean findColumn = false;
+            List<String> columnSpecs = new LinkedList<>();
+            for (String spe:columnDataType.getColumnSpecs()) {//对一before，after进行处理
+                spe = StringUtil.replace(spe,"`","");
+                if (findColumn){
+                    //对before（after）字段进行映射
+                    String finalSpe = spe;
+                    FieldMapping fieldMapping = fieldMappingList.stream().filter(x->x.getSource().getName().equals(
+                            finalSpe)).findFirst().get();
+                    columnSpecs.add(fieldMapping.getTarget().getName());
+                    findColumn = false;
+                    continue;
+                }
+
+                if (StringUtil.equalsIgnoreCase(spe,"before") || StringUtil.equalsIgnoreCase(spe,"after")){
+                    findColumn =true;
+                }
+                columnSpecs.add(spe);
+            }
+            columnDataType.setColumnSpecs(columnSpecs);
+            String columName = columnDataType.getColumnName();
+            columName = StringUtil.replace(columName, "`", "");
+            Field field = new Field(columName,columnDataType.getColDataType().getDataType(),0);//感觉不需要都行，只需要名称，后续可以自己刷新
+            ddlConfig.getAddFields().add(field);
+        }
+    }
+}

dbsyncer-parser/src/main/java/org/dbsyncer/parser/ddl/alter/AlterStrategy.java

@@ -43,107 +43,23 @@ public class AlterStrategy implements JsqlParserStrategy {
         alter.getTable().setName(targetName);
         for (AlterExpression expression : alter.getAlterExpressions()) {
             AlterOperation alterOperation = expression.getOperation();
+            ExpressionStrategy expressionStrategy = null;
             if (alterOperation == AlterOperation.MODIFY) {//修改属性
                 ddlConfig.setDdlOperationEnum(DDLOperationEnum.ALTER_MODIFY);
-                parseModify(expression);
+                expressionStrategy = new ModifyStrategy();
             } else if (AlterOperation.ADD == alterOperation) {//新增字段，只需要替换表名不需要替换sql
                 ddlConfig.setDdlOperationEnum(DDLOperationEnum.ALTER_ADD);
-                parseAdd(expression);
+                expressionStrategy =new AddStrategy();
             }else if (AlterOperation.CHANGE == alterOperation){
                 ddlConfig.setDdlOperationEnum(DDLOperationEnum.ALTER_CHANGE);
-                parseChange(expression);
+                expressionStrategy = new ChangeStrategy();
             }else if (AlterOperation.DROP == alterOperation){
                 ddlConfig.setDdlOperationEnum(DDLOperationEnum.ALTER_DROP);
-                parseDrop(expression);
+                expressionStrategy = new DropStrategy();
             }
+            expressionStrategy.parse(expression,ddlConfig,fieldMappingList);
             ddlConfig.setSql(alter.toString());
         }
 
     }
-
-    //解析drop
-    //example: ALTER TABLE test_table DROP dis;
-    private void parseDrop(AlterExpression expression) {
-        String columName = expression.getColumnName();
-        columName = StringUtil.replace(columName, "`", "");
-        Field field = new Field(columName,null,0);
-        //需要把列替换成目标的列名
-        String finalColumName = columName;
-        FieldMapping fieldMapping =  fieldMappingList.stream().filter(x->x.getSource().getName().equals(
-                finalColumName)).findFirst().orElse(null);
-        if (fieldMapping !=null){
-            expression.setColumnName(fieldMapping.getTarget().getName());
-        }
-        //加入还是原名
-        ddlConfig.getRemoveFields().add(field);
-    }
-
-    //解析change属性
-    //exampleSql: ALTER TABLE test_table CHANGE duan1  duan2 INT(10)
-    private void parseChange(AlterExpression expression) {
-        String oldColumnName = expression.getColumnOldName();
-        oldColumnName =StringUtil.replace(oldColumnName,"`","");
-        ddlConfig.setSourceColumnName(oldColumnName);
-        String finalOldColumnName = oldColumnName;
-        FieldMapping fieldMapping = fieldMappingList.stream().filter(x->x.getSource().getName().equals(
-                finalOldColumnName)).findFirst().orElse(null);
-        if (fieldMapping != null) {
-            expression.setColumnOldName(fieldMapping.getTarget().getName());
-            for (AlterExpression.ColumnDataType columnDataType : expression.getColDataTypeList()) {
-                ddlConfig.setChangedColumnName(columnDataType.getColumnName());
-            }
-        }
-
-    }
-
-    //解析add的属性
-    //exampleSql: ALTER TABLE cost ADD duan INT after(before) `tmp`;
-    private void parseAdd(AlterExpression expression) {
-        for (AlterExpression.ColumnDataType columnDataType : expression.getColDataTypeList()) {
-            boolean findColumn = false;
-            List<String> columnSpecs = new LinkedList<>();
-            for (String spe:columnDataType.getColumnSpecs()) {//对一before，after进行处理
-                spe = StringUtil.replace(spe,"`","");
-                if (findColumn){
-                    //对before（after）字段进行映射
-                    String finalSpe = spe;
-                    FieldMapping fieldMapping = fieldMappingList.stream().filter(x->x.getSource().getName().equals(
-                            finalSpe)).findFirst().get();
-                    columnSpecs.add(fieldMapping.getTarget().getName());
-                    findColumn = false;
-                    continue;
-                }
-
-                if (StringUtil.equalsIgnoreCase(spe,"before") || StringUtil.equalsIgnoreCase(spe,"after")){
-                   findColumn =true;
-                }
-                columnSpecs.add(spe);
-            }
-            columnDataType.setColumnSpecs(columnSpecs);
-            String columName = columnDataType.getColumnName();
-            columName = StringUtil.replace(columName, "`", "");
-            Field field = new Field(columName,columnDataType.getColDataType().getDataType(),0);//感觉不需要都行，只需要名称，后续可以自己刷新
-            ddlConfig.getAddFields().add(field);
-        }
-    }
-
-    //解析modify的属性
-    //exampleSql: ALTER TABLE `test`.`test_table` MODIFY COLUMN `test` varchar(251) NULL DEFAULT NULL
-    private void parseModify(AlterExpression expression){
-        //先查找到当前的表和目标的表对应的字段
-        for (AlterExpression.ColumnDataType columnDataType : expression.getColDataTypeList()) {
-            String columName = columnDataType.getColumnName();
-            columName = StringUtil.replace(columName, "`", "");
-            for (FieldMapping fieldMapping : fieldMappingList) {
-                if (StringUtil.equals(fieldMapping.getSource().getName(), columName)) {
-                    //TODO life 找到目标的表名，先是alter进行属性替换，然后config记录新的
-                    columnDataType.setColumnName(
-                            fieldMapping.getTarget().getName());//alter语法树进行替换
-                    //因为只是修改属性，所以表名称没有变化
-                    ddlConfig.setSourceColumnName(fieldMapping.getSource().getName());
-                    ddlConfig.setChangedColumnName(fieldMapping.getSource().getName());
-                }
-            }
-        }
-    }
 }

dbsyncer-parser/src/main/java/org/dbsyncer/parser/ddl/alter/ChangeStrategy.java

@@ -0,0 +1,34 @@
+package org.dbsyncer.parser.ddl.alter;
+
+import java.util.List;
+import net.sf.jsqlparser.statement.alter.AlterExpression;
+import org.dbsyncer.common.util.StringUtil;
+import org.dbsyncer.connector.config.DDLConfig;
+import org.dbsyncer.parser.model.FieldMapping;
+
+/**
+ * 解析change属性
+ * exampleSql: ALTER TABLE test_table CHANGE duan1  duan2 INT(10)
+ * @author life
+ */
+public class ChangeStrategy implements ExpressionStrategy{
+
+
+    @Override
+    public void parse(AlterExpression expression, DDLConfig ddlConfig,
+            List<FieldMapping> fieldMappingList) {
+        String oldColumnName = expression.getColumnOldName();
+        oldColumnName = StringUtil.replace(oldColumnName,"`","");
+        ddlConfig.setSourceColumnName(oldColumnName);
+        String finalOldColumnName = oldColumnName;
+        FieldMapping fieldMapping = fieldMappingList.stream().filter(x->x.getSource().getName().equals(
+                finalOldColumnName)).findFirst().orElse(null);
+        if (fieldMapping != null) {
+            expression.setColumnOldName(fieldMapping.getTarget().getName());
+            for (AlterExpression.ColumnDataType columnDataType : expression.getColDataTypeList()) {
+                ddlConfig.setChangedColumnName(columnDataType.getColumnName());
+            }
+        }
+
+    }
+}

dbsyncer-parser/src/main/java/org/dbsyncer/parser/ddl/alter/DropStrategy.java

@@ -0,0 +1,31 @@
+package org.dbsyncer.parser.ddl.alter;
+
+import java.util.List;
+import net.sf.jsqlparser.statement.alter.AlterExpression;
+import org.dbsyncer.common.util.StringUtil;
+import org.dbsyncer.connector.config.DDLConfig;
+import org.dbsyncer.connector.model.Field;
+import org.dbsyncer.parser.model.FieldMapping;
+
+/**
+ * 解析drop
+ * example: ALTER TABLE test_table DROP dis;
+ * @author life
+ */
+public class DropStrategy implements ExpressionStrategy{
+
+    @Override
+    public void parse(AlterExpression expression, DDLConfig ddlConfig,
+            List<FieldMapping> fieldMappingList) {
+        String columName = expression.getColumnName();
+        columName = StringUtil.replace(columName, "`", "");
+        Field field = new Field(columName,null,0);
+        //需要把列替换成目标的列名
+        String finalColumName = columName;
+        fieldMappingList.stream().filter(x -> x.getSource().getName().equals(
+                finalColumName)).findFirst().ifPresent(
+                fieldMapping -> expression.setColumnName(fieldMapping.getTarget().getName()));
+        //加入还是原名
+        ddlConfig.getRemoveFields().add(field);
+    }
+}

dbsyncer-parser/src/main/java/org/dbsyncer/parser/ddl/alter/ExpressionStrategy.java

@@ -0,0 +1,13 @@
+package org.dbsyncer.parser.ddl.alter;
+
+import java.util.List;
+import net.sf.jsqlparser.statement.alter.AlterExpression;
+import org.dbsyncer.connector.config.DDLConfig;
+import org.dbsyncer.parser.model.FieldMapping;
+
+/**
+ * @author life
+ */
+public interface ExpressionStrategy {
+    void parse(AlterExpression expression, DDLConfig ddlConfig, List<FieldMapping> fieldMappingList);
+}

dbsyncer-parser/src/main/java/org/dbsyncer/parser/ddl/alter/ModifyStrategy.java

@@ -0,0 +1,36 @@
+package org.dbsyncer.parser.ddl.alter;
+
+import java.util.List;
+import net.sf.jsqlparser.statement.alter.AlterExpression;
+import org.dbsyncer.common.util.StringUtil;
+import org.dbsyncer.connector.config.DDLConfig;
+import org.dbsyncer.parser.model.FieldMapping;
+
+/**
+ * 解析modify的属性
+ * exampleSql: ALTER TABLE `test`.`test_table` MODIFY COLUMN `test` varchar(251) NULL DEFAULT NULL
+ * alter modify parser
+ * @author life
+ */
+public class ModifyStrategy implements ExpressionStrategy{
+
+    @Override
+    public void parse(AlterExpression expression, DDLConfig ddlConfig,
+            List<FieldMapping> fieldMappingList) {
+        //先查找到当前的表和目标的表对应的字段
+        for (AlterExpression.ColumnDataType columnDataType : expression.getColDataTypeList()) {
+            String columName = columnDataType.getColumnName();
+            columName = StringUtil.replace(columName, "`", "");
+            for (FieldMapping fieldMapping : fieldMappingList) {
+                if (StringUtil.equals(fieldMapping.getSource().getName(), columName)) {
+                    //TODO life 找到目标的表名，先是alter进行属性替换，然后config记录新的
+                    columnDataType.setColumnName(
+                            fieldMapping.getTarget().getName());//alter语法树进行替换
+                    //因为只是修改属性，所以表名称没有变化
+                    ddlConfig.setSourceColumnName(fieldMapping.getSource().getName());
+                    ddlConfig.setChangedColumnName(fieldMapping.getSource().getName());
+                }
+            }
+        }
+    }
+}