From 4403a428edc18482c5aa561e603b26ebe13d328f Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期二, 02 四月 2024 18:27:02 +0800
Subject: [PATCH] 部门端登录、基础数据导入优化、得分计算优化

---
 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TbBasicDataServiceImpl.java |  198 +++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 165 insertions(+), 33 deletions(-)

diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TbBasicDataServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TbBasicDataServiceImpl.java
index d755803..77e4182 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TbBasicDataServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TbBasicDataServiceImpl.java
@@ -1,23 +1,25 @@
 package com.ruoyi.system.service.impl;
 
 import com.alibaba.excel.EasyExcel;
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.google.common.collect.Lists;
 import com.ruoyi.common.basic.PageDTO;
 import com.ruoyi.common.core.domain.R;
-import com.ruoyi.common.enums.ReportingStatusEnum;
-import com.ruoyi.common.enums.ShowStatusEnum;
+import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.common.core.domain.model.LoginUser;
+import com.ruoyi.common.enums.*;
 import com.ruoyi.common.exception.ServiceException;
-import com.ruoyi.common.utils.BeanUtils;
-import com.ruoyi.common.utils.CollUtils;
-import com.ruoyi.common.utils.DateUtils;
-import com.ruoyi.common.utils.NumberDisplaceChineseUtil;
+import com.ruoyi.common.utils.*;
 import com.ruoyi.system.domain.*;
 import com.ruoyi.system.dto.BasicDataDTO;
+import com.ruoyi.system.handler.CustomerHandler;
+import com.ruoyi.system.handler.SelectedSheetWriteHandler;
 import com.ruoyi.system.listener.BasicDataListener;
 import com.ruoyi.system.mapper.TbBasicDataMapper;
+import com.ruoyi.system.mapper.TbDeptMapper;
 import com.ruoyi.system.query.ScoreQuery;
 import com.ruoyi.system.service.*;
 import com.ruoyi.system.vo.BasicDataReportingVO;
@@ -51,18 +53,19 @@
 @RequiredArgsConstructor
 public class TbBasicDataServiceImpl extends ServiceImpl<TbBasicDataMapper, TbBasicData> implements TbBasicDataService {
 
-    private final TbDeptService tbDeptService;
+    private final TbDeptMapper tbDeptMapper;
     private final HttpServletResponse response;
     private final TbFieldService tbFieldService;
     private final TbBasicDataFieldService tbBasicDataFieldService;
     private final TbBasicDataConfigService tbBasicDataConfigService;
     private final TbBasicDataConfigDetailService tbBasicDataConfigDetailService;
-    private final TbScoreService scoreService;
+    private final TbScoreService tbScoreService;
+
     @Override
     public R<BasicDataReportingVO> getBasicFields(String deptAreaCode) throws Exception {
         BasicDataReportingVO vo = new BasicDataReportingVO();
         //校验区划代码
-        TbDept dept = tbDeptService.getOne(Wrappers.<TbDept>lambdaQuery().eq(TbDept::getAreaCode, deptAreaCode));
+        TbDept dept = tbDeptMapper.selectOne(Wrappers.<TbDept>lambdaQuery().eq(TbDept::getAreaCode, deptAreaCode));
         if (Objects.isNull(dept)) {
             throw new ServiceException(String.format("区划代码%s不存在", deptAreaCode));
         }
@@ -106,7 +109,7 @@
             List<TbBasicDataField> basicDataFields = tbBasicDataFieldService.lambdaQuery()
                     .eq(TbBasicDataField::getBasicDataId, basicData.getId()).list();
             if (CollUtils.isNotEmpty(basicDataFields)) {
-                Map<Integer, TbBasicDataField> fieldMap = basicDataFields.stream()
+                Map<Long, TbBasicDataField> fieldMap = basicDataFields.stream()
                         .collect(Collectors.toMap(TbBasicDataField::getFieldId, Function.identity()));
                 vo.getFields().forEach(item -> {
                     TbBasicDataField tbBasicDataField = fieldMap.get(item.getId());
@@ -123,21 +126,47 @@
     @Override
     @Transactional(rollbackFor = Exception.class)
     public void saveBasicData(BasicDataDTO dto) {
-        //TODO
-        //LoginUser loginUser = SecurityUtils.getLoginUser();
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        String areaCode = loginUser.getUser().getAreaCode();
         //数据校验
         if (Objects.isNull(dto) || CollUtils.isEmpty(dto.getFields())) {
             return;
         }
-
         TbBasicData tbBasicData = BeanUtils.copyBean(dto, TbBasicData.class);
+        tbBasicData.setDeptAreaCode(areaCode);
+        tbBasicData.setStatus(ReportingStatusEnum.MISSING_DATA);
         this.saveOrUpdate(tbBasicData);
         //保存基础数据动态字段数据
         List<TbBasicDataField> tbBasicDataFields = BeanUtils.copyList(dto.getFields(), TbBasicDataField.class);
         tbBasicDataFields.forEach(item -> item.setBasicDataId(tbBasicData.getId()));
+        //查询需要填写的动态字段
+        List<TbField> fieldList = tbFieldService.lambdaQuery().eq(TbField::getStatus, ShowStatusEnum.SHOW).list();
+        //添加固定字段 转移支付规模、当期GDP
+        TbBasicDataField transferPaymentScale = new TbBasicDataField();
+        transferPaymentScale.setBasicDataId(tbBasicData.getId());
+        transferPaymentScale.setFieldId(-1L);
+        transferPaymentScale.setFieldValue(tbBasicData.getTransferPaymentScale());
+        tbBasicDataFields.add(transferPaymentScale);
+        TbBasicDataField currentGdp = new TbBasicDataField();
+        transferPaymentScale.setBasicDataId(tbBasicData.getId());
+        transferPaymentScale.setFieldId(-2L);
+        transferPaymentScale.setFieldValue(tbBasicData.getCurrentGdp());
+        tbBasicDataFields.add(currentGdp);
         //将该基础数据的动态字段数据全部删除
-        tbBasicDataFieldService.remove(null);
+        tbBasicDataFieldService.remove(Wrappers.<TbBasicDataField>lambdaQuery().eq(TbBasicDataField::getBasicDataId, tbBasicData.getId()));
         tbBasicDataFieldService.saveBatch(tbBasicDataFields);
+        //需要填写的动态字段
+        Set<Long> fieldIds = fieldList.stream().map(TbField::getId).collect(Collectors.toSet());
+        //用户填写的动态字段
+        Set<Long> deptFieldIds = tbBasicDataFields.stream().map(TbBasicDataField::getFieldId).collect(Collectors.toSet());
+        boolean flag = deptFieldIds.containsAll(fieldIds) && (StringUtils.isNotEmpty(dto.getTransferPaymentScale())&& StringUtils.isNotEmpty(dto.getCurrentGdp()));
+        if (flag) {
+            tbBasicData.setStatus(ReportingStatusEnum.FILLED);
+            this.updateById(tbBasicData);
+        }
+        if (ReportingStatusEnum.FILLED.equals(tbBasicData.getStatus())) {
+            calculateScore(tbBasicData);
+        }
     }
 
     @Override
@@ -149,16 +178,47 @@
         response.setHeader("Content-disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8") + ".xlsx");
         //查询需要填写的动态字段
         List<TbField> list = tbFieldService.lambdaQuery().eq(TbField::getStatus, ShowStatusEnum.SHOW).list();
+        List<List<String>> head = head(list);
+        List<TbField> collect = list.stream()
+                .filter(item -> FieldTypeEnum.TEXT.equals(item.getFieldType()))
+                .filter(item -> FieldInputTypeEnum.FIXED_CONTENT.equals(item.getTextInputType()))
+                .collect(Collectors.toList());
+        Map<Integer, String[]> selectedMap = new HashMap<>();
+        if (CollUtils.isNotEmpty(collect)) {
+            selectedMap = new HashMap<>();
+            for (TbField tbField : collect) {
+                int outerListIndex = getOuterListIndex(head, tbField.getFieldName());
+                String[] selectStr = tbField.getTextContent().split(",");
+                selectedMap.put(outerListIndex, selectStr);
+            }
+        }
         // 这里需要设置不关闭流
-        EasyExcel.write(response.getOutputStream()).head(head(list))
-                .autoCloseStream(Boolean.FALSE).sheet("模板")
+        EasyExcel.write(response.getOutputStream()).head(head)
+                .autoCloseStream(Boolean.TRUE).sheet("模板")
+                .registerWriteHandler(new SelectedSheetWriteHandler(selectedMap))
+                .registerWriteHandler(new CustomerHandler())
+                .registerWriteHandler(EasyExcelUtil.getStyleStrategy())
                 .doWrite(dataList(list));
     }
 
+    /**
+     * 根据字段名获取该字段下标
+     * @param list 表头
+     * @param value 字段名
+     * @return 下标
+     */
+    public int getOuterListIndex(List<List<String>> list, String value) {
+        for (int i = 0; i < list.size(); i++) {
+            if (list.get(i).contains(value)) {
+                return i;
+            }
+        }
+        return -1; // 返回-1表示未找到
+    }
     private List<List<Object>> dataList(List<TbField> list) throws Exception {
-        //TODO
-        //LoginUser loginUser = SecurityUtils.getLoginUser();
-        TbDept dept = tbDeptService.getById(44);
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        String areaName = loginUser.getUser().getAreaName();
+
         //当前所在季度
         int quarterOfYear = DateUtils.getQuarterOfYear(new Date());
         String quarterOfYearStr = NumberDisplaceChineseUtil.numberToChinese(quarterOfYear);
@@ -169,7 +229,7 @@
             columnNo.add(String.valueOf(i));
         }
         excellist.add(columnNo);
-        excellist.add(Lists.newArrayList(dept.getAreaName(), String.format("%s季度",quarterOfYearStr)));
+        excellist.add(Lists.newArrayList(areaName, String.format("%s季度",quarterOfYearStr)));
         return excellist;
     }
 
@@ -181,31 +241,103 @@
         headTitles.add(Lists.newArrayList("转移支付规模"));
         headTitles.add(Lists.newArrayList("当期GDP"));
         list.forEach(item ->{
-            headTitles.add(Lists.newArrayList(item.getLevelOneCategory(), item.getLevelTwoCategory(), item.getLevelThreeCategory(), item.getFieldName()));
+            String levelOneCategory = item.getLevelOneCategory();
+            String levelTwoCategory = item.getLevelTwoCategory();
+            String levelThreeCategory = item.getLevelThreeCategory();
+            String fieldName = item.getFieldName();
+            headTitles.add(Lists.newArrayList(levelOneCategory,StringUtils.isBlank(levelTwoCategory) ? fieldName : levelTwoCategory, StringUtils.isBlank(levelThreeCategory) ? fieldName : levelThreeCategory, fieldName));
         });
         headTitles.add(Lists.newArrayList("备注"));
         return headTitles;
     }
-
     @Override
     @Transactional(rollbackFor = Exception.class)
     public void importBasicData(MultipartFile file) throws Exception {
-        //TODO
-        //LoginUser loginUser = SecurityUtils.getLoginUser();
-        TbDept dept = tbDeptService.getById(44);
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        String areaCode = loginUser.getUser().getAreaCode();
         //查询需要填写的动态字段
         List<TbField> fieldList = tbFieldService.lambdaQuery().eq(TbField::getStatus, ShowStatusEnum.SHOW).list();
         EasyExcel.read(file.getInputStream(), new BasicDataListener(this,fieldList,
-                tbFieldService,dept,tbBasicDataFieldService,tbBasicDataConfigService,
-                tbBasicDataConfigDetailService,scoreService)).sheet().doRead();
+                tbFieldService,areaCode,tbBasicDataFieldService,tbBasicDataConfigService,
+                tbBasicDataConfigDetailService,tbScoreService)).sheet().doRead();
     }
-
+    private void calculateScore(TbBasicData tbBasicData) {
+        //计算得分
+        List<TbBasicDataConfig> list = tbBasicDataConfigService.lambdaQuery().eq(TbBasicDataConfig::getStatus, ShowStatusEnum.SHOW).list();
+        if (CollectionUtils.isEmpty(list)) {
+            throw new ServiceException("计算得分失败,平台未配置得分计算规则");
+        }
+        List<TbBasicDataConfig> numCalculates = list.stream()
+                .filter(item -> CalculateTypeEnum.NUMBER.equals(item.getCalculateType()))
+                .collect(Collectors.toList());
+        List<TbBasicDataConfig> textAndPercentages = list.stream()
+                .filter(item -> !CalculateTypeEnum.NUMBER.equals(item.getCalculateType()))
+                .collect(Collectors.toList());
+        if (CollectionUtils.isNotEmpty(numCalculates)) {
+            numCalculates.forEach(item -> {
+                Map<String, Object> valueMap = new HashMap<>();
+                String numberCalculateFormula = item.getNumberCalculateFormula();
+                Map<String, Integer> fieldsAndValue = CalculateUtil.getFieldsAndValue(numberCalculateFormula);
+                for (Map.Entry<String, Integer> stringIntegerEntry : fieldsAndValue.entrySet()) {
+                    Optional<TbBasicDataField> tbBasicDataField = tbBasicDataFieldService.lambdaQuery()
+                            .eq(TbBasicDataField::getBasicDataId, tbBasicData.getId())
+                            .eq(TbBasicDataField::getFieldId, stringIntegerEntry.getValue())
+                            .oneOpt();
+                    tbBasicDataField.ifPresent(basicDataField -> valueMap.put(stringIntegerEntry.getKey(), basicDataField.getFieldValue()));
+                }
+                double score = CalculateUtil.calculate(numberCalculateFormula, valueMap);
+                TbScore tbScore = new TbScore();
+                tbScore.setBasicDataId(tbBasicData.getId());
+                tbScore.setScore(score);
+                tbScore.setBasicDataConfigId(item.getId());
+                tbScoreService.save(tbScore);
+            });
+        }
+        if (CollectionUtils.isNotEmpty(textAndPercentages)) {
+            for (TbBasicDataConfig textAndPercentage : textAndPercentages) {
+                TbScore tbScore = new TbScore();
+                List<TbBasicDataConfigDetail> details = tbBasicDataConfigDetailService.lambdaQuery()
+                        .eq(TbBasicDataConfigDetail::getBasicDataConfigId, textAndPercentage.getId())
+                        .list();
+                Map<String, String> scoreMap = details.stream().collect(Collectors.toMap(TbBasicDataConfigDetail::getKey, TbBasicDataConfigDetail::getValue));
+                if (CollectionUtils.isNotEmpty(details)) {
+                    Optional<TbBasicDataField> tbBasicDataFieldOptional = tbBasicDataFieldService.lambdaQuery()
+                            .eq(TbBasicDataField::getBasicDataId, tbBasicData.getId())
+                            .eq(TbBasicDataField::getFieldId, textAndPercentage.getFieldIdStr())
+                            .oneOpt();
+                    if (tbBasicDataFieldOptional.isPresent()) {
+                        TbBasicDataField tbBasicDataField = tbBasicDataFieldOptional.get();
+                        if (CalculateTypeEnum.TEXT.equals(textAndPercentage.getCalculateType())) {
+                            String score = scoreMap.get(tbBasicDataField.getFieldValue());
+                            tbScore.setBasicDataId(tbBasicData.getId());
+                            tbScore.setScore(Double.parseDouble(score));
+                            tbScore.setBasicDataConfigId(textAndPercentage.getId());
+                            tbScoreService.save(tbScore);
+                        }
+                        if (CalculateTypeEnum.PERCENTAGE.equals(textAndPercentage.getCalculateType())) {
+                            for (Map.Entry<String, String> stringStringEntry : scoreMap.entrySet()) {
+                                String[] split = stringStringEntry.getKey().split("-");
+                                double v = Double.parseDouble(tbBasicDataField.getFieldValue());
+                                double min = Double.parseDouble(split[0]);
+                                double max = Double.parseDouble(split[1]);
+                                if (v >= min && v <= max) {
+                                    tbScore.setScore(Double.parseDouble(stringStringEntry.getValue()));
+                                }
+                            }
+                            tbScore.setBasicDataId(tbBasicData.getId());
+                            tbScore.setBasicDataConfigId(textAndPercentage.getId());
+                            tbScoreService.save(tbScore);
+                        }
+                    }
+                }
+            }
+        }
+    }
     @Override
     public PageDTO<ScoreVO> pageScore(ScoreQuery query) {
         Page<ScoreVO> page = new Page<>(query.getPageNum(), query.getPageSize());
-        //todo 获取当前登录部门
-        //校验区划代码
-        TbDept dept = tbDeptService.getById(44);
+        SysUser user = SecurityUtils.getLoginUser().getUser();
+        String areaCode = user.getAreaCode();
         Map<String, Date> quarterDate = DateUtils.getQuarterDate(new Date());
         //当前季度开始
         Date quarterStart = quarterDate.get("first");
@@ -213,14 +345,14 @@
         Date quarterEnd = quarterDate.get("last");
         //查询是否有当前季度的填报记录
         TbBasicData basicData = this.getOne(Wrappers.<TbBasicData>lambdaQuery()
-                .eq(TbBasicData::getDeptAreaCode, dept.getAreaCode())
+                .eq(TbBasicData::getDeptAreaCode, areaCode)
                 .between(TbBasicData::getCreateTime, quarterStart,quarterEnd));
         if (Objects.isNull(basicData)) {
             return PageDTO.empty(page);
         }
         query.setBasicDataId(basicData.getId());
         //查询对应的基础数据配置
-        List<ScoreVO> vos = scoreService.pageScore(query,page);
+        List<ScoreVO> vos = tbScoreService.pageScore(query,page);
         return PageDTO.of(page, vos);
     }
 }

--
Gitblit v1.7.1