修改使用统计报表接口

jyzb-biz/src/main/java/com/junmp/jyzb/service/impl/TjServiceImpl.java

@@ -737,15 +737,46 @@ public class TjServiceImpl implements TjService {
         PubOrg pubOrg = pubOrgService.PubOrgExist(req.getOrgId());
         List<String> orgList= inventorySummaryMapper.getTotalData(String.valueOf(req.getOrgId()));
         List<TjOrgCountDto> tjOrgCountList=new ArrayList<>();
-        for (String orgId:orgList) {
-            TjOrgCountDto orgData =equipmentCountSummaryMapper.TjOrgCount(orgId,
-                    req.getTypeIdsList(),req.getSizeNameList(),req.getYear(),req.getTypeParentId(),
-                    req.getMonthList());
-            orgData.setNumber(ObjectUtil.isNull(orgData.getNumber())?0:orgData.getNumber());
-            orgData.setUseCount(ObjectUtil.isNull(orgData.getUseCount())?0:orgData.getUseCount());
-            orgData.setFixCount(ObjectUtil.isNull(orgData.getFixCount())?0:orgData.getFixCount());
-            tjOrgCountList.add(orgData);
-        }
+//        for (String orgId:orgList) {
+//            TjOrgCountDto orgData =equipmentCountSummaryMapper.TjOrgCount(orgId,
+//                    req.getTypeIdsList(),req.getSizeNameList(),req.getYear(),req.getTypeParentId(),
+//                    req.getMonthList());
+//            orgData.setNumber(ObjectUtil.isNull(orgData.getNumber())?0:orgData.getNumber());
+//            orgData.setUseCount(ObjectUtil.isNull(orgData.getUseCount())?0:orgData.getUseCount());
+//            orgData.setFixCount(ObjectUtil.isNull(orgData.getFixCount())?0:orgData.getFixCount());
+//            tjOrgCountList.add(orgData);
+//        }
+
+        List<CompletableFuture<TjOrgCountDto>> futures = orgList.stream()
+                .map(orgId -> CompletableFuture.supplyAsync(() ->
+                        equipmentCountSummaryMapper.TjOrgCount(orgId,
+                                req.getTypeIdsList(), req.getSizeNameList(),
+                                req.getYear(), req.getTypeParentId(), req.getMonthList()))
+                )
+                .collect(Collectors.toList());
+// 等待所有 CompletableFuture 完成
+        CompletableFuture<Void> allOf = CompletableFuture.allOf(
+                futures.toArray(new CompletableFuture[0])
+        );
+        // 处理每个 CompletableFuture 的结果
+        allOf.thenRun(() -> {
+            for (CompletableFuture<TjOrgCountDto> future : futures) {
+                try {
+                    TjOrgCountDto orgData = future.get(); // 获取 CompletableFuture 的结果
+                    if (orgData != null) {
+                        orgData.setNumber(ObjectUtil.isNull(orgData.getNumber()) ? 0 : orgData.getNumber());
+                        orgData.setUseCount(ObjectUtil.isNull(orgData.getUseCount()) ? 0 : orgData.getUseCount());
+                        orgData.setFixCount(ObjectUtil.isNull(orgData.getFixCount()) ? 0 : orgData.getFixCount());
+                        tjOrgCountList.add(orgData);
+                    }
+                } catch (InterruptedException | ExecutionException e) {
+                    // 处理异常
+                }
+            }
+        });
+
+// 等待所有 CompletableFuture 完成
+        allOf.join();
         if (req.getOrgId()!=1369509498032808905L){
             TjOrgCountDto orgData =equipmentCountSummaryMapper.TjOrgCountleve4(req.getOrgId(),
                     req.getTypeIdsList(),req.getSizeNameList(),req.getYear(),req.getTypeParentId(),