From 088bc1e162261a91a8779d8079bef9d6a5d6d3e9 Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期二, 03 九月 2024 10:06:22 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java |   14 ++++++++++++--
 1 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java
index 1b5c7fa..61ebd1f 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java
@@ -216,8 +216,18 @@
 
     @Override
     public Result workList(DistributeWorkOrderQuery query) {
-        List<YwUnitVO> entities = baseMapper.workList(query);
-        return Result.ok().data(entities);
+        List<YwUnit> ywUnits = baseMapper.selectList(null);
+        List<YwUnitVO> lists = ywUnits.stream().map(item -> {
+            YwUnitVO vo = new YwUnitVO();
+            BeanUtils.copyProperties(item, vo);
+            return vo;
+        }).collect(Collectors.toList());
+        for (YwUnitVO ywUnit : lists) {
+            query.setUnitId(ywUnit.getId());
+            Integer workOrderNum = baseMapper.workList(query);
+            ywUnit.setWorkOrderCount(workOrderNum);
+        }
+        return Result.ok().data(lists);
     }
 
     @Override

--
Gitblit v1.8.0