From 7fa07718de92093c6315c5cb37bbc8dd7cdafa3c Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期三, 25 十二月 2024 10:28:43 +0800
Subject: [PATCH] 考核积分修改、点位导出修改

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java |   20 ++++++++++++++++++--
 1 files changed, 18 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..76da7e9 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,24 @@
 
     @Override
     public Result workList(DistributeWorkOrderQuery query) {
-        List<YwUnitVO> entities = baseMapper.workList(query);
-        return Result.ok().data(entities);
+        if(query.getStart()!=null) {
+            query.setStart(DateUtils.getDayStart(query.getStart()));
+        }
+        if(query.getEnd()!=null) {
+            query.setEnd(DateUtils.getDayEnd(query.getEnd()));
+        }
+        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