尝试Pairs upsert

src/main/java/com/luca/pilot/service/impl/PlanServiceImpl.java

 package com.luca.pilot.service.impl;
 
+
 import cn.hutool.core.map.MapUtil;
 import cn.hutool.json.JSONUtil;
 import com.luca.pilot.bizEnum.TableEnum;
@@ -18,6 +19,7 @@ import org.springframework.data.mongodb.core.mapping.Document;
 import org.springframework.data.mongodb.core.query.Criteria;
 import org.springframework.data.mongodb.core.query.Query;
 import org.springframework.data.mongodb.core.query.Update;
+import org.springframework.data.util.Pair;
 import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
@@ -76,7 +78,7 @@ public class PlanServiceImpl implements PlanService {
 
         long totalPlanModuleCount = mongoTemplate.count(new Query(), PlanModule.class);
 
-//        BulkOperations bulkOps = mongoTemplate.bulkOps(BulkOperations.BulkMode.UNORDERED, PlanModule.class);
+        BulkOperations bulkOps = mongoTemplate.bulkOps(BulkOperations.BulkMode.UNORDERED, PlanModule.class);
 
         for (int i = 0; i < skip; i++) {
             countQuery.skip(i * 10000L);
@@ -103,6 +105,7 @@ public class PlanServiceImpl implements PlanService {
             Map<Long, Map<Long, String>> digitalMedicalTypeMap = queryDigitalMedicalType(planTaskDetails);
 
             List<PlanModule> planMoudles = new ArrayList<>();
+            List<Pair<Query, Update>> pairs = new ArrayList<>();
 
             //以 planTaskDetail 和 planMoudleRelate 为基准
             // 一个 planTaskDetail 能对应多个 planMoudleRelate，即一次任务配置了多少个模块（问卷，量表，数字靶点）
@@ -232,21 +235,23 @@ public class PlanServiceImpl implements PlanService {
                     if (totalPlanModuleCount > 0) {
                         Query upsertQuery = new Query(where("key").is(planMoudle.getKey()));
                         Update update = buildUpdate(planMoudle);
-//                        bulkOps.upsert(upsertQuery, update);
-                        mongoTemplate.upsert(upsertQuery, update, PlanModule.class);
-                        log.info("mongoTemplate upsert 1 条数据");
+                        bulkOps.upsert(upsertQuery, update);
+                        Pair<Query, Update> pair = Pair.of(upsertQuery, update);
+                        pairs.add(pair);
+                    }else {
+                        planMoudles.add(planMoudle);
                     }
-                    planMoudles.add(planMoudle);
                 }
             }
 
 
             // 是否第一次全量插入
             if (totalPlanModuleCount > 0) {
-//                CompletableFuture.runAsync(() -> {
-//                bulkOps.execute();
-//                log.info("bulkOps 更新或插入 {}  条数据", planMoudles.size());
-//                });
+                CompletableFuture.runAsync(() -> {
+                    bulkOps.upsert(pairs);
+                    bulkOps.execute();
+                    log.info("bulkOps 更新或插入 {}  条数据", pairs.size());
+                });
             } else {
                 if (!CollectionUtils.isEmpty(planMoudles)) {
                     mongoTemplate.insert(planMoudles, PlanModule.class);