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 | 446 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 446 insertions(+), 0 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
new file mode 100644
index 0000000..eb9098a
--- /dev/null
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/CheckScoreServiceImpl.java
@@ -0,0 +1,446 @@
+package com.ycl.platform.service.impl;
+
+import annotation.DataScope;
+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;
+import com.ycl.platform.mapper.CheckTemplateMapper;
+import com.ycl.platform.mapper.CheckTemplateRuleMapper;
+import com.ycl.platform.service.ICheckIndexCarService;
+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;
+
+import java.io.IOException;
+import java.text.SimpleDateFormat;
+import java.time.LocalDateTime;
+import java.util.*;
+import java.util.stream.Collectors;
+
+/**
+ * 鑰冩牳绉垎鏄庣粏Service涓氬姟灞傚鐞�
+ *
+ * @author ruoyi
+ * @date 2024-04-22
+ */
+@Service
+@Slf4j
+public class CheckScoreServiceImpl extends ServiceImpl<CheckScoreMapper, CheckScore> implements ICheckScoreService {
+ @Autowired
+ private CheckScoreMapper scoreMapper;
+ @Autowired
+ private CheckTemplateMapper templateMapper;
+ @Autowired
+ private CheckTemplateRuleMapper templateRuleMapper;
+ @Autowired
+ private CheckRuleMapper ruleMapper;
+ @Autowired
+ private ICheckIndexCarService indexCarService;
+ @Autowired
+ private ICheckIndexFaceService indexFaceService;
+ @Autowired
+ private ICheckIndexVideoService indexVideoService;
+ @Autowired
+ private ISysDeptService deptService;
+
+ /**
+ * 鏌ヨ鑰冩牳绉垎鎸囨爣
+ *
+ * @param id 鑰冩牳绉垎涓婚敭
+ * @return 鑰冩牳绉垎
+ */
+ @Override
+ @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(templateId);
+
+ List<CheckRule> checkRules = new ArrayList<>();
+ //璇诲彇妯℃澘瀵瑰簲鎵�鏈夎鍒�
+ Map<String, Object> scoreMap = new HashMap<>();
+ List<Integer> ruleIds = templateRuleList.stream().map(checkTemplateRule -> checkTemplateRule.getCheckRuleId())
+ .collect(Collectors.toList());
+ List<CheckRule> ruleIndex = ruleMapper.selectBatchIds(ruleIds);
+
+ checkRules.addAll(ruleIndex);
+ //鏍规嵁examineCategory 璇诲彇涓嶅悓index琛�
+ Short examineCategory = checkScore.getExamineCategory();
+ String[] indexTableArr = {"", "t_check_index_video", "t_check_index_car", "t_check_index_face"};
+ String tableName = indexTableArr[examineCategory];
+
+ ScoreIndexDTO scoreIndexDTO = new ScoreIndexDTO()
+ .setTableName(tableName)
+ .setDeptId(checkScore.getDeptId())
+ .setDate(checkScoreIndexDTO.getDate())
+ .setQuarter(checkScoreIndexDTO.getQuarter());
+ scoreIndexDTO.setParams(checkScoreIndexDTO.getParams());
+ scoreIndexDTO.setExamineTag(checkScoreIndexDTO.getExamineTag());
+ //鑾峰綋鏈堜唤
+ 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);
+
+ checkScoreDetailVO.setCheckRuleList(templateRuleList);
+ checkScoreDetailVO.setScoreMap(scoreMap);
+ return checkScoreDetailVO;
+ }
+
+
+ /**
+ * 鏌ヨ鑰冩牳绉垎鍗＄墖鍒楄〃
+ *
+ * @param checkScore 鑰冩牳绉垎
+ * @return 鑰冩牳绉垎
+ */
+ @Override
+ @DataScope(deptAlias = "d", userAlias = "u")
+ public Map<Long, List<CheckScore>> selectCheckScoreList(CheckScore checkScore) {
+ //鍖哄幙鍙兘鐪嬪凡鍙戝竷
+ roleControl(checkScore);
+
+ // 鑾峰彇鏁版嵁鏃ユ湡鏃堕棿
+ Calendar calendar = Calendar.getInstance();
+ getCheckScore(checkScore, calendar);
+
+ // 涓�鍙锋煡璇箣鍓嶇殑鏁版嵁
+ if (LocalDateTime.now().getDayOfMonth() == 1) {
+ calendar.add(Calendar.DAY_OF_MONTH, -1);
+ }
+
+ List<CheckScore> checkScores = scoreMapper.selectCheckScoreMap(checkScore);
+
+ // 濡傛灉鏁版嵁涓虹┖锛屽垯鏌ヨ涔嬪墠鐨勬暟鎹�
+ if (checkScores.isEmpty()) {
+ CheckScore one = scoreMapper.getLast(checkScore);
+ if (Objects.nonNull(one)) {
+ calendar.setTime(one.getCreateTime());
+ getCheckScore(checkScore, calendar);
+ checkScores = scoreMapper.selectCheckScoreMap(checkScore);
+ }
+ }
+ return checkScores.stream().collect(Collectors.groupingBy(CheckScore::getDeptId));
+ }
+
+ private void getCheckScore(CheckScore checkScore, Calendar calendar) {
+ // 0 鐪佸巺鏈堝害 1 甯傚眬鏈堝害 2 鐪佸巺瀛ｅ害 3 甯傚眬瀛ｅ害
+ switch (checkScore.getExamineTag()) {
+ case 0, 1:
+ checkScore.setEndDate(calendar.getTime());
+ calendar.set(Calendar.DAY_OF_MONTH, 1);
+ checkScore.setStartDate(calendar.getTime());
+ break;
+ case 2, 3:
+ checkScore.setStartDate(DateUtils.getQuarterStart(calendar).getTime());
+ checkScore.setEndDate(DateUtils.getQuarterEnd(calendar).getTime());
+ checkScore.setExamineTag(checkScore.getExamineTag() == 2 ? 0 : 1);
+ break;
+ }
+ }
+
+ /**
+ * 鏌ヨ鑰冩牳绉垎鍗＄墖鍒楄〃
+ *
+ * @param checkScore 鑰冩牳绉垎
+ * @return 鑰冩牳绉垎
+ */
+ @Override
+ @DataScope(deptAlias = "d", userAlias = "u")
+ public Map<Long, List<CheckScore>> selectCheckScoreChart(CheckScore checkScore) {
+ //鍖哄幙鍙兘鐪嬪凡鍙戝竷
+ roleControl(checkScore);
+
+ //鏌ヨ鎴愮哗
+ String date = checkScore.getDate();
+ if (StringUtils.isEmpty(date)) {
+ //濡傛灉涓虹┖鏌ユ湰鏈堢殑鏁版嵁
+ Calendar now = Calendar.getInstance();
+ //鍙栨槰澶�
+ // now.add(Calendar.DATE, -1);
+ Date yesterday = now.getTime();
+ SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM");
+ date = simpleDateFormat.format(yesterday);
+ checkScore.setDate(date);
+ }
+ List<CheckScore> checkScores = scoreMapper.selectCheckScoreList(checkScore);
+ return checkScores.stream().collect(Collectors.groupingBy(CheckScore::getDeptId));
+ }
+
+ /**
+ * 鍖哄幙璇︽儏
+ *
+ * @param checkScore 鑰冩牳绉垎
+ * @return 鑰冩牳绉垎
+ */
+ @Override
+ @DataScope(deptAlias = "d", userAlias = "u")
+ public List<CheckScore> page(CheckScore checkScore) {
+ //鍖哄幙鍙兘鐪嬪凡鍙戝竷
+ roleControl(checkScore);
+ return scoreMapper.selectCheckScoreList(checkScore);
+ }
+
+ /**
+ * 鍙戝竷鑰冩牳绉垎淇℃伅
+ *
+ * @param checkScoreDTO 鑰冩牳绉垎
+ * @return 缁撴灉
+ */
+ @Override
+ @Transactional(rollbackFor = Exception.class)
+ public int publishCheckScore(CheckScoreDTO checkScoreDTO) {
+ if (CollectionUtils.isEmpty(checkScoreDTO.getId())) {
+ return 0;
+ }
+ PublishType publishType = null;
+ try {
+ publishType = PublishType.valueOf(checkScoreDTO.getPublish());
+ } catch (IllegalArgumentException e) {
+ throw new IllegalArgumentException("鍙傛暟绫诲瀷涓嶅尮閰�");
+ }
+ String code = publishType.getCode();
+ checkScoreDTO.setPublish(code);
+ 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;
+ }
+
+ /**
+ * 瀵煎嚭鎸囨爣
+ *
+ * @param checkScore 鑰冩牳绉垎
+ * @return 鑰冩牳绉垎
+ */
+ @Override
+ public void exportIndex(HttpServletResponse response, CheckScore checkScoreDTO) throws IOException {
+ String date = checkScoreDTO.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);
+ }
+
+ /** 瀵间笁寮爏heet */
+ //杞﹁締
+ CheckIndexCar checkIndexCar = new CheckIndexCar();
+ checkIndexCar.setDate(date);
+ checkIndexCar.setQuarter(checkScoreDTO.getQuarter());
+ //鏉冮檺鎺у埗 鍙兘鏌ョ湅宸插彂甯�
+ roleControl(checkIndexCar);
+ List<CheckIndexCar> checkIndexCars = indexCarService.selectCheckIndexCarList(checkIndexCar);
+ 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);
+ 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);
+ 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);
+
+ }
+
+ 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()) {
+ 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);
+ }
+
+ // /**
+// * 鏂板鑰冩牳绉垎
+// *
+// * @param checkScore 鑰冩牳绉垎
+// * @return 缁撴灉
+// */
+// @Override
+// public int insertCheckScore(CheckScore checkScore)
+// {
+// checkScore.setCreateTime(DateUtils.getNowDate());
+// return scoreMapper.insertCheckScore(checkScore);
+// }
+//
+// /**
+// * 淇敼鑰冩牳鏄庣粏
+// *
+// * @param checkScore 鑰冩牳绉垎鏄庣粏
+// * @return 缁撴灉
+// */
+// @Override
+// public int updateCheckScore(CheckScore checkScore)
+// {
+//
+// return scoreMapper.updateCheckScore(checkScore);
+// }
+//
+// /**
+// * 鎵归噺鍒犻櫎鑰冩牳绉垎
+// *
+// * @param ids 闇�瑕佸垹闄ょ殑鑰冩牳绉垎涓婚敭
+// * @return 缁撴灉
+// */
+// @Override
+// public int deleteCheckScoreByIds(Long[] ids)
+// {
+// return scoreMapper.deleteCheckScoreByIds(ids);
+// }
+//
+// /**
+// * 鍒犻櫎鑰冩牳绉垎淇℃伅
+// *
+// * @param id 鑰冩牳绉垎涓婚敭
+// * @return 缁撴灉
+// */
+// @Override
+// public int deleteCheckScoreById(Long id)
+// {
+// return scoreMapper.deleteCheckScoreById(id);
+// }
+}
--
Gitblit v1.8.0