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