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 | 118 ++++++++++++++++++++++++++++++++++++++++++++++++----------- 1 files changed, 96 insertions(+), 22 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 19c36b7..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,17 +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; @@ -77,8 +85,18 @@ 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<>(); //璇诲彇妯℃澘瀵瑰簲鎵�鏈夎鍒� @@ -99,7 +117,7 @@ .setDate(checkScoreIndexDTO.getDate()) .setQuarter(checkScoreIndexDTO.getQuarter()); scoreIndexDTO.setParams(checkScoreIndexDTO.getParams()); - scoreIndexDTO.setExamineTag(checkScore.getExamineTag()); + scoreIndexDTO.setExamineTag(checkScoreIndexDTO.getExamineTag()); //鑾峰綋鏈堜唤 if (StringUtils.isEmpty(scoreIndexDTO.getDate()) && CollectionUtils.isEmpty(checkScoreIndexDTO.getQuarter())) { //濡傛灉鏌ヨ鏉′欢涓嶅惈鍙傛暟锛屾煡璇㈢Н鍒嗗搴斿垱寤烘椂闂� @@ -108,6 +126,8 @@ scoreIndexDTO.setDate(formatter.format(createTime)); } + //鏉冮檺鎺у埗 + roleControl(scoreIndexDTO); List<Map> map = scoreMapper.selectScoreIndex(scoreIndexDTO); scoreMap.put("tableData", checkRules); @@ -144,7 +164,7 @@ // 濡傛灉鏁版嵁涓虹┖锛屽垯鏌ヨ涔嬪墠鐨勬暟鎹� 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); @@ -219,6 +239,7 @@ * @return 缁撴灉 */ @Override + @Transactional(rollbackFor = Exception.class) public int publishCheckScore(CheckScoreDTO checkScoreDTO) { if (CollectionUtils.isEmpty(checkScoreDTO.getId())) { return 0; @@ -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; } /** @@ -244,14 +296,12 @@ @Override public void exportIndex(HttpServletResponse response, CheckScore checkScoreDTO) throws IOException { String date = checkScoreDTO.getDate(); - Integer examineTag = null; //濡傛灉璇锋眰鍙傛暟鏃堕棿涓虹┖锛屾煡杩欐潯璁板綍鐨勬椂闂� 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); - examineTag = checkScore.getExamineTag(); } /** 瀵间笁寮爏heet */ @@ -259,29 +309,41 @@ CheckIndexCar checkIndexCar = new CheckIndexCar(); checkIndexCar.setDate(date); checkIndexCar.setQuarter(checkScoreDTO.getQuarter()); - if (examineTag != null) { - checkIndexCar.setExamineTag(Short.parseShort(examineTag + "")); - } + //鏉冮檺鎺у埗 鍙兘鏌ョ湅宸插彂甯� + 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()); - if (examineTag != null) { - checkIndexFace.setExamineTag(Short.parseShort(examineTag + "")); - } + //鏉冮檺鎺у埗 鍙兘鏌ョ湅宸插彂甯� + 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()); - if (examineTag != null) { - checkIndexVideo.setExamineTag(Short.parseShort(examineTag + "")); - } + //鏉冮檺鎺у埗 鍙兘鏌ョ湅宸插彂甯� + 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); @@ -292,12 +354,24 @@ } - 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:publish")) { + 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()); } -- Gitblit v1.8.0