From 0ccc9a0dcadfb32aa8c63b718518ec0d216fae8f Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期四, 05 九月 2024 17:06:16 +0800
Subject: [PATCH] 保存到mongo时设置标签信息

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java |   37 +++++++++++++++++++++++--------------
 1 files changed, 23 insertions(+), 14 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java
index 309b427..35bf6ab 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java
@@ -6,12 +6,16 @@
 import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ycl.dataListener.CurrencyDataListener;
+import com.ycl.platform.domain.entity.ReportAuditingRecord;
 import com.ycl.platform.domain.entity.YwPeople;
 import com.ycl.platform.domain.entity.YwPoint;
 import com.ycl.platform.domain.excel.PointExport;
 import com.ycl.platform.domain.form.BatchEditPointForm;
 import com.ycl.platform.domain.form.YwPointForm;
 import com.ycl.platform.domain.query.YwPointQuery;
+import com.ycl.platform.domain.result.BaseResult;
+import com.ycl.platform.domain.vo.CheckTagVO;
+import com.ycl.platform.domain.vo.ReportAuditingRecordVO;
 import com.ycl.platform.domain.vo.YwPointVO;
 import com.ycl.platform.mapper.YwPeopleMapper;
 import com.ycl.platform.mapper.YwPointMapper;
@@ -200,20 +204,8 @@
 
     @Override
     public Result select(String keyword) {
-        Long userId = SecurityUtils.getUserId();
-        // 鑾峰彇褰撳墠杩愮淮浜哄憳鐨勮繍缁村崟浣�
-        YwPeople people = new LambdaQueryChainWrapper<>(ywPeopleMapper)
-                .eq(YwPeople::getUserId, userId)
-                .one();
-        Integer unitId = null;
-        if (Objects.nonNull(people)) {
-            unitId = people.getBelongUnit();
-        }
-
-        List<YwPoint> list = new LambdaQueryChainWrapper<>(baseMapper)
-                .eq(YwPoint::getUnitId, unitId)
-                .like(YwPoint::getPointName, keyword)
-                .list();
+        Integer unitId = SecurityUtils.getUnitId();
+        List<YwPoint> list = baseMapper.select(keyword, unitId);
         List data = list.stream().map(item -> {
             Map map = new HashMap<String, Object>();
             map.put("id", item.getSerialNumber());
@@ -260,6 +252,11 @@
         };
         EasyExcel.read(file.getInputStream(), PointExport.class , new CurrencyDataListener(consumer)).headRowNumber(1).doReadAll();
         return Result.ok();
+    }
+
+    @Override
+    public ReportAuditingRecordVO getReportResult(String serialNumber, String reportType) {
+        return baseMapper.getReportResult(serialNumber, reportType);
     }
 
     /**
@@ -315,4 +312,16 @@
         log.error("鎬诲叡鑰楁椂锛�" + (endTime.getTime() - startTime.getTime()));
     }
 
+    @Override
+    public <T extends BaseResult> void setDeviceTagByGB(List<T> dataList) {
+        List<String> gbList = dataList.stream().map(BaseResult::getNo).collect(Collectors.toList());
+        List<CheckTagVO> pointList = baseMapper.getTagsByGB(gbList);
+        Map<String, CheckTagVO> pointMap = pointList.stream().collect(Collectors.toMap(CheckTagVO::getNo, item -> item));
+        for (BaseResult data : dataList) {
+            CheckTagVO tag = pointMap.get(data.getNo());
+            if (Objects.nonNull(tag)) {
+                BeanUtils.copyProperties(tag, data);
+            }
+        }
+    }
 }

--
Gitblit v1.8.0