From a7a8a50f48c1bff5ddc52b3bad1782328f8043d8 Mon Sep 17 00:00:00 2001 From: mitao <2763622819@qq.com> Date: 星期五, 12 四月 2024 18:50:20 +0800 Subject: [PATCH] 基础数据配置接口修改 --- ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TbBasicDataServiceImpl.java | 836 +++++++++++++++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 743 insertions(+), 93 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 e6639e2..91d061c 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,50 +1,84 @@ package com.ruoyi.system.service.impl; import com.alibaba.excel.EasyExcel; +import com.alibaba.excel.ExcelWriter; +import com.alibaba.excel.write.metadata.WriteSheet; +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.NumberDisplaceChineseUtil; +import com.ruoyi.common.basic.PageDTO; import com.ruoyi.common.core.domain.R; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.enums.CalculateTypeEnum; import com.ruoyi.common.enums.ReportingStatusEnum; import com.ruoyi.common.enums.ShowStatusEnum; +import com.ruoyi.common.enums.UserTypeEnum; import com.ruoyi.common.exception.ServiceException; import com.ruoyi.common.utils.BeanUtils; +import com.ruoyi.common.utils.CalculateUtil; import com.ruoyi.common.utils.CollUtils; import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.EasyExcelUtil; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; import com.ruoyi.system.domain.TbBasicData; +import com.ruoyi.system.domain.TbBasicDataConfig; +import com.ruoyi.system.domain.TbBasicDataConfigDetail; import com.ruoyi.system.domain.TbBasicDataField; import com.ruoyi.system.domain.TbDept; import com.ruoyi.system.domain.TbField; +import com.ruoyi.system.domain.TbScore; import com.ruoyi.system.dto.BasicDataDTO; +import com.ruoyi.system.handler.CustomCellWriteHandler; import com.ruoyi.system.listener.BasicDataListener; +import com.ruoyi.system.listener.HistoryDataListener; import com.ruoyi.system.mapper.TbBasicDataMapper; +import com.ruoyi.system.mapper.TbDeptMapper; +import com.ruoyi.system.query.CurrentFieldsQuery; +import com.ruoyi.system.query.ScoreCalculateQuery; +import com.ruoyi.system.query.ScoreQuery; +import com.ruoyi.system.service.ISysUserService; +import com.ruoyi.system.service.TbBasicDataConfigDetailService; +import com.ruoyi.system.service.TbBasicDataConfigService; import com.ruoyi.system.service.TbBasicDataFieldService; import com.ruoyi.system.service.TbBasicDataService; -import com.ruoyi.system.service.TbDeptService; import com.ruoyi.system.service.TbFieldService; +import com.ruoyi.system.service.TbScoreService; +import com.ruoyi.system.utils.FieldBuildUtil; import com.ruoyi.system.vo.BasicDataReportingVO; -import com.ruoyi.system.vo.FieldReportingVO; +import com.ruoyi.system.vo.CurrentFieldsAllVO; +import com.ruoyi.system.vo.CurrentFieldsDetailVO; +import com.ruoyi.system.vo.CurrentFieldsVO; +import com.ruoyi.system.vo.FieldsTreeVO; +import com.ruoyi.system.vo.ScoreCalculateVO; +import com.ruoyi.system.vo.ScoreVO; +import java.io.IOException; +import java.net.URLEncoder; +import java.time.Instant; +import java.time.LocalDate; +import java.time.ZoneId; +import java.util.ArrayList; +import java.util.Date; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.Objects; +import java.util.Optional; +import java.util.Set; +import java.util.function.Function; +import java.util.stream.Collectors; +import javax.servlet.http.HttpServletResponse; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import org.springframework.web.multipart.MultipartFile; -import javax.servlet.http.HttpServletResponse; -import java.io.IOException; -import java.net.URLEncoder; -import java.time.Instant; -import java.time.LocalDate; -import java.time.ZoneId; -import java.util.*; -import java.util.function.Function; -import java.util.stream.Collectors; - /** - * <p> * 基础数据表 服务实现类 - * </p> * * @author mitao * @since 2024-03-13 @@ -52,69 +86,108 @@ @Slf4j @Service @RequiredArgsConstructor -public class TbBasicDataServiceImpl extends ServiceImpl<TbBasicDataMapper, TbBasicData> implements TbBasicDataService { +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 tbScoreService; + private final ISysUserService sysUserService; + + public static void setFieldValues( + List<FieldsTreeVO> fields, Map<Long, TbBasicDataField> fieldMap) { + for (FieldsTreeVO field : fields) { + TbBasicDataField tbBasicDataField = fieldMap.get(field.getId()); + if (tbBasicDataField != null) { + field.setValue( + FieldBuildUtil.formatNumberWithCommas(tbBasicDataField.getFieldValue())); + } + if (field.getChildren() != null && !field.getChildren().isEmpty()) { + setFieldValues(field.getChildren(), fieldMap); + } + } + } @Override - public R<BasicDataReportingVO> getBasicFields(String deptAreaCode) throws Exception { + public R<BasicDataReportingVO> getBasicFields() throws Exception { + SysUser user = SecurityUtils.getLoginUser().getUser(); + String deptAreaCode = user.getAreaCode(); 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)); } Date date = new Date(); - //当前所在季度 - int quarterOfYear = DateUtils.getQuarterOfYear(date); - String quarterOfYearStr = NumberDisplaceChineseUtil.numberToChinese(quarterOfYear); Map<String, Date> quarterDate = DateUtils.getQuarterDate(date); - //当前季度开始 + // 当前季度开始 Date quarterStart = quarterDate.get("first"); - //当前季度结束 + // 当前季度结束 Date quarterEnd = quarterDate.get("last"); - //判断当前时间是否在季度初1-15号 + // 判断当前时间是否在季度初1-15号 Instant instant = quarterStart.toInstant(); LocalDate quarterStartLocalDate = instant.atZone(ZoneId.systemDefault()).toLocalDate(); LocalDate fifteenDaysLimit = quarterStartLocalDate.plusDays(15); LocalDate now = LocalDate.now(); - vo.setQuarter(String.format("%s年%s季度",now.getYear(),quarterOfYearStr)); + vo.setQuarter(DateUtils.getNowQuarter()); vo.setStatus(ReportingStatusEnum.UNFILLED); - //如果当前时间不在规定范围内:季度初1-15号 + // 如果当前时间不在规定范围内:季度初1-15号 if (now.isBefore(quarterStartLocalDate) || now.isAfter(fifteenDaysLimit)) { - return R.ok(vo,"请于季度初1-15号上传季度数据。"); + return R.ok(vo, "请于季度初1-15号上传季度数据。"); } - //查询是否有当前季度的填报记录 - TbBasicData basicData = this.getOne(Wrappers.<TbBasicData>lambdaQuery() - .eq(TbBasicData::getDeptAreaCode, dept.getAreaCode()) - .between(TbBasicData::getCreateTime, quarterStart,quarterEnd)); - //查询需要填写的字段 - List<TbField> list = tbFieldService.lambdaQuery().eq(TbField::getStatus, ShowStatusEnum.SHOW).list(); + // 查询是否有当前季度的填报记录 + TbBasicData basicData = + this.getOne( + Wrappers.<TbBasicData>lambdaQuery() + .eq(TbBasicData::getDeptAreaCode, dept.getAreaCode()) + .between(TbBasicData::getCreateTime, quarterStart, quarterEnd)); + // 查询需要填写的字段 + List<TbField> list = + tbFieldService.lambdaQuery().eq(TbField::getStatus, ShowStatusEnum.SHOW).list(); + List<FieldsTreeVO> roots = new ArrayList<>(); + FieldsTreeVO fieldsTreeVO = new FieldsTreeVO(); + fieldsTreeVO.setId(-1L); + fieldsTreeVO.setName("转移支付规模"); + fieldsTreeVO.setCategory(Boolean.FALSE); + FieldsTreeVO fieldsTreeVO2 = new FieldsTreeVO(); + fieldsTreeVO2.setId(-2L); + fieldsTreeVO2.setName("当期GDP"); + fieldsTreeVO2.setCategory(Boolean.FALSE); + roots.add(fieldsTreeVO); + roots.add(fieldsTreeVO2); if (CollUtils.isNotEmpty(list)) { - List<FieldReportingVO> fieldReportingVOS = BeanUtils.copyList(list, FieldReportingVO.class); - vo.setFields(fieldReportingVOS); + FieldBuildUtil.buildTreeFromTbFieldList(list, roots); + FieldsTreeVO remark = new FieldsTreeVO(); + remark.setId(-3L); + remark.setName("备注"); + roots.add(remark); + vo.setFields(roots); } if (Objects.isNull(basicData)) { - vo.setQuarter(String.format("%s年%s季度",now.getYear(),quarterOfYearStr)); + vo.setQuarter(DateUtils.getNowQuarter()); vo.setStatus(ReportingStatusEnum.UNFILLED); return R.ok(vo); - }else { + } else { vo.setStatus(basicData.getStatus()); - //查询已填报数据 包含数据缺失和已填报 - List<TbBasicDataField> basicDataFields = tbBasicDataFieldService.lambdaQuery() - .eq(TbBasicDataField::getBasicDataId, basicData.getId()).list(); + // 查询已填报数据 包含数据缺失和已填报 + List<TbBasicDataField> basicDataFields = + tbBasicDataFieldService + .lambdaQuery() + .eq(TbBasicDataField::getBasicDataId, basicData.getId()) + .list(); + if (CollUtils.isNotEmpty(basicDataFields)) { - Map<Integer, TbBasicDataField> fieldMap = basicDataFields.stream() - .collect(Collectors.toMap(TbBasicDataField::getFieldId, Function.identity())); - vo.getFields().forEach(item -> { - TbBasicDataField tbBasicDataField = fieldMap.get(item.getId()); - if (Objects.nonNull(tbBasicDataField)) { - item.setValue(tbBasicDataField.getFieldValue()); - } - }); + Map<Long, TbBasicDataField> fieldMap = + basicDataFields.stream() + .collect(Collectors.toMap(TbBasicDataField::getFieldId, + Function.identity())); + setFieldValues(vo.getFields(), fieldMap); return R.ok(vo); } } @@ -123,78 +196,655 @@ @Override @Transactional(rollbackFor = Exception.class) - public void saveBasicData(BasicDataDTO dto) { - //TODO - //LoginUser loginUser = SecurityUtils.getLoginUser(); - //数据校验 + public void saveBasicData(BasicDataDTO dto) throws Exception { + LoginUser loginUser = SecurityUtils.getLoginUser(); + String areaCode = loginUser.getUser().getAreaCode(); + // 数据校验 if (Objects.isNull(dto) || CollUtils.isEmpty(dto.getFields())) { return; } - + String nowQuarter = DateUtils.getNowQuarter(); TbBasicData tbBasicData = BeanUtils.copyBean(dto, TbBasicData.class); + tbBasicData.setDeptAreaCode(areaCode); + tbBasicData.setStatus(ReportingStatusEnum.MISSING_DATA); + tbBasicData.setQuarter(nowQuarter); + tbBasicData.setReportingTime(DateUtils.getQuarterDate(nowQuarter)); this.saveOrUpdate(tbBasicData); - //保存基础数据动态字段数据 - List<TbBasicDataField> tbBasicDataFields = BeanUtils.copyList(dto.getFields(), TbBasicDataField.class); - tbBasicDataFields.forEach(item -> item.setBasicDataId(tbBasicData.getId())); - //将该基础数据的动态字段数据全部删除 - tbBasicDataFieldService.remove(null); + // 保存基础数据动态字段数据 + List<TbBasicDataField> tbBasicDataFields = + BeanUtils.copyList(dto.getFields(), TbBasicDataField.class); + // 查询需要填写的动态字段 + List<TbField> fieldList = + tbFieldService.lambdaQuery().eq(TbField::getStatus, ShowStatusEnum.SHOW).list(); + Map<Long, TbField> fieldMap = + fieldList.stream().collect(Collectors.toMap(TbField::getId, e -> e)); + tbBasicDataFields.forEach( + item -> { + item.setBasicDataId(tbBasicData.getId()); + TbField tbField = fieldMap.get(item.getFieldId()); + if (Objects.nonNull(tbField)) { + item.setFieldName(tbField.getFieldName()); + } + }); + // 添加固定字段 转移支付规模、当期GDP + TbBasicDataField transferPaymentScale = new TbBasicDataField(); + transferPaymentScale.setBasicDataId(tbBasicData.getId()); + transferPaymentScale.setFieldId(-1L); + transferPaymentScale.setFieldValue(tbBasicData.getTransferPaymentScale()); + tbBasicDataFields.add(transferPaymentScale); + TbBasicDataField currentGdp = new TbBasicDataField(); + currentGdp.setBasicDataId(tbBasicData.getId()); + currentGdp.setFieldId(-2L); + currentGdp.setFieldValue(tbBasicData.getCurrentGdp()); + tbBasicDataFields.add(currentGdp); + TbBasicDataField remark = new TbBasicDataField(); + remark.setBasicDataId(tbBasicData.getId()); + remark.setFieldId(-3L); + remark.setFieldValue(tbBasicData.getRemark()); + tbBasicDataFields.add(remark); + // 将该基础数据的动态字段数据全部删除 + 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 - public void downloadImportTemplate() throws Exception { + @Transactional(rollbackFor = Exception.class) + public void importBasicData(MultipartFile file) throws Exception { + LoginUser loginUser = SecurityUtils.getLoginUser(); + String areaCode = loginUser.getUser().getAreaCode(); + EasyExcel.read( + file.getInputStream(), + new BasicDataListener( + this, + 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()); + SysUser user = SecurityUtils.getLoginUser().getUser(); + String areaCode = user.getAreaCode(); + Map<String, Date> quarterDate = DateUtils.getQuarterDate(new Date()); + // 当前季度开始 + Date quarterStart = quarterDate.get("first"); + // 当前季度结束 + Date quarterEnd = quarterDate.get("last"); + // 查询是否有当前季度的填报记录 + TbBasicData basicData = + this.getOne( + Wrappers.<TbBasicData>lambdaQuery() + .eq(TbBasicData::getDeptAreaCode, areaCode) + .between(TbBasicData::getCreateTime, quarterStart, quarterEnd)); + if (Objects.isNull(basicData)) { + return PageDTO.empty(page); + } + query.setBasicDataId(basicData.getId()); + // 查询对应的基础数据配置 + Page<ScoreVO> pageVO = tbScoreService.pageScore(query, page); + return PageDTO.of(pageVO); + } + + @Override + public R<CurrentFieldsDetailVO> fieldsDetails(Long id) { + // 从数据库中获取基本数据 + TbBasicData basicData = this.getById(id); + if (Objects.isNull(basicData)) { + throw new ServiceException("非法参数"); + } + CurrentFieldsDetailVO vo = BeanUtils.copyBean(basicData, CurrentFieldsDetailVO.class); + // 查询用户信息 + sysUserService + .lambdaQuery() + .eq(SysUser::getAreaCode, basicData.getDeptAreaCode()) + .eq(SysUser::getUserType, UserTypeEnum.DEPARTMENT.getCode()) + .oneOpt() + .ifPresent( + item -> { + vo.setAreaName(item.getAreaName()); + vo.setPhoneNumber(item.getPhoneNumber()); + vo.setPersonInCharge(item.getPersonInCharge()); + }); + // 查询动态字段 + List<TbBasicDataField> basicDataFields = + tbBasicDataFieldService + .lambdaQuery() + .eq(TbBasicDataField::getBasicDataId, basicData.getId()) + .list(); + // 获取所有字段ID + Set<Long> fieldIds = + basicDataFields.stream().map(TbBasicDataField::getFieldId) + .collect(Collectors.toSet()); + // 根据字段ID查询字段信息并构建字段ID到字段对象的映射 + Map<Long, TbField> fieldMap = + tbFieldService + .lambdaQuery() + .in(!fieldIds.isEmpty(), TbField::getId, fieldIds) + .list() + .stream() + .collect(Collectors.toMap(TbField::getId, e -> e)); + // 根节点 + List<FieldsTreeVO> root = new ArrayList<>(); + basicDataFields.stream() + .filter(item -> item.getFieldId() == -1) + .findFirst() + .ifPresent( + item -> { + FieldsTreeVO fieldsTreeVO = new FieldsTreeVO(); + fieldsTreeVO.setId(item.getFieldId()); + fieldsTreeVO.setName("转移支付规模"); + fieldsTreeVO.setValue(item.getFieldValue()); + fieldsTreeVO.setCategory(Boolean.FALSE); + root.add(fieldsTreeVO); + }); + basicDataFields.stream() + .filter(item -> item.getFieldId() == -2) + .findFirst() + .ifPresent( + item -> { + FieldsTreeVO fieldsTreeVO = new FieldsTreeVO(); + fieldsTreeVO.setId(item.getFieldId()); + fieldsTreeVO.setName("当期GDP"); + fieldsTreeVO.setValue(item.getFieldValue()); + fieldsTreeVO.setCategory(Boolean.FALSE); + root.add(fieldsTreeVO); + }); + FieldBuildUtil.buildTreeStructure(basicDataFields, fieldMap, root); + basicDataFields.stream() + .filter(item -> item.getFieldId() == -3) + .findFirst() + .ifPresent( + item -> { + FieldsTreeVO fieldsTreeVO = new FieldsTreeVO(); + fieldsTreeVO.setId(item.getFieldId()); + fieldsTreeVO.setName("备注"); + fieldsTreeVO.setValue(item.getFieldValue()); + fieldsTreeVO.setCategory(Boolean.FALSE); + root.add(fieldsTreeVO); + }); + vo.setFields(root); + return R.ok(vo); + } + + @Override + public R<PageDTO<CurrentFieldsVO>> fieldsStatics(CurrentFieldsQuery dto) { + Page<CurrentFieldsVO> page = new Page<>(dto.getPageNum(), dto.getPageSize()); + return R.ok(PageDTO.of(baseMapper.fieldsStatics(page, dto))); + } + + @Override + public CurrentFieldsAllVO fieldsStaticsAll() throws Exception { + CurrentFieldsAllVO vo = new CurrentFieldsAllVO(); + List<FieldsTreeVO> roots = new ArrayList<>(); + // 获取表头 + FieldsTreeVO area = new FieldsTreeVO(); + area.setCategory(Boolean.FALSE); + area.setName("地区"); + FieldsTreeVO quarter = new FieldsTreeVO(); + quarter.setCategory(Boolean.FALSE); + quarter.setName("填报季度"); + FieldsTreeVO transferPaymentScale = new FieldsTreeVO(); + transferPaymentScale.setName("转移支付规模"); + FieldsTreeVO currentGdp = new FieldsTreeVO(); + currentGdp.setCategory(Boolean.FALSE); + currentGdp.setName("当期GDP"); + roots.add(area); + roots.add(quarter); + roots.add(transferPaymentScale); + roots.add(currentGdp); + // 查询当前季度填写字段 + List<TbField> fieldList = + tbFieldService.lambdaQuery().eq(TbField::getStatus, ShowStatusEnum.SHOW).list(); + FieldBuildUtil.buildTreeFromTbFieldList(fieldList, roots); + FieldsTreeVO remark = new FieldsTreeVO(); + remark.setCategory(Boolean.FALSE); + remark.setName("备注"); + roots.add(remark); + vo.setFields(roots); + String nowQuarter = DateUtils.getNowQuarter(); + // 查询上报的基础数据 + List<TbBasicData> basicDataList = + this.lambdaQuery() + .eq(TbBasicData::getQuarter, nowQuarter) + .eq(TbBasicData::getStatus, ReportingStatusEnum.FILLED) + .list(); + if (CollUtils.isNotEmpty(basicDataList)) { + Set<String> areaCodeList = + basicDataList.stream().map(TbBasicData::getDeptAreaCode) + .collect(Collectors.toSet()); + Map<String, SysUser> userMap = + sysUserService.lambdaQuery().in(SysUser::getAreaCode, areaCodeList).list() + .stream() + .collect(Collectors.toMap(SysUser::getAreaCode, e -> e)); + Set<Long> basicDataIds = + basicDataList.stream().map(TbBasicData::getId).collect(Collectors.toSet()); + Map<Long, TbBasicDataField> basicDataFieldMap = + tbBasicDataFieldService + .lambdaQuery() + .in(TbBasicDataField::getBasicDataId, basicDataIds) + .list() + .stream() + .collect(Collectors.toMap(TbBasicDataField::getFieldId, e -> e)); + // 值 + List<Object> result = new ArrayList<>(); + for (TbBasicData tbBasicData : basicDataList) { + SysUser user = userMap.get(tbBasicData.getDeptAreaCode()); + List<String> item = new ArrayList<>(); + item.add(Objects.nonNull(user) ? user.getAreaName() : ""); + item.add(tbBasicData.getQuarter()); + item.add(tbBasicData.getTransferPaymentScale()); + item.add(tbBasicData.getCurrentGdp()); + for (TbField tbField : fieldList) { + TbBasicDataField tbBasicDataField = basicDataFieldMap.get(tbField.getId()); + item.add( + Objects.nonNull(tbBasicDataField) + ? FieldBuildUtil.formatNumberWithCommas( + tbBasicDataField.getFieldValue()) + : ""); + } + item.add(tbBasicData.getRemark()); + result.add(item); + } + vo.setValue(result); + } + return vo; + } + + @Override + public R<PageDTO<CurrentFieldsVO>> historyFieldsStatics(CurrentFieldsQuery dto) { + Page<CurrentFieldsVO> page = new Page<>(dto.getPageNum(), dto.getPageSize()); + // 当前所在季度 + return R.ok(PageDTO.of(baseMapper.fieldsStatics(page, dto))); + } + + @Override + public PageDTO<ScoreCalculateVO> scoreCalculatePage(ScoreCalculateQuery query) { + Page<ScoreCalculateVO> page = new Page<>(query.getPageNum(), query.getPageSize()); + Page<ScoreCalculateVO> pageVO = tbBasicDataConfigService.scoreCalculatePage(query, page); + return PageDTO.of(pageVO); + } + + @Override + public CurrentFieldsAllVO fieldsStaticsAll(String quarterStr) { + CurrentFieldsAllVO vo = new CurrentFieldsAllVO(); + List<FieldsTreeVO> roots = new ArrayList<>(); + // 获取表头 + FieldsTreeVO area = new FieldsTreeVO(); + area.setCategory(Boolean.FALSE); + area.setName("地区"); + FieldsTreeVO quarter = new FieldsTreeVO(); + quarter.setCategory(Boolean.FALSE); + quarter.setName("填报季度"); + FieldsTreeVO transferPaymentScale = new FieldsTreeVO(); + transferPaymentScale.setName("转移支付规模"); + FieldsTreeVO currentGdp = new FieldsTreeVO(); + currentGdp.setCategory(Boolean.FALSE); + currentGdp.setName("当期GDP"); + roots.add(area); + roots.add(quarter); + roots.add(transferPaymentScale); + roots.add(currentGdp); + List<TbBasicData> basicDataList = + this.lambdaQuery() + .eq(TbBasicData::getQuarter, quarterStr) + .eq(TbBasicData::getStatus, ReportingStatusEnum.FILLED) + .list(); + if (CollUtils.isEmpty(basicDataList)) { + throw new ServiceException("非法参数"); + } + Long basicDataId = basicDataList.get(0).getId(); + Set<Long> fieldIds = + tbBasicDataFieldService + .lambdaQuery() + .eq(TbBasicDataField::getBasicDataId, basicDataId) + .ne(TbBasicDataField::getFieldId, -1L) + .ne(TbBasicDataField::getFieldId, -2L) + .list() + .stream() + .map(TbBasicDataField::getFieldId) + .collect(Collectors.toSet()); + List<TbField> fieldList = tbFieldService.lambdaQuery().in(TbField::getId, fieldIds).list(); + if (CollUtils.isEmpty(fieldList)) { + FieldsTreeVO remark = new FieldsTreeVO(); + remark.setCategory(Boolean.FALSE); + remark.setName("备注"); + roots.add(remark); + vo.setFields(roots); + return vo; + } + FieldBuildUtil.buildTreeFromTbFieldList(fieldList, roots); + FieldsTreeVO remark = new FieldsTreeVO(); + remark.setCategory(Boolean.FALSE); + remark.setName("备注"); + roots.add(remark); + vo.setFields(roots); + Set<String> areaCodeList = + basicDataList.stream().map(TbBasicData::getDeptAreaCode) + .collect(Collectors.toSet()); + Map<String, SysUser> userMap = + sysUserService.lambdaQuery().in(SysUser::getAreaCode, areaCodeList).list().stream() + .collect(Collectors.toMap(SysUser::getAreaCode, e -> e)); + Set<Long> basicDataIds = + basicDataList.stream().map(TbBasicData::getId).collect(Collectors.toSet()); + List<TbBasicDataField> basicDataFieldList = tbBasicDataFieldService + .lambdaQuery() + .in(TbBasicDataField::getBasicDataId, basicDataIds) + .list(); + // 值 + List<Object> result = new ArrayList<>(); + for (TbBasicData tbBasicData : basicDataList) { + Map<Long, TbBasicDataField> basicDataFieldMap = basicDataFieldList.stream() + .filter(item -> tbBasicData.getId().equals(item.getBasicDataId())) + .collect(Collectors.toMap(TbBasicDataField::getFieldId, e -> e)); + SysUser user = userMap.get(tbBasicData.getDeptAreaCode()); + List<String> item = new ArrayList<>(); + item.add(Objects.nonNull(user) ? user.getAreaName() : ""); + item.add(tbBasicData.getQuarter()); + item.add(tbBasicData.getTransferPaymentScale()); + item.add(tbBasicData.getCurrentGdp()); + for (TbField tbField : fieldList) { + TbBasicDataField tbBasicDataField = basicDataFieldMap.get(tbField.getId()); + item.add( + Objects.nonNull(tbBasicDataField) + ? FieldBuildUtil.formatNumberWithCommas( + tbBasicDataField.getFieldValue()) + : ""); + } + item.add(tbBasicData.getRemark()); + result.add(item); + } + vo.setValue(result); + return vo; + } + + @Override + @Transactional(rollbackFor = Exception.class) + public void importData(MultipartFile file, String quarter) throws IOException { + //校验季度 + Date quarterDate = DateUtils.getQuarterDate(quarter); + + Map<String, Date> quarterMap = DateUtils.getQuarterDate(new Date()); + Date date = quarterMap.get("first"); + LocalDate quarterLocalDate = DateUtils.dateToLocalDate(quarterDate); + LocalDate dateLocalDate = DateUtils.dateToLocalDate(date); + if (quarterLocalDate.isAfter(dateLocalDate) || quarterLocalDate.equals(dateLocalDate)) { + throw new ServiceException("请导入过去季度的数据。"); + } + + EasyExcel.read( + file.getInputStream(), + new HistoryDataListener( + this, + tbFieldService, + tbBasicDataFieldService, + tbBasicDataConfigService, + tbBasicDataConfigDetailService, + tbScoreService, sysUserService, quarter)) + .sheet() + .doRead(); + } + + @Override + public void exportData(List<String> quarterList) throws Exception { response.setContentType("application/vnd.ms-excel"); response.setCharacterEncoding("utf-8"); String fileName = "地方财政运行及“三保”情况统计表"; - // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系 - response.setHeader("Content-disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8") + ".xlsx"); - //查询需要填写的动态字段 - List<TbField> list = tbFieldService.lambdaQuery().eq(TbField::getStatus, ShowStatusEnum.SHOW).list(); - // 这里需要设置不关闭流 - EasyExcel.write(response.getOutputStream()).head(head(list)) - .autoCloseStream(Boolean.FALSE).sheet("模板") - .doWrite(dataList(list)); + response.setHeader( + "Content-disposition", + "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8") + ".xlsx"); + //季度数据,用于反查字段 + List<TbBasicData> tbBasicDataList = this.lambdaQuery() + .in(TbBasicData::getQuarter, quarterList) + .eq(TbBasicData::getStatus, ReportingStatusEnum.FILLED) + .groupBy(TbBasicData::getQuarter).list(); + //符合导出条件的基础数据 + List<TbBasicData> tbBasicDataListAll = this.lambdaQuery() + .in(TbBasicData::getQuarter, quarterList) + .eq(TbBasicData::getStatus, ReportingStatusEnum.FILLED) + .list(); + //基础数据id列表 + Set<Long> basicDataIdList = tbBasicDataListAll.stream().map(TbBasicData::getId) + .collect(Collectors.toSet()); + //涉及到的部门编码 + Set<String> userAreaCodeList = tbBasicDataListAll.stream().map(TbBasicData::getDeptAreaCode) + .collect(Collectors.toSet()); + //用户信息map + Map<String, SysUser> userMap = sysUserService.lambdaQuery() + .in(SysUser::getAreaCode, userAreaCodeList).list().stream() + .collect(Collectors.toMap(SysUser::getAreaCode, e -> e)); + //基础数据对应的字段值列表 + List<TbBasicDataField> tbBasicDataFieldList = tbBasicDataFieldService.lambdaQuery() + .in(TbBasicDataField::getBasicDataId, basicDataIdList) + .list(); + //构建Excel写对象 + try (ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()) + .registerWriteHandler(new CustomCellWriteHandler()) + .registerWriteHandler(EasyExcelUtil.getStyleStrategy()).build()) { + WriteSheet writeSheet; + //遍历季度基础数据 + for (TbBasicData tbBasicData : tbBasicDataList) { + //该季度的动态字段id列表 + Set<Long> fieldIdList = tbBasicDataFieldList.stream() + .filter(item -> tbBasicData.getId().equals(item.getBasicDataId())) + .map(TbBasicDataField::getFieldId) + .collect(Collectors.toSet()); + //字段列表 + List<TbField> fieldList = tbFieldService.lambdaQuery() + .in(TbField::getId, fieldIdList) + .list(); + //表头 + List<List<String>> head = head(fieldList); + //过滤基础数据为遍历元素的季度 + List<TbBasicData> list = tbBasicDataListAll.stream() + .filter(item -> tbBasicData.getQuarter().equals(item.getQuarter())) + .collect(Collectors.toList()); + List<List<Object>> dataList = dataList(fieldList, list, tbBasicDataFieldList, + userMap, + head); + // 构建sheet对象 + writeSheet = EasyExcel.writerSheet(tbBasicData.getQuarter()).head(head).build(); + // 写出sheet数据 + excelWriter.write(dataList, writeSheet); + } + } } - private List<List<Object>> dataList(List<TbField> list) throws Exception { - //TODO - //LoginUser loginUser = SecurityUtils.getLoginUser(); - TbDept dept = tbDeptService.getById(44); - //当前所在季度 - int quarterOfYear = DateUtils.getQuarterOfYear(new Date()); - String quarterOfYearStr = NumberDisplaceChineseUtil.numberToChinese(quarterOfYear); - List<List<Object>> excellist = new ArrayList<List<Object>>(); - List<List<String>> head = head(list); + private List<List<Object>> dataList(List<TbField> fieldList, List<TbBasicData> list, + List<TbBasicDataField> tbBasicDataFieldList, Map<String, SysUser> userMap, + List<List<String>> head) { + //所有数据集合 + List<List<Object>> excellist = new ArrayList<>(); + //构建栏号行 List<Object> columnNo = Lists.newArrayList("栏号"); for (int i = 1; i < head.size(); i++) { columnNo.add(String.valueOf(i)); } excellist.add(columnNo); - excellist.add(Lists.newArrayList(dept.getAreaName(), String.format("%s季度",quarterOfYearStr))); + //动态字段数据行 + for (TbBasicData tbBasicData : list) { + //转换为map,方便遍历的时候取 + Map<Long, TbBasicDataField> basicDataFieldMap = tbBasicDataFieldList.stream() + .filter(item -> tbBasicData.getId().equals(item.getBasicDataId())) + .collect(Collectors.toMap(TbBasicDataField::getFieldId, e -> e)); + SysUser sysUser = userMap.get(tbBasicData.getDeptAreaCode()); + if (CollUtils.isEmpty(basicDataFieldMap) || Objects.isNull(sysUser)) { + continue; + } + List<Object> valueList = Lists.newArrayList(sysUser.getAreaName(), + tbBasicData.getQuarter()); + TbBasicDataField transferPaymentScale = basicDataFieldMap.get(-1L); + valueList.add( + Objects.nonNull(transferPaymentScale) + ? transferPaymentScale.getFieldValue() + : ""); + TbBasicDataField currentGdp = basicDataFieldMap.get(-2L); + valueList.add( + Objects.nonNull(currentGdp) + ? currentGdp.getFieldValue() + : ""); + for (TbField tbField : fieldList) { + TbBasicDataField tbBasicDataField = basicDataFieldMap.get(tbField.getId()); + valueList.add( + Objects.nonNull(tbBasicDataField) + ? tbBasicDataField.getFieldValue() + : ""); + } + excellist.add(valueList); + valueList.add(tbBasicData.getRemark()); + } return excellist; } + private List<List<String>> head(List<TbField> list) { List<List<String>> headTitles = Lists.newArrayList(); - //固定字段 + // 固定字段 headTitles.add(Lists.newArrayList("地区")); headTitles.add(Lists.newArrayList("填报季度")); headTitles.add(Lists.newArrayList("转移支付规模")); headTitles.add(Lists.newArrayList("当期GDP")); - list.forEach(item ->{ - headTitles.add(Lists.newArrayList(item.getLevelOneCategory(), item.getLevelTwoCategory(), item.getLevelThreeCategory(), item.getFieldName())); - }); + list.forEach( + item -> { + 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 - public void importBasicData(MultipartFile file) throws IOException { - //TODO - //LoginUser loginUser = SecurityUtils.getLoginUser(); - TbDept dept = tbDeptService.getById(44); - //查询需要填写的动态字段 - List<TbField> fieldList = tbFieldService.lambdaQuery().eq(TbField::getStatus, ShowStatusEnum.SHOW).list(); - EasyExcel.read(file.getInputStream(), new BasicDataListener(this,fieldList,tbFieldService,dept,tbBasicDataFieldService)).sheet().doRead(); } } -- Gitblit v1.7.1