From b14531e3b850fe6d2fa916ba7b88b3e2bd2ff30a Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期三, 11 九月 2024 16:47:48 +0800
Subject: [PATCH] OSD加标签

---
 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