update field mapping

dbsyncer-biz/src/main/java/org/dbsyncer/biz/checker/impl/tablegroup/TableGroupChecker.java

@@ -1,17 +1,24 @@
 package org.dbsyncer.biz.checker.impl.tablegroup;
 
+import org.dbsyncer.biz.BizException;
 import org.dbsyncer.biz.checker.AbstractChecker;
 import org.dbsyncer.common.util.JsonUtil;
+import org.dbsyncer.connector.config.Field;
 import org.dbsyncer.manager.Manager;
 import org.dbsyncer.parser.model.FieldMapping;
 import org.dbsyncer.parser.model.TableGroup;
 import org.dbsyncer.storage.constant.ConfigConstant;
+import org.json.JSONArray;
+import org.json.JSONException;
+import org.json.JSONObject;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import org.springframework.util.Assert;
 
+import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
@@ -42,12 +49,51 @@ public class TableGroupChecker extends AbstractChecker {
         // 字段映射关系
         String fieldMappingJson = params.get("fieldMapping");
         Assert.hasText(fieldMappingJson, "TableGroupChecker check params fieldMapping is empty");
-        List<FieldMapping> fieldMapping = JsonUtil.jsonToObj(fieldMappingJson, List.class);
-        tableGroup.setFieldMapping(fieldMapping);
+        setFieldMapping(tableGroup, fieldMappingJson);
 
         // 修改高级配置：过滤条件/转换配置/插件配置
         this.modifySuperConfigModel(tableGroup, params);
 
         return JsonUtil.objToJson(tableGroup);
     }
+
+    /**
+     * 解析映射关系
+     *
+     * @param tableGroup
+     * @param json [{"source":"id","target":"id"}]
+     * @return
+     */
+    private void setFieldMapping(TableGroup tableGroup, String json) {
+        try {
+            JSONArray mapping = new JSONArray(json);
+            if(null == mapping){
+                throw new BizException("映射关系不能为空");
+            }
+
+            final Map<String, Field> sMap = convert2Map(tableGroup.getSourceTable().getColumn());
+            final Map<String, Field> tMap = convert2Map(tableGroup.getTargetTable().getColumn());
+            int length = mapping.length();
+            List<FieldMapping> list = new ArrayList<>();
+            JSONObject row = null;
+            Field s = null;
+            Field t = null;
+            for (int i = 0; i < length; i++) {
+                row = mapping.getJSONObject(i);
+                s = sMap.get(row.getString("source"));
+                t = tMap.get(row.getString("target"));
+                list.add(new FieldMapping(s, t));
+            }
+            tableGroup.setFieldMapping(list);
+        } catch (JSONException e) {
+            logger.error(e.getMessage());
+            throw new BizException(e.getMessage());
+        }
+    }
+
+    private Map<String, Field> convert2Map(List<Field> col) {
+        final Map<String, Field> map = new HashMap<>();
+        col.forEach(f -> map.put(f.getName(), f));
+        return map;
+    }
 }

dbsyncer-parser/src/main/java/org/dbsyncer/parser/model/FieldMapping.java

@@ -15,6 +15,14 @@ public class FieldMapping {
 
     private Field target;
 
+    public FieldMapping() {
+    }
+
+    public FieldMapping(Field source, Field target) {
+        this.source = source;
+        this.target = target;
+    }
+
     public Field getSource() {
         return source;
     }