添加耗材转产品数据转换

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

@@ -36,6 +36,18 @@ public abstract class BaseTask implements SqlTask {
         return null;
     }
 
+    /**
+     * 获取树形字典id
+     */
+    public Long getDicTreeId(JdbcTemplate jt,String key, String value) {
+        List<Map<String,Object>> list = jt.queryForList("SELECT id from itsm_base_dictionary_tree WHERE _key = ? and _value = ?",key,value);
+        if(list != null && list.size() > 0) {
+            Object obj = list.get(0).get("id");
+            return obj == null ? null : Long.valueOf(obj.toString());
+        }
+        return null;
+    }
+
     public Object getSingleObj(JdbcTemplate jt,String sql, Object... args) {
         List<Map<String,Object>> list = jt.queryForList(sql,args);
         if(list != null && list.size() > 0) {

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

@@ -1,6 +1,5 @@
 package com.dashitech.migration;
 
-import com.dashitech.migration.core.FieldMapper;
 import com.dashitech.utils.SpringContextHolder;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
@@ -40,8 +39,8 @@ public class StartApplication {
 //        task.start();
 //        task = SpringContextHolder.getBean("companyTask");
 //        task.start();
-        task = SpringContextHolder.getBean("userTask");
-        task.start();
+//        task = SpringContextHolder.getBean("userTask");
+//        task.start();
 //        task = SpringContextHolder.getBean("userGroupTask");
 //        task.start();
 //        task = SpringContextHolder.getBean("roleUserTask");
@@ -68,8 +67,10 @@ public class StartApplication {
 //        task.start();
 //        task = SpringContextHolder.getBean("solutionIntroduceTask");
 //        task.start();
-//        task = SpringContextHolder.getBean("consumableTask");
+//        task = SpringContextHolder.getBean("productTask");
 //        task.start();
+        task = SpringContextHolder.getBean("consumableTask");
+        task.start();
 //        task = SpringContextHolder.getBean("consumableHistoryTask");
 //        task.start();
 //        task = SpringContextHolder.getBean("repairSummaryTask");
@@ -80,7 +81,6 @@ public class StartApplication {
 //        task.start();
 //        task = SpringContextHolder.getBean("hjzxRecordTask");
 //        task.start();
-		
 
 
         log.info("########### 迁移任务完成！###########");

src/main/java/com/dashitech/migration/core/DataMapper.java

@@ -19,10 +19,14 @@ public class DataMapper implements Serializable{
     private Object sourceValue;
     private Long sourceId;
 
+    //匹配字段
+    private String matchFieldName;
+
     //操作类型
     private boolean remove = false;
     private boolean update = false;
     private boolean add = false;
+    private boolean addFromMatchField = false;
     private boolean replace = false;
     private boolean nullDefault = false;
 
@@ -31,6 +35,8 @@ public class DataMapper implements Serializable{
      */
     private Map<Long, Long> idCache = new HashMap<>();
 
+    private Map<String, Long> targetCache = new HashMap<>();
+
     public DataMapper() {
     }
 
@@ -39,6 +45,12 @@ public class DataMapper implements Serializable{
         this.sourceName = sourceName;
     }
 
+    public DataMapper(String targetName, String sourceName,String matchFieldName) {
+        this.targetName = targetName;
+        this.sourceName = sourceName;
+        this.matchFieldName = matchFieldName;
+    }
+
     public boolean isNullDefault() {
         return nullDefault;
     }
@@ -135,4 +147,27 @@ public class DataMapper implements Serializable{
         this.sourceId = sourceId;
     }
 
+    public Map<String, Long> getTargetCache() {
+        return targetCache;
+    }
+
+    public void setTargetCache(Map<String, Long> targetCache) {
+        this.targetCache = targetCache;
+    }
+
+    public boolean isAddFromMatchField() {
+        return addFromMatchField;
+    }
+
+    public void setAddFromMatchField(boolean addFromMatchField) {
+        this.addFromMatchField = addFromMatchField;
+    }
+
+    public String getMatchFieldName() {
+        return matchFieldName;
+    }
+
+    public void setMatchFieldName(String matchFieldName) {
+        this.matchFieldName = matchFieldName;
+    }
 }

src/main/java/com/dashitech/migration/core/FieldMapper.java

@@ -64,6 +64,15 @@ public class FieldMapper {
                             row.put(dataMapper.getTargetName(), dataMapper.getTargetValue());
                         }
                     }
+                    //如果是根据指定字段匹配新增
+                    if (dataMapper.isAddFromMatchField() && row.containsKey(dataMapper.getMatchFieldName())) {
+                        Object value = row.get(dataMapper.getMatchFieldName());
+                        if (value != null) {
+                            row.put(dataMapper.getSourceName(), dataMapper.getTargetCache().get(value));
+                        } else {
+                            row.put(dataMapper.getSourceName(), null);
+                        }
+                    }
                     //如果是要替换的，使用id缓存替换
                     if (dataMapper.isReplace() && row.containsKey(dataMapper.getSourceName())) {
                         Object value = row.get(dataMapper.getSourceName());
@@ -142,6 +151,17 @@ public class FieldMapper {
     }
 
     /**
+     * 待新增的字段 需要通过指定sourceName字段匹配
+     */
+    public FieldMapper addOperationAdd(String name, String matchFieldName,Map<String,Long> targetCache) {
+        DataMapper mapper = new DataMapper(name, name, matchFieldName);
+        mapper.setAddFromMatchField(true);
+        mapper.setTargetCache(targetCache);
+        rowsOperationList.add(mapper);
+        return this;
+    }
+
+    /**
      * 待替换的字段，每次加入都在队列最前面。注意：如果是loadCache过滤，使用target名称，如果是start转换，使用source名称
      */
     public FieldMapper addOperationReplace(String name, Map<Long, Long> idCache) {

src/main/java/com/dashitech/migration/task/consumable/ConsumableTask.java

@@ -29,6 +29,9 @@ public class ConsumableTask extends BaseTask {
     @Autowired
     private CompanyTask companyTask;
 
+    @Autowired
+    private ProductTask productTask;
+
     @Override
     @Transactional(rollbackFor = Exception.class)
     public void start() {
@@ -39,6 +42,7 @@ public class ConsumableTask extends BaseTask {
         FieldMapper mapper = new FieldMapper();
         mapper.addOperationReplace("hos_id", hospitalTask.loadCache().getIdCache());
         mapper.addOperationReplace("supplier_id", companyTask.loadCache().getIdCache());
+        mapper.addOperationAdd("product_id","spd_code", productTask.loadCache().getTargetCache());
         mapper.trans2JdbcRow(consumableList);
 
         JdbcUtil jdbcUtil = new JdbcUtil(jdbcTemplateHsms);

src/main/java/com/dashitech/migration/task/consumable/ProductTask.java

@@ -0,0 +1,61 @@
+package com.dashitech.migration.task.consumable;
+
+import com.dashitech.migration.BaseTask;
+import com.dashitech.migration.core.DataMigrationter;
+import com.dashitech.migration.core.FieldMapper;
+import com.dashitech.migration.task.uc.HospitalTask;
+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 org.springframework.transaction.annotation.Transactional;
+
+import java.util.List;
+import java.util.Map;
+
+/**
+ * @author song
+ * @date 2024年10月12日
+ */
+@Component
+public class ProductTask extends BaseTask {
+
+    public static Log log = LogFactory.getLog(ProductTask.class);
+
+    @Autowired
+    private HospitalTask hospitalTask;
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void start() {
+        log.info("耗材转产品ProductTask转换开始");
+        StringBuffer sql = new StringBuffer("SELECT NAME,IFNULL(spd_code,spdcode) as code,brand_model as brand,specification as model,");
+        sql.append("unit,duty_id as hos_id,delete_flag FROM itsm_asset_consumable");
+        List<Map<String, Object>> productList = jdbcTemplateMdv2.queryForList(sql.toString());
+        FieldMapper mapper = new FieldMapper();
+        mapper.addOperationReplace("hos_id", hospitalTask.loadCache().getIdCache());
+        mapper.addOperationAdd("main_type", getDicTreeId(jdbcTemplateHsms, "product_category_type", "1008"));
+        mapper.trans2JdbcRow(productList);
+
+        JdbcUtil jdbcUtil = new JdbcUtil(jdbcTemplateHsms);
+        jdbcUtil.batchPageInsert("itsm_itsm_asset_product", mapper.getRows());
+
+        mapper.clear();
+
+    }
+
+    @Override
+    public FieldMapper loadCache() {
+        log.info(" *** 加载product缓存 *** ");
+        StringBuffer sql = new StringBuffer("select id,code from itsm_itsm_asset_product");
+        List<Map<String, Object>> productList = jdbcTemplateHsms.queryForList(sql.toString());
+        sql = new StringBuffer("SELECT id,IFNULL(spd_code,spdcode) as code FROM itsm_asset_consumable");
+        List<Map<String, Object>> consumableList = jdbcTemplateMdv2.queryForList(sql.toString());
+        FieldMapper mapper = new FieldMapper();
+        mapper.addFieldMapper("code", "code");
+
+        DataMigrationter.loadCache(productList, consumableList, mapper);
+        return mapper;
+    }
+}