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/CheckScoreServiceImpl.java |  244 ++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 210 insertions(+), 34 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/CheckScoreServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/CheckScoreServiceImpl.java
index 0137fda..4a3be48 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/CheckScoreServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/CheckScoreServiceImpl.java
@@ -2,13 +2,15 @@
 
 import annotation.DataScope;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
-import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ycl.platform.base.CheckIndex;
+import com.ycl.platform.domain.dto.CheckResultExportDTO;
 import com.ycl.platform.domain.dto.CheckScoreDTO;
 import com.ycl.platform.domain.dto.CheckScoreIndexDTO;
 import com.ycl.platform.domain.dto.ScoreIndexDTO;
 import com.ycl.platform.domain.entity.*;
-import com.ycl.platform.domain.vo.CheckScoreDetailVO;
+import com.ycl.platform.domain.vo.*;
 import com.ycl.platform.mapper.CheckRuleMapper;
 import com.ycl.platform.mapper.CheckScoreMapper;
 import com.ycl.platform.mapper.CheckTemplateMapper;
@@ -17,6 +19,7 @@
 import com.ycl.platform.service.ICheckIndexFaceService;
 import com.ycl.platform.service.ICheckIndexVideoService;
 import com.ycl.platform.service.ICheckScoreService;
+import com.ycl.system.entity.BaseEntity;
 import com.ycl.system.entity.SysRole;
 import com.ycl.system.entity.SysUser;
 import com.ycl.system.service.ISysDeptService;
@@ -29,14 +32,18 @@
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.CollectionUtils;
 import pojo.ExcelExp;
 import utils.poi.ExcelUtilManySheet;
 
 import java.io.IOException;
+import java.math.BigDecimal;
+import java.math.RoundingMode;
 import java.text.SimpleDateFormat;
 import java.time.LocalDateTime;
 import java.util.*;
+import java.util.function.Function;
 import java.util.stream.Collectors;
 
 /**
@@ -65,6 +72,7 @@
     @Autowired
     private ISysDeptService deptService;
 
+
     /**
      * 鏌ヨ鑰冩牳绉垎鎸囨爣
      *
@@ -82,10 +90,10 @@
         Integer templateId = 0;
         //鏂逛究鍒囨崲鏀逛负锛氭牴鎹畇core鑰冩牳绫诲埆鍜宒to鑰冩牳鏍囩鏌ユā鏉胯鍒欏拰鏉冮噸,娉ㄦ剰鍚岀绫诲悓鏍囩鍙兘鏈変竴涓ā鏉�
         QueryWrapper<CheckTemplate> wrapper = new QueryWrapper<>();
-        wrapper.eq("examine_tag",checkScoreIndexDTO.getExamineTag());
-        wrapper.eq("examine_category",checkScore.getExamineCategory());
+        wrapper.eq("examine_tag", checkScoreIndexDTO.getExamineTag());
+        wrapper.eq("examine_category", checkScore.getExamineCategory());
         Optional<CheckTemplate> first = templateMapper.selectList(wrapper).stream().findFirst();
-        if(first.isPresent()){
+        if (first.isPresent()) {
             CheckTemplate checkTemplate = first.get();
             templateId = checkTemplate.getId();
         }
@@ -120,6 +128,8 @@
             scoreIndexDTO.setDate(formatter.format(createTime));
         }
 
+        //鏉冮檺鎺у埗
+        roleControl(scoreIndexDTO);
         List<Map> map = scoreMapper.selectScoreIndex(scoreIndexDTO);
 
         scoreMap.put("tableData", checkRules);
@@ -163,7 +173,9 @@
                 checkScores = scoreMapper.selectCheckScoreMap(checkScore);
             }
         }
-        return checkScores.stream().collect(Collectors.groupingBy(CheckScore::getDeptId));
+        //鍒嗘暟淇濈暀涓�浣嶅皬鏁�
+        checkScores.stream().forEach(item ->item.setScore(item.getScore().setScale(1,RoundingMode.HALF_UP)));
+        return  checkScores.stream().collect(Collectors.groupingBy(CheckScore::getDeptId));
     }
 
     private void getCheckScore(CheckScore checkScore, Calendar calendar) {
@@ -183,7 +195,7 @@
     }
 
     /**
-     * 鏌ヨ鑰冩牳绉垎鍗$墖鍒楄〃
+     * 鏌ヨ鑰冩牳绉垎鍗$墖鎶樼嚎鍥�
      *
      * @param checkScore 鑰冩牳绉垎
      * @return 鑰冩牳绉垎
@@ -207,6 +219,8 @@
             checkScore.setDate(date);
         }
         List<CheckScore> checkScores = scoreMapper.selectCheckScoreList(checkScore);
+        //鍒嗘暟淇濈暀涓�浣嶅皬鏁�
+        checkScores.stream().forEach(item ->item.setScore(item.getScore().setScale(1,RoundingMode.HALF_UP)));
         return checkScores.stream().collect(Collectors.groupingBy(CheckScore::getDeptId));
     }
 
@@ -221,7 +235,9 @@
     public List<CheckScore> page(CheckScore checkScore) {
         //鍖哄幙鍙兘鐪嬪凡鍙戝竷
         roleControl(checkScore);
-        return scoreMapper.selectCheckScoreList(checkScore);
+        List<CheckScore> scoreList = scoreMapper.selectCheckScoreList(checkScore);
+        scoreList.stream().forEach(item ->item.setScore(item.getScore().setScale(1,RoundingMode.HALF_UP)));
+        return scoreList;
     }
 
     /**
@@ -231,6 +247,7 @@
      * @return 缁撴灉
      */
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public int publishCheckScore(CheckScoreDTO checkScoreDTO) {
         if (CollectionUtils.isEmpty(checkScoreDTO.getId())) {
             return 0;
@@ -243,8 +260,33 @@
         }
         String code = publishType.getCode();
         checkScoreDTO.setPublish(code);
-
-        return scoreMapper.publishCheckScore(checkScoreDTO);
+        int i = scoreMapper.publishCheckScore(checkScoreDTO);
+        //鎺у埗index琛ㄧ殑鍙戝竷鐘舵��
+        List<Integer> ids = checkScoreDTO.getId();
+        QueryWrapper<CheckScore> queryWrapper = new QueryWrapper<>();
+        queryWrapper.in("id", ids);
+        List<CheckScore> scoreList = scoreMapper.selectList(queryWrapper);
+        //鏈�澶氫负3娆″惊鐜�
+        for (CheckScore checkScore : scoreList) {
+            Short examineCategory = checkScore.getExamineCategory();
+            if (CheckConstants.Rule_Category_Video.equals(examineCategory)) {
+                UpdateWrapper<CheckIndexVideo> updateWrapper = new UpdateWrapper<>();
+                updateWrapper.set("publish", code);
+                updateWrapper.eq("id", checkScore.getIndexId());
+                indexVideoService.update(updateWrapper);
+            } else if (CheckConstants.Rule_Category_Car.equals(examineCategory)) {
+                UpdateWrapper<CheckIndexCar> updateWrapper = new UpdateWrapper<>();
+                updateWrapper.set("publish", code);
+                updateWrapper.eq("id", checkScore.getIndexId());
+                indexCarService.update(updateWrapper);
+            } else if (CheckConstants.Rule_Category_Face.equals(examineCategory)) {
+                UpdateWrapper<CheckIndexFace> updateWrapper = new UpdateWrapper<>();
+                updateWrapper.set("publish", code);
+                updateWrapper.eq("id", checkScore.getIndexId());
+                indexFaceService.update(updateWrapper);
+            }
+        }
+        return i;
     }
 
     /**
@@ -254,55 +296,189 @@
      * @return 鑰冩牳绉垎
      */
     @Override
-    public void exportIndex(HttpServletResponse response, CheckScore checkScoreDTO) throws IOException {
-        String date = checkScoreDTO.getDate();
+    public void exportIndex(HttpServletResponse response, CheckResultExportDTO exportDTO) throws IOException {
+//        String date = exportDTO.getDate();
         //濡傛灉璇锋眰鍙傛暟鏃堕棿涓虹┖锛屾煡杩欐潯璁板綍鐨勬椂闂�
-        if (StringUtils.isEmpty(checkScoreDTO.getDate()) && CollectionUtils.isEmpty(checkScoreDTO.getQuarter())) {
-            CheckScore checkScore = getById(checkScoreDTO.getId());
-            Date createTime = checkScore.getCreateTime();
-            SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM");
-            date = formatter.format(createTime);
-        }
+//        if (StringUtils.isEmpty(exportDTO.getDate()) && CollectionUtils.isEmpty(exportDTO.getQuarter())) {
+//            SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM");
+//            if (exportDTO.getId() != null) {
+//                CheckScore checkScore = getById(exportDTO.getId());
+//                Date createTime = checkScore.getCreateTime();
+//                date = formatter.format(createTime);
+//            } else {
+//                //闃叉鍗$墖椤甸潰瀵煎嚭涓嶄紶scoreId锛屽鑷存姤閿�
+//                date = formatter.format(new Date());
+//            }
+//        }
 
         /** 瀵间笁寮爏heet */
         //杞﹁締
         CheckIndexCar checkIndexCar = new CheckIndexCar();
-        checkIndexCar.setDate(date);
-        checkIndexCar.setQuarter(checkScoreDTO.getQuarter());
+//        checkIndexCar.setDate(date);
+        checkIndexCar.setQuarter(exportDTO.getQuarter());
+        checkIndexCar.setDeptId(exportDTO.getDeptId());
+        checkIndexCar.setDeptIds(exportDTO.getDeptIds());
+        checkIndexCar.setExamineTags(exportDTO.getExamineTags());
+        //鏉冮檺鎺у埗 鍙兘鏌ョ湅宸插彂甯�
+        roleControl(checkIndexCar);
         List<CheckIndexCar> checkIndexCars = indexCarService.selectCheckIndexCarList(checkIndexCar);
-        for (CheckIndexCar indexCar : checkIndexCars) {
-            indexCar.setExamineTagStr(CheckConstants.Examine_Tag_Province.equals(indexCar.getExamineTag()) ? "鐪佸巺鑰冩牳" : "鍖哄幙鑰冩牳");
+        //璁$畻骞冲潎鍊兼斁鍦╡xcel鏈�鍚�
+        if (exportDTO.getAverage() != null && exportDTO.getAverage() && !CollectionUtils.isEmpty(checkIndexCars)) {
+            //鍏堝尯鍒嗚�冩牳鏍囩锛屽啀鍖哄垎鍖哄幙
+            List<CheckIndexCar> countyList = checkIndexCars.stream().filter(car -> CheckConstants.Examine_Tag_County.equals(car.getExamineTag())).collect(Collectors.toList());
+            if (!CollectionUtils.isEmpty(countyList)) {
+                Map<Long, List<CheckIndexCar>> carMap = countyList.stream().collect(Collectors.groupingBy(CheckIndexCar::getDeptId));
+                List<CheckIndexCar> cars = new ArrayList<>();
+                carMap.forEach((deptId, list) -> {
+                    CheckIndexCar car = CheckIndexCar.calculateAverage(list);
+                    cars.add(car);
+                });
+                //鏍规嵁绉垎闄嶅簭
+                checkIndexCars.addAll(cars.stream().sorted(Comparator.comparing(CheckIndexCar::getScore).reversed()).collect(Collectors.toList()));
+            }
+            List<CheckIndexCar> provinceList = checkIndexCars.stream().filter(car -> CheckConstants.Examine_Tag_Province.equals(car.getExamineTag())).collect(Collectors.toList());
+            if (!CollectionUtils.isEmpty(provinceList)) {
+                Map<Long, List<CheckIndexCar>> carMap = provinceList.stream().collect(Collectors.groupingBy(CheckIndexCar::getDeptId));
+                List<CheckIndexCar> cars = new ArrayList<>();
+                carMap.forEach((deptId, list) -> {
+                    CheckIndexCar car = CheckIndexCar.calculateAverage(list);
+                    cars.add(car);
+                });
+                //鏍规嵁绉垎闄嶅簭
+                checkIndexCars.addAll(cars.stream().sorted(Comparator.comparing(CheckIndexCar::getScore).reversed()).collect(Collectors.toList()));
+            }
         }
-        ExcelExp e1 = new ExcelExp("杞﹁締鑰冩牳鎸囨爣鏁版嵁", checkIndexCars, CheckIndexCar.class);
+        List<CheckIndexCarVO> checkIndexCarVOS = new ArrayList<>();
+        for (CheckIndexCar indexCar : checkIndexCars) {
+            CheckIndexCarVO excelVo = CheckIndexCar.getExcelVo(indexCar);
+            checkIndexCarVOS.add(excelVo);
+        }
+        ExcelExp e1 = new ExcelExp("杞﹁締鑰冩牳鎸囨爣鏁版嵁", checkIndexCarVOS, CheckIndexCarVO.class);
         //浜鸿劯
         CheckIndexFace checkIndexFace = new CheckIndexFace();
-        checkIndexFace.setDate(date);
-        checkIndexFace.setQuarter(checkScoreDTO.getQuarter());
+//        checkIndexFace.setDate(date);
+        checkIndexFace.setQuarter(exportDTO.getQuarter());
+        checkIndexFace.setDeptId(exportDTO.getDeptId());
+        checkIndexFace.setDeptIds(exportDTO.getDeptIds());
+        checkIndexFace.setExamineTags(exportDTO.getExamineTags());
+        //鏉冮檺鎺у埗 鍙兘鏌ョ湅宸插彂甯�
+        roleControl(checkIndexFace);
         List<CheckIndexFace> checkIndexFaces = indexFaceService.selectCheckIndexFaceList(checkIndexFace);
-        for (CheckIndexFace indexFace : checkIndexFaces) {
-            indexFace.setExamineTagStr(CheckConstants.Examine_Tag_Province.equals(indexFace.getExamineTag()) ? "鐪佸巺鑰冩牳" : "鍖哄幙鑰冩牳");
+        //璁$畻骞冲潎鍊兼斁鍦╡xcel鏈�鍚�
+        if (exportDTO.getAverage() != null && exportDTO.getAverage() && !CollectionUtils.isEmpty(checkIndexFaces)) {
+            //鍏堝尯鍒嗚�冩牳鏍囩锛屽啀鍖哄垎鍖哄幙
+            List<CheckIndexFace> countyList = checkIndexFaces.stream().filter(face -> CheckConstants.Examine_Tag_County.equals(face.getExamineTag())).collect(Collectors.toList());
+            if (!CollectionUtils.isEmpty(countyList)) {
+                Map<Long, List<CheckIndexFace>> faceMap = countyList.stream().collect(Collectors.groupingBy(CheckIndexFace::getDeptId));
+                List<CheckIndexFace> faces = new ArrayList<>();
+                faceMap.forEach((deptId, list) -> {
+                    CheckIndexFace face = CheckIndexFace.calculateAverage(list);
+                    faces.add(face);
+                });
+                //鏍规嵁绉垎闄嶅簭
+                checkIndexFaces.addAll(faces.stream().sorted(Comparator.comparing(CheckIndexFace::getScore).reversed()).collect(Collectors.toList()));
+            }
+            List<CheckIndexFace> provinceList = checkIndexFaces.stream().filter(face -> CheckConstants.Examine_Tag_Province.equals(face.getExamineTag())).collect(Collectors.toList());
+            if (!CollectionUtils.isEmpty(provinceList)) {
+                Map<Long, List<CheckIndexFace>> faceMap = provinceList.stream().collect(Collectors.groupingBy(CheckIndexFace::getDeptId));
+                List<CheckIndexFace> faces = new ArrayList<>();
+                faceMap.forEach((deptId, list) -> {
+                    CheckIndexFace face = CheckIndexFace.calculateAverage(list);
+                    faces.add(face);
+                });
+                //鏍规嵁绉垎闄嶅簭
+                checkIndexFaces.addAll(faces.stream().sorted(Comparator.comparing(CheckIndexFace::getScore).reversed()).collect(Collectors.toList()));
+            }
         }
-        ExcelExp e2 = new ExcelExp("浜鸿劯鑰冩牳鎸囨爣鏁版嵁", checkIndexFaces, CheckIndexFace.class);
+        List<CheckIndexFaceVO> checkIndexFaceVOS = new ArrayList<>();
+        for (CheckIndexFace indexFace : checkIndexFaces) {
+            CheckIndexFaceVO excelVo = CheckIndexFace.getExcelVo(indexFace);
+            checkIndexFaceVOS.add(excelVo);
+        }
+        ExcelExp e2 = new ExcelExp("浜鸿劯鑰冩牳鎸囨爣鏁版嵁", checkIndexFaceVOS, CheckIndexFaceVO.class);
         //瑙嗛
         CheckIndexVideo checkIndexVideo = new CheckIndexVideo();
-        checkIndexVideo.setDate(date);
-        checkIndexVideo.setQuarter(checkScoreDTO.getQuarter());
+//        checkIndexVideo.setDate(date);
+        checkIndexVideo.setQuarter(exportDTO.getQuarter());
+        checkIndexVideo.setDeptId(exportDTO.getDeptId());
+        checkIndexVideo.setDeptIds(exportDTO.getDeptIds());
+        checkIndexVideo.setExamineTags(exportDTO.getExamineTags());
+        //鏉冮檺鎺у埗 鍙兘鏌ョ湅宸插彂甯�
+        roleControl(checkIndexVideo);
         List<CheckIndexVideo> checkIndexVideos = indexVideoService.selectCheckIndexVideoList(checkIndexVideo);
-        for (CheckIndexVideo indexVideo : checkIndexVideos) {
-            indexVideo.setExamineTagStr(CheckConstants.Examine_Tag_Province.equals(indexVideo.getExamineTag()) ? "鐪佸巺鑰冩牳" : "鍖哄幙鑰冩牳");
+        //璁$畻骞冲潎鍊兼斁鍦╡xcel鏈�鍚�
+        if (exportDTO.getAverage() != null && exportDTO.getAverage() && !CollectionUtils.isEmpty(checkIndexVideos)) {
+            //鍏堝尯鍒嗚�冩牳鏍囩锛屽啀鍖哄垎鍖哄幙
+            List<CheckIndexVideo> countyList = checkIndexVideos.stream().filter(video -> CheckConstants.Examine_Tag_County.equals(video.getExamineTag())).collect(Collectors.toList());
+            if (!CollectionUtils.isEmpty(countyList)) {
+                Map<Long, List<CheckIndexVideo>> videoMap = countyList.stream().collect(Collectors.groupingBy(CheckIndexVideo::getDeptId));
+                List<CheckIndexVideo> videos = new ArrayList<>();
+                videoMap.forEach((deptId, list) -> {
+                    CheckIndexVideo video = CheckIndexVideo.calculateAverage(list);
+                    videos.add(video);
+                });
+                //鏍规嵁绉垎闄嶅簭
+                checkIndexVideos.addAll(videos.stream().sorted(Comparator.comparing(CheckIndexVideo::getScore).reversed()).collect(Collectors.toList()));
+            }
+            List<CheckIndexVideo> provinceList = checkIndexVideos.stream().filter(video -> CheckConstants.Examine_Tag_Province.equals(video.getExamineTag())).collect(Collectors.toList());
+            if (!CollectionUtils.isEmpty(provinceList)) {
+                Map<Long, List<CheckIndexVideo>> videoMap = provinceList.stream().collect(Collectors.groupingBy(CheckIndexVideo::getDeptId));
+                List<CheckIndexVideo> videos = new ArrayList<>();
+                videoMap.forEach((deptId, list) -> {
+                    CheckIndexVideo video = CheckIndexVideo.calculateAverage(list);
+                    videos.add(video);
+                });
+                //鏍规嵁绉垎闄嶅簭
+                checkIndexVideos.addAll(videos.stream().sorted(Comparator.comparing(CheckIndexVideo::getScore).reversed()).collect(Collectors.toList()));
+            }
         }
-        ExcelExp e3 = new ExcelExp("瑙嗛鑰冩牳鎸囨爣鏁版嵁", checkIndexVideos, CheckIndexVideo.class);
+        List<CheckIndexVideoVO> checkIndexVideoVOS = new ArrayList<>();
+        for (CheckIndexVideo indexVideo : checkIndexVideos) {
+            CheckIndexVideoVO excelVo = CheckIndexVideo.getExcelVo(indexVideo);
+            checkIndexVideoVOS.add(excelVo);
+        }
+        ExcelExp e3 = new ExcelExp("瑙嗛鑰冩牳鎸囨爣鏁版嵁", checkIndexVideoVOS, CheckIndexVideoVO.class);
 
         List<ExcelExp> mysheet = new ArrayList<>();
         mysheet.add(e1);
         mysheet.add(e2);
         mysheet.add(e3);
+        //瑙勫垯鏄庣粏
+        List<RuleExcelVO> allRuleTemplate = templateRuleMapper.getAllRuleTemplate();
+        //鏁村悎鏁版嵁
+        List<RuleExcelVO> countyRuleList = allRuleTemplate.stream().filter(vo -> vo.getExamineTag().equals(Integer.parseInt(CheckConstants.Examine_Tag_County + ""))).collect(Collectors.toList());
+        List<RuleExcelVO> provinceRuleList = allRuleTemplate.stream().filter(vo -> vo.getExamineTag().equals(Integer.parseInt(CheckConstants.Examine_Tag_Province + ""))).collect(Collectors.toList());
+        //鐢ㄥ尯鍘胯鍒欏仛涓讳綋灞曠ず
+        for (RuleExcelVO countyRule : countyRuleList) {
+            countyRule.setCountyWeight(countyRule.getWeight());
+            //绛涢�夊悓绉嶈鍒欐潈閲�
+            for (RuleExcelVO provinceRule : provinceRuleList) {
+                if (provinceRule.getRuleId().equals(countyRule.getRuleId()) && provinceRule.getExamineCategory().equals(countyRule.getExamineCategory())) {
+                    countyRule.setProvinceWeight(provinceRule.getWeight());
+                }
+            }
+        }
+        ExcelExp e4 = new ExcelExp("瑙勫垯鏄庣粏", countyRuleList, RuleExcelVO.class);
+        mysheet.add(e4);
         ExcelUtilManySheet<List<ExcelExp>> util = new ExcelUtilManySheet<>(mysheet);
         util.exportExcelManySheet(response, mysheet);
 
     }
 
-    private void roleControl(CheckScore checkScore) {
+    private void roleControl(CheckIndex checkIndex) {
+        List<SysRole> roles = SecurityUtils.getLoginUser().getUser().getRoles();
+        SysUser user = SecurityUtils.getLoginUser().getUser();
+        if (!user.isAdmin()) {
+            for (SysRole role : roles) {
+                if (role.getPermissions().contains("check:score:role:publish")) {
+                    Map<String, Object> params = checkIndex.getParams();
+                    params.put("publish", PublishType.PUBLISHED.getCode());
+                }
+            }
+        }
+    }
+
+    private void roleControl(BaseEntity checkScore) {
         List<SysRole> roles = SecurityUtils.getLoginUser().getUser().getRoles();
         SysUser user = SecurityUtils.getLoginUser().getUser();
         if (!user.isAdmin()) {

--
Gitblit v1.8.0