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/CheckScoreServiceImpl.java |  181 +++++++++++++++++++++++++++++++++++---------
 1 files changed, 142 insertions(+), 39 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 21d675d..eb9098a 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
@@ -1,12 +1,17 @@
 package com.ycl.platform.service.impl;
 
 import annotation.DataScope;
-import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+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.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.CheckIndexCarVO;
+import com.ycl.platform.domain.vo.CheckIndexFaceVO;
+import com.ycl.platform.domain.vo.CheckIndexVideoVO;
 import com.ycl.platform.domain.vo.CheckScoreDetailVO;
 import com.ycl.platform.mapper.CheckRuleMapper;
 import com.ycl.platform.mapper.CheckScoreMapper;
@@ -16,16 +21,20 @@
 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;
 import com.ycl.utils.DateUtils;
 import com.ycl.utils.SecurityUtils;
 import com.ycl.utils.StringUtils;
+import constant.CheckConstants;
 import enumeration.general.PublishType;
 import jakarta.servlet.http.HttpServletResponse;
 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;
@@ -44,8 +53,7 @@
  */
 @Service
 @Slf4j
-public class CheckScoreServiceImpl extends ServiceImpl<CheckScoreMapper, CheckScore> implements ICheckScoreService
-{
+public class CheckScoreServiceImpl extends ServiceImpl<CheckScoreMapper, CheckScore> implements ICheckScoreService {
     @Autowired
     private CheckScoreMapper scoreMapper;
     @Autowired
@@ -62,6 +70,7 @@
     private ICheckIndexVideoService indexVideoService;
     @Autowired
     private ISysDeptService deptService;
+
     /**
      * 鏌ヨ鑰冩牳绉垎鎸囨爣
      *
@@ -69,20 +78,29 @@
      * @return 鑰冩牳绉垎
      */
     @Override
-    @DataScope(deptAlias = "d",userAlias = "u")
-    public CheckScoreDetailVO selectCheckScoreById(CheckScoreIndexDTO checkScoreIndexDTO)
-    {
+    @DataScope(deptAlias = "d", userAlias = "u")
+    public CheckScoreDetailVO selectCheckScoreById(CheckScoreIndexDTO checkScoreIndexDTO) {
         Long checkScoreId = checkScoreIndexDTO.getId();
 
         CheckScoreDetailVO checkScoreDetailVO = new CheckScoreDetailVO();
         //鏍规嵁id璇诲彇score
         CheckScore checkScore = scoreMapper.selectCheckScoreById(checkScoreId);
+        Integer templateId = 0;
+        //鏂逛究鍒囨崲鏀逛负锛氭牴鎹畇core鑰冩牳绫诲埆鍜宒to鑰冩牳鏍囩鏌ユā鏉胯鍒欏拰鏉冮噸,娉ㄦ剰鍚岀绫诲悓鏍囩鍙兘鏈変竴涓ā鏉�
+        QueryWrapper<CheckTemplate> wrapper = new QueryWrapper<>();
+        wrapper.eq("examine_tag",checkScoreIndexDTO.getExamineTag());
+        wrapper.eq("examine_category",checkScore.getExamineCategory());
+        Optional<CheckTemplate> first = templateMapper.selectList(wrapper).stream().findFirst();
+        if(first.isPresent()){
+            CheckTemplate checkTemplate = first.get();
+            templateId = checkTemplate.getId();
+        }
         //璇诲彇瑙勫垯浠ュ強鏉冮噸
-        List<CheckTemplateRule> templateRuleList = templateRuleMapper.selectListByTemplateId(checkScore.getTemplateId());
+        List<CheckTemplateRule> templateRuleList = templateRuleMapper.selectListByTemplateId(templateId);
 
         List<CheckRule> checkRules = new ArrayList<>();
         //璇诲彇妯℃澘瀵瑰簲鎵�鏈夎鍒�
-        Map<String,Object> scoreMap = new HashMap<>();
+        Map<String, Object> scoreMap = new HashMap<>();
         List<Integer> ruleIds = templateRuleList.stream().map(checkTemplateRule -> checkTemplateRule.getCheckRuleId())
                 .collect(Collectors.toList());
         List<CheckRule> ruleIndex = ruleMapper.selectBatchIds(ruleIds);
@@ -90,7 +108,7 @@
         checkRules.addAll(ruleIndex);
         //鏍规嵁examineCategory 璇诲彇涓嶅悓index琛�
         Short examineCategory = checkScore.getExamineCategory();
-        String[] indexTableArr ={"","t_check_index_video","t_check_index_car","t_check_index_face"};
+        String[] indexTableArr = {"", "t_check_index_video", "t_check_index_car", "t_check_index_face"};
         String tableName = indexTableArr[examineCategory];
 
         ScoreIndexDTO scoreIndexDTO = new ScoreIndexDTO()
@@ -99,19 +117,21 @@
                 .setDate(checkScoreIndexDTO.getDate())
                 .setQuarter(checkScoreIndexDTO.getQuarter());
         scoreIndexDTO.setParams(checkScoreIndexDTO.getParams());
-
+        scoreIndexDTO.setExamineTag(checkScoreIndexDTO.getExamineTag());
         //鑾峰綋鏈堜唤
-        if(StringUtils.isEmpty(scoreIndexDTO.getDate()) && CollectionUtils.isEmpty(checkScoreIndexDTO.getQuarter())){
+        if (StringUtils.isEmpty(scoreIndexDTO.getDate()) && CollectionUtils.isEmpty(checkScoreIndexDTO.getQuarter())) {
             //濡傛灉鏌ヨ鏉′欢涓嶅惈鍙傛暟锛屾煡璇㈢Н鍒嗗搴斿垱寤烘椂闂�
             Date createTime = checkScore.getCreateTime();
             SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM");
             scoreIndexDTO.setDate(formatter.format(createTime));
         }
 
+        //鏉冮檺鎺у埗
+        roleControl(scoreIndexDTO);
         List<Map> map = scoreMapper.selectScoreIndex(scoreIndexDTO);
 
-        scoreMap.put("tableData",checkRules);
-        scoreMap.put("dataMap",map);
+        scoreMap.put("tableData", checkRules);
+        scoreMap.put("dataMap", map);
 
         checkScoreDetailVO.setCheckRuleList(templateRuleList);
         checkScoreDetailVO.setScoreMap(scoreMap);
@@ -126,9 +146,8 @@
      * @return 鑰冩牳绉垎
      */
     @Override
-    @DataScope(deptAlias = "d",userAlias = "u")
-    public Map<Long, List<CheckScore>> selectCheckScoreList(CheckScore checkScore)
-    {
+    @DataScope(deptAlias = "d", userAlias = "u")
+    public Map<Long, List<CheckScore>> selectCheckScoreList(CheckScore checkScore) {
         //鍖哄幙鍙兘鐪嬪凡鍙戝竷
         roleControl(checkScore);
 
@@ -137,12 +156,15 @@
         getCheckScore(checkScore, calendar);
 
         // 涓�鍙锋煡璇箣鍓嶇殑鏁版嵁
-        if (LocalDateTime.now().getDayOfMonth() == 1) { calendar.add(Calendar.DAY_OF_MONTH, -1); }
+        if (LocalDateTime.now().getDayOfMonth() == 1) {
+            calendar.add(Calendar.DAY_OF_MONTH, -1);
+        }
 
         List<CheckScore> checkScores = scoreMapper.selectCheckScoreMap(checkScore);
+
         // 濡傛灉鏁版嵁涓虹┖锛屽垯鏌ヨ涔嬪墠鐨勬暟鎹�
         if (checkScores.isEmpty()) {
-            CheckScore one = new LambdaQueryChainWrapper<>(baseMapper).select(CheckScore::getCreateTime).last("order by create_time desc limit 1").one();
+            CheckScore one = scoreMapper.getLast(checkScore);
             if (Objects.nonNull(one)) {
                 calendar.setTime(one.getCreateTime());
                 getCheckScore(checkScore, calendar);
@@ -152,8 +174,7 @@
         return checkScores.stream().collect(Collectors.groupingBy(CheckScore::getDeptId));
     }
 
-    private void getCheckScore(CheckScore checkScore, Calendar calendar)
-    {
+    private void getCheckScore(CheckScore checkScore, Calendar calendar) {
         // 0 鐪佸巺鏈堝害 1 甯傚眬鏈堝害 2 鐪佸巺瀛e害 3 甯傚眬瀛e害
         switch (checkScore.getExamineTag()) {
             case 0, 1:
@@ -176,15 +197,14 @@
      * @return 鑰冩牳绉垎
      */
     @Override
-    @DataScope(deptAlias = "d",userAlias = "u")
-    public Map<Long, List<CheckScore>> selectCheckScoreChart(CheckScore checkScore)
-    {
+    @DataScope(deptAlias = "d", userAlias = "u")
+    public Map<Long, List<CheckScore>> selectCheckScoreChart(CheckScore checkScore) {
         //鍖哄幙鍙兘鐪嬪凡鍙戝竷
         roleControl(checkScore);
 
         //鏌ヨ鎴愮哗
         String date = checkScore.getDate();
-        if(StringUtils.isEmpty(date)){
+        if (StringUtils.isEmpty(date)) {
             //濡傛灉涓虹┖鏌ユ湰鏈堢殑鏁版嵁
             Calendar now = Calendar.getInstance();
             //鍙栨槰澶�
@@ -205,7 +225,7 @@
      * @return 鑰冩牳绉垎
      */
     @Override
-    @DataScope(deptAlias = "d",userAlias = "u")
+    @DataScope(deptAlias = "d", userAlias = "u")
     public List<CheckScore> page(CheckScore checkScore) {
         //鍖哄幙鍙兘鐪嬪凡鍙戝竷
         roleControl(checkScore);
@@ -219,8 +239,9 @@
      * @return 缁撴灉
      */
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public int publishCheckScore(CheckScoreDTO checkScoreDTO) {
-        if(CollectionUtils.isEmpty(checkScoreDTO.getId())){
+        if (CollectionUtils.isEmpty(checkScoreDTO.getId())) {
             return 0;
         }
         PublishType publishType = null;
@@ -231,8 +252,39 @@
         }
         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("dept_id",checkScore.getDeptId());
+                updateWrapper.eq("examine_tag",checkScore.getExamineTag());
+                updateWrapper.between("create_time",DateUtils.getDayStart(checkScore.getCreateTime()),DateUtils.getDayEnd(checkScore.getCreateTime()));
+                indexVideoService.update(updateWrapper);
+            }else if(CheckConstants.Rule_Category_Car.equals(examineCategory)){
+                UpdateWrapper<CheckIndexCar> updateWrapper = new UpdateWrapper<>();
+                updateWrapper.set("publish",code);
+                updateWrapper.eq("dept_id",checkScore.getDeptId());
+                updateWrapper.eq("examine_tag",checkScore.getExamineTag());
+                updateWrapper.between("create_time",DateUtils.getDayStart(checkScore.getCreateTime()),DateUtils.getDayEnd(checkScore.getCreateTime()));
+                indexCarService.update(updateWrapper);
+            }else if(CheckConstants.Rule_Category_Face.equals(examineCategory)){
+                UpdateWrapper<CheckIndexFace> updateWrapper = new UpdateWrapper<>();
+                updateWrapper.set("publish",code);
+                updateWrapper.eq("dept_id",checkScore.getDeptId());
+                updateWrapper.eq("examine_tag",checkScore.getExamineTag());
+                updateWrapper.between("create_time",DateUtils.getDayStart(checkScore.getCreateTime()),DateUtils.getDayEnd(checkScore.getCreateTime()));
+                indexFaceService.update(updateWrapper);
+            }
+        }
+        return i;
     }
 
     /**
@@ -242,10 +294,10 @@
      * @return 鑰冩牳绉垎
      */
     @Override
-    public void exportIndex(HttpServletResponse response,CheckScore checkScoreDTO) throws IOException {
+    public void exportIndex(HttpServletResponse response, CheckScore checkScoreDTO) throws IOException {
         String date = checkScoreDTO.getDate();
         //濡傛灉璇锋眰鍙傛暟鏃堕棿涓虹┖锛屾煡杩欐潯璁板綍鐨勬椂闂�
-        if(StringUtils.isEmpty(checkScoreDTO.getDate()) && CollectionUtils.isEmpty(checkScoreDTO.getQuarter())){
+        if (StringUtils.isEmpty(checkScoreDTO.getDate()) && CollectionUtils.isEmpty(checkScoreDTO.getQuarter())) {
             CheckScore checkScore = getById(checkScoreDTO.getId());
             Date createTime = checkScore.getCreateTime();
             SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM");
@@ -257,39 +309,90 @@
         CheckIndexCar checkIndexCar = new CheckIndexCar();
         checkIndexCar.setDate(date);
         checkIndexCar.setQuarter(checkScoreDTO.getQuarter());
+        //鏉冮檺鎺у埗 鍙兘鏌ョ湅宸插彂甯�
+        roleControl(checkIndexCar);
         List<CheckIndexCar> checkIndexCars = indexCarService.selectCheckIndexCarList(checkIndexCar);
-        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());
+        //鏉冮檺鎺у埗 鍙兘鏌ョ湅宸插彂甯�
+        roleControl(checkIndexFace);
         List<CheckIndexFace> checkIndexFaces = indexFaceService.selectCheckIndexFaceList(checkIndexFace);
-        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());
+        //鏉冮檺鎺у埗 鍙兘鏌ョ湅宸插彂甯�
+        roleControl(checkIndexVideo);
         List<CheckIndexVideo> checkIndexVideos = indexVideoService.selectCheckIndexVideoList(checkIndexVideo);
-        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);
         ExcelUtilManySheet<List<ExcelExp>> util = new ExcelUtilManySheet<>(mysheet);
-        util.exportExcelManySheet(response,mysheet);
+        util.exportExcelManySheet(response, mysheet);
 
     }
 
-    private void roleControl(CheckScore checkScore) {
+    private void roleControl(CheckIndex checkIndex) {
         List<SysRole> roles = SecurityUtils.getLoginUser().getUser().getRoles();
-        SysRole sysRole = roles.get(0);
-        if("zg_province".equals(sysRole.getRoleKey())){
-            Map<String, Object> params = checkScore.getParams();
-            params.put("publish",PublishType.PUBLISHED.getCode());
+        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()) {
+            for (SysRole role : roles) {
+                if (role.getPermissions().contains("check:score:role:publish")) {
+                    Map<String, Object> params = checkScore.getParams();
+                    params.put("publish", PublishType.PUBLISHED.getCode());
+                }
+            }
         }
     }
 
+    @Override
+    public List<Map<String, Object>> home() {
+        return baseMapper.home();
+    }
+
+
+    @Override
+    public List<Map<String, Object>> calculate(String category) {
+        Calendar calendar = Calendar.getInstance();
+        Date endDate = calendar.getTime();
+        calendar.set(Calendar.DAY_OF_MONTH, 1);
+        Date startDate = calendar.getTime();
+        return scoreMapper.calculate(startDate, endDate, category);
+    }
 
     //    /**
 //     * 鏂板鑰冩牳绉垎

--
Gitblit v1.8.0