From 165541429aa52cb97b2cea18f7701bff2efed1f2 Mon Sep 17 00:00:00 2001
From: 龚焕茏 <2842157468@qq.com>
Date: 星期一, 08 四月 2024 18:01:20 +0800
Subject: [PATCH] 运维单位

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/ReportServiceImpl.java |   47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 47 insertions(+), 0 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/ReportServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/ReportServiceImpl.java
index 415104c..f863d3f 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/ReportServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/ReportServiceImpl.java
@@ -26,6 +26,7 @@
 import org.springframework.util.Assert;
 import org.springframework.util.CollectionUtils;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Objects;
 import java.util.stream.Collectors;
@@ -187,4 +188,50 @@
                 .collect(Collectors.toList());
         return Result.ok().data(vos);
     }
+
+    public List<ReportVO> export(ReportQuery query) {
+
+        if (StringUtils.isNotBlank(query.getPointId())) {
+            LambdaQueryWrapper<YwPoint> queryWrapper = new LambdaQueryWrapper<>();
+            queryWrapper.like(YwPoint::getPointName, query.getPointId());
+            query.setPointIdList(ywpointMapper.selectList(queryWrapper).stream().map(YwPoint::getId).collect(Collectors.toList()));
+            if(CollectionUtils.isEmpty(query.getPointIdList())) {
+                return new ArrayList<>();
+            }
+        }
+        if (StringUtils.isNotBlank(query.getPeopleId())) {
+            LambdaQueryWrapper<YwPeople> queryWrapper = new LambdaQueryWrapper<>();
+            queryWrapper.like(YwPeople::getYwPersonName, query.getPeopleId());
+            query.setPeopleIdList(peopleMapper.selectList(queryWrapper).stream().map(YwPeople::getId).collect(Collectors.toList()));
+            if(CollectionUtils.isEmpty(query.getPeopleIdList())) {
+                return new ArrayList<>();
+            }
+        }
+
+        IPage<Report> page = new LambdaQueryChainWrapper<>(baseMapper)
+                .eq(StringUtils.isNotBlank(query.getReportType()), Report::getReportType, query.getReportType())
+                .le(Objects.nonNull(query.getBeginCreateTime()), Report::getBeginCreateTime, query.getBeginCreateTime())
+                .ge(Objects.nonNull(query.getEndCreateTime()), Report::getEndCreateTime, query.getEndCreateTime())
+                .in(!CollectionUtils.isEmpty(query.getPointIdList()), Report::getPointId, query.getPointIdList())
+                .in(!CollectionUtils.isEmpty(query.getPeopleIdList()), Report::getPeopleId, query.getPeopleIdList())
+                .orderByDesc(Report::getCreateTime)
+                .page(PageUtil.getPage(query, Report.class));
+
+        List<ReportVO> vos = page.getRecords().stream()
+                .map(
+                        entity -> {
+                            ReportVO vo = ReportVO.getVoByEntity(entity, null);
+                            YwUnit unit = unitMapper.selectById(vo.getUnitId());
+                            vo.setUnitName(unit.getUnitName());
+                            YwPeople ywPeople = peopleMapper.selectById(vo.getPeopleId());
+                            vo.setPeopleName(ywPeople.getYwPersonName());
+                            YwPoint ywPoint = ywpointMapper.selectById(vo.getPointId());
+                            vo.setPointName(ywPoint.getPointName());
+                            return vo;
+                        }
+                )
+                .collect(Collectors.toList());
+        return vos;
+    }
+
 }

--
Gitblit v1.8.0