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