perf: 对执行对象任务写入 db 进行分批处理 #3312

TencentBlueKing · Nov 28, 2024 · a9ad191 · a9ad191
1 parent 66a3252
commit a9ad191
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/...ce-job-execute/src/main/java/com/tencent/bk/job/execute/dao/impl/TaskInstanceDAOImpl.java b/...ce-job-execute/src/main/java/com/tencent/bk/job/execute/dao/impl/TaskInstanceDAOImpl.java
@@ -30,7 +30,7 @@
 import com.tencent.bk.job.common.mysql.dynamic.ds.DbOperationEnum;
 import com.tencent.bk.job.common.mysql.dynamic.ds.MySQLOperation;
 import com.tencent.bk.job.common.mysql.jooq.JooqDataTypeUtil;
-import com.tencent.bk.job.common.util.CollectionUtil;
+import com.tencent.bk.job.common.util.BatchUtil;
 import com.tencent.bk.job.execute.common.constants.RunStatusEnum;
 import com.tencent.bk.job.execute.dao.TaskInstanceDAO;
 import com.tencent.bk.job.execute.dao.common.DSLContextProviderFactory;
@@ -527,11 +527,7 @@ public List<Long> listTaskInstanceId(Long appId, Long fromTime, Long toTime, int
     public void saveTaskInstanceHosts(long appId,
                                       long taskInstanceId,
                                       Collection<HostDTO> hosts) {
-        if (CollectionUtils.isEmpty(hosts)) {
-            return;
-        }
-        List<List<HostDTO>> hostBatches = CollectionUtil.partitionCollection(hosts, 2000);
-        hostBatches.forEach(batchHosts -> {
+        BatchUtil.executeBatch(hosts, 2000, batchHosts -> {
             BatchBindStep batchInsert = dsl().batch(
                 dsl().insertInto(
                         TASK_INSTANCE_HOST,