添加项目标识获取不同缓存和运行不同转换

src/main/java/com/dashitech/migration/BaseTask.java

@@ -2,6 +2,7 @@ package com.dashitech.migration;
 
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.jdbc.core.JdbcTemplate;
 
 import java.util.LinkedHashMap;
@@ -14,6 +15,9 @@ public abstract class BaseTask implements SqlTask {
 
     protected Map cache = new LinkedHashMap();
 
+    @Value("${application.name}")
+    protected String appName;
+
     @Autowired
     @Qualifier("jdbcTemplateHsms")
     protected JdbcTemplate jdbcTemplateHsms;

src/main/java/com/dashitech/migration/StartApplication.java

@@ -1,9 +1,11 @@
 package com.dashitech.migration;
 
 import com.dashitech.migration.task.uc.HospitalTask;
+import com.dashitech.migration.task.uc.XyGroupTask;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 
 /**
@@ -24,13 +26,29 @@ public class StartApplication {
     @Autowired
     HospitalTask hospitalTask;
 
+    @Autowired
+    XyGroupTask xyGroupTask;
+
+    @Value("${application.name}")
+    private String appName;
+
     public void start() {
         log.info("########### 开始迁移任务！###########");
 
-        //工作组
-//        groupTask.start();
+        if ("xiangyang".equals(appName)) {
+            //责任科室转院区
+            //hospitalTask.start();
+            //工作组
+            //xyGroupTask.start();
+        } else {
+            //工作组
+            groupTask.start();
+        }
+
+
+
 
-//        userTask.start();
+        userTask.start();
 
         groupTask.loadCache();
 

src/main/java/com/dashitech/migration/task/uc/DictionaryTask.java

@@ -1,4 +1,4 @@
-package com.dashitech.migration.task.uc;
+package com.dashitech.migration.task.other;
 
 import com.dashitech.migration.BaseTask;
 import com.dashitech.migration.core.FieldMapper;

src/main/java/com/dashitech/migration/task/uc/SolutionTask.java

@@ -1,4 +1,4 @@
-package com.dashitech.migration.task.uc;
+package com.dashitech.migration.task.solution;
 
 import com.dashitech.migration.BaseTask;
 import com.dashitech.migration.core.FieldMapper;

src/main/java/com/dashitech/migration/task/uc/HospitalTask.java

@@ -1,6 +1,7 @@
 package com.dashitech.migration.task.uc;
 
 import com.dashitech.migration.BaseTask;
+import com.dashitech.migration.core.DataMigrationter;
 import com.dashitech.migration.core.FieldMapper;
 import com.dashitech.migration.util.JdbcUtil;
 import org.apache.commons.logging.Log;
@@ -27,9 +28,10 @@ public class HospitalTask extends BaseTask {
         sql.append(" FROM `itsm_uc_duty_dept` where dept not like 'Z%'  and dept not like '%保卫%' and id not in (20,21)");
         List<Map<String, Object>> hosList = jdbcTemplateMdv2.queryForList(sql.toString());
         FieldMapper mapper = new FieldMapper();
+        mapper.addOperationAdd("type", 20390L);//院区类型 维修科室
         mapper.trans2JdbcRow(hosList);
         JdbcUtil jdbcUtil = new JdbcUtil(jdbcTemplateHsms);
-        //jdbcUtil.batchPageInsert("itsm_uc_hospital", mapper.getRows());
+        jdbcUtil.batchPageInsert("itsm_uc_hospital", mapper.getRows());
 
         //clear释放内存
         mapper.clear();
@@ -45,11 +47,14 @@ public class HospitalTask extends BaseTask {
     }
 
     public FieldMapper loadXiangyangCache() {
+        StringBuffer sql = new StringBuffer("SELECT id,dept FROM `itsm_uc_duty_dept` where dept not like 'Z%'  and dept not like '%保卫%' and id not in (20,21)");
+        List<Map<String, Object>> dutyList = jdbcTemplateMdv2.queryForList(sql.toString());
+        sql = new StringBuffer("select id,_hos_name from itsm_uc_hospital where type = (select id from itsm_base_dictionary where _key = 'hospital_type' and _value = '6')");
+        List<Map<String, Object>> hosList = jdbcTemplateHsms.queryForList(sql.toString());
         FieldMapper mapper = new FieldMapper();
-        Map<Long, Long> hosIdCache = new HashMap<>();
-        mapper.setIdCache(hosIdCache);
-
+        mapper.addFieldMapper("_hos_name", "dept");
 
+        DataMigrationter.loadCache(hosList, dutyList, mapper);
 
         return mapper;
     }

src/main/java/com/dashitech/migration/task/uc/UserTask.java

@@ -22,6 +22,9 @@ public class UserTask extends BaseTask {
     @Autowired
     SqlTask groupTask;
 
+    @Autowired
+    HospitalTask hospitalTask;
+
     public static Log log = LogFactory.getLog(UserTask.class);
 
     @Override
@@ -35,6 +38,12 @@ public class UserTask extends BaseTask {
         List list = jdbcTemplateHsms.queryForList(sb.toString());
 
         FieldMapper groupMapper = groupTask.loadCache();
+        Map<Long, Long> hosIdCache = null;
+        if ("xiangyang".equals(appName)) {
+            hosIdCache = hospitalTask.loadXiangyangCache().getIdCache();
+        } else {
+            hosIdCache = hospitalTask.loadNanjingCache().getIdCache();
+        }
         System.out.println(groupMapper.getIdCache());
 
 

src/main/java/com/dashitech/migration/task/uc/XyGroupTask.java

@@ -0,0 +1,56 @@
+package com.dashitech.migration.task.uc;
+
+import com.dashitech.migration.BaseTask;
+import com.dashitech.migration.core.FieldMapper;
+import com.dashitech.migration.util.JdbcUtil;
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.Map;
+
+/**
+ * @author song
+ * @date 2024年10月12日
+ */
+@Component
+public class XyGroupTask extends BaseTask {
+
+    public static Log log = LogFactory.getLog(XyGroupTask.class);
+
+    @Autowired
+    private HospitalTask hospitalTask;
+
+
+    @Override
+    public void start() {
+        log.info(" *** 襄阳工作组group转换开始 *** ");
+
+        StringBuffer sql = new StringBuffer("select 0 as version,a.groupname,a.delete_flag,a.duty_id ,now() as operator_time,1 as schedule_class_id");
+        sql.append(" from itsm_uc_groups a join itsm_uc_duty_dept b on a.duty_id = b.id ");
+        sql.append("where b.dept not like 'Z%' and dept not like '%保卫%' and b.id not in (20,21)");
+
+        List<Map<String, Object>> groupList = jdbcTemplateMdv2.queryForList(sql.toString());
+
+        //测试
+        FieldMapper mapper = new FieldMapper();
+        mapper.addOperationUpdate("_hospital_id", "duty_id");
+        mapper.addOperationAdd("type", 3);
+        Map<Long, Long> hosIdCache = hospitalTask.loadXiangyangCache().getIdCache();
+        mapper.addOperationReplace("duty_id", hosIdCache);
+        mapper.trans2JdbcRow(groupList);
+
+        JdbcUtil jdbcUtil = new JdbcUtil(jdbcTemplateHsms);
+        //jdbcUtil.batchPageInsert("itsm_uc_groups", mapper.getRows());
+
+        //clear释放内存
+        mapper.clear();
+    }
+
+    @Override
+    public FieldMapper loadCache() {
+        return null;
+    }
+}

src/main/resources/application.properties

@@ -53,3 +53,6 @@ server.tomcat.compression=on
 
 spring.jackson.parser.allow-numeric-leading-zeros=true
 
+#\u5F53\u524D\u73AF\u5883\u540D\u79F0
+application.name=xiangyang
+