222

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

@@ -36,4 +36,15 @@ public abstract class BaseTask implements SqlTask {
         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) {
+            Map<String, Object> map = list.get(0);
+            for (Map.Entry<String, Object> entry : map.entrySet()) {
+                return entry.getValue();
+            }
+        }
+        return null;
+    }
+
 }

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

@@ -31,6 +31,8 @@ public class StartApplication {
     XyGroupTask xyGroupTask;
     @Autowired
     CompanyTask companyTask;
+    @Autowired
+    SqlTask roleTask;
 
     @Value("${application.name}")
     private String appName;
@@ -51,7 +53,7 @@ public class StartApplication {
 
 //            userGroupTask.start();
 
-            userTask.loadCache();
+//            roleTask.start();
 
             System.out.println("AAAAA");
         }

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

@@ -1,12 +1,17 @@
 package com.dashitech.migration.task.uc;
 
 import com.dashitech.migration.BaseTask;
+import com.dashitech.migration.SqlTask;
 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 org.springframework.transaction.annotation.Transactional;
 
+import java.util.List;
+
 /**
  * Created by CX on 2024/10/10.
  */
@@ -15,20 +20,40 @@ public class RoleTask extends BaseTask {
 
     public static Log log = LogFactory.getLog(RoleTask.class);
 
+    @Autowired
+    SqlTask userTask;
+
     @Override
     @Transactional
     public void start() {
+        log.info(" *** 加载角色Role缓存 *** ");
 
-    }
+        StringBuffer sb = new StringBuffer();
+        //只拿运维人员
+        sb.append(" SELECT id as 'user_id' from itsm_uc_user where ACCOUNT != 'dsadmin' and delete_flag = 0 ");
 
-    @Override
-    public FieldMapper loadCache() {
-        log.info(" *** 加载角色Role缓存 *** ");
+        Object roleId = getSingleObj(jdbcTemplateHsms," SELECT id from itsm_uc_role WHERE rolecode = 'first-line support' ");
+
+        List list = jdbcTemplateMdv2.queryForList(sb.toString());
 
         FieldMapper mapper = new FieldMapper();
+        FieldMapper userMapper = userTask.loadCache();
+        mapper.addOperationAdd("role_id", Long.valueOf(roleId.toString()));
+        mapper.addOperationReplace("user_id", userMapper.getIdCache());
+        mapper.trans2JdbcRow(list);
+
+        JdbcUtil jdbcUtil = new JdbcUtil(jdbcTemplateHsms);
+        jdbcUtil.batchPageInsert("itsm_uc_user_role", mapper.getRows());
 
+        //clear释放内存
+        mapper.clear();
+        userMapper.clear();
+    }
 
+    @Override
+    public FieldMapper loadCache() {
+        log.info(" *** 加载角色Role缓存 *** ");
 
-        return mapper;
+        return null;
     }
 }

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

@@ -48,8 +48,6 @@ public class UserGroupTask extends BaseTask {
         mapper.addOperationReplace("user_id", userMapper.getIdCache());
         mapper.trans2JdbcRow(list);
 
-
-
         JdbcUtil jdbcUtil = new JdbcUtil(jdbcTemplateHsms);
         jdbcUtil.batchPageInsert("itsm_uc_user_group", mapper.getRows());