kcim-cost/src/main/java/com/imed/costaccount/service/impl/ShareParamValueServiceImpl.java

package com.imed.costaccount.service.impl;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.StrUtil;
import com.alibaba.druid.util.StringUtils;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.imed.costaccount.common.exception.CostException;
import com.imed.costaccount.common.util.*;
import com.imed.costaccount.constants.NumberConstant;
import com.imed.costaccount.enums.DateStyleEnum;
import com.imed.costaccount.mapper.ShareParamValueMapper;
import com.imed.costaccount.model.*;
import com.imed.costaccount.model.dto.CopyShareParamValueDTO;
import com.imed.costaccount.model.dto.ShareParamValueEditDTO;
import com.imed.costaccount.model.dto.ShareParamValueSaveDTO;
import com.imed.costaccount.model.dto.ShareParamValueVO;
import com.imed.costaccount.model.vo.IncomeErrorMessage;
import com.imed.costaccount.service.*;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.web.multipart.MultipartFile;

import java.math.BigDecimal;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Collectors;

import static cn.hutool.core.date.DatePattern.PURE_DATE_PATTERN;


@Service("shareParamValueService")
public class ShareParamValueServiceImpl extends ServiceImpl<ShareParamValueMapper, ShareParamValue> implements ShareParamValueService {

    private final CostIncomeGroupServiceImpl costIncomeGroupService;

    private final CostShareParamService costShareParamService;

    private final CostIncomeFileService costIncomeFileService;

    private final ResponsibilityService responsibilityService;

    private final IncomeCollectionService incomeCollectionService;


    public ShareParamValueServiceImpl(CostIncomeGroupServiceImpl costIncomeGroupService,
                                      CostShareParamService costShareParamService,
                                      CostIncomeFileService costIncomeFileService,
                                      ResponsibilityService responsibilityService, IncomeCollectionService incomeCollectionService) {
        this.costIncomeGroupService = costIncomeGroupService;
        this.costShareParamService = costShareParamService;
        this.costIncomeFileService = costIncomeFileService;
        this.responsibilityService = responsibilityService;
        this.incomeCollectionService = incomeCollectionService;
    }


    /**
     * 添加分摊参数值
     *
     * @param shareParamValueSaveDTO {@link ShareParamValueSaveDTO}
     * @param hospId                 医院id
     */
    @Override
    @Transactional(propagation = Propagation.REQUIRED, rollbackFor = Throwable.class)
    public void addData(ShareParamValueSaveDTO shareParamValueSaveDTO, Long hospId) {
        ShareParamValue paramValue = BeanUtil.convertObj(shareParamValueSaveDTO, ShareParamValue.class);
        DateTime dateTime = DateUtil.parse(shareParamValueSaveDTO.getDate(), PURE_DATE_PATTERN);
        // 校验是否唯一
        Boolean flag = checkUniq(shareParamValueSaveDTO.getShareParamCode(), shareParamValueSaveDTO.getResponsibilityCode(), hospId, dateTime);
        if (flag) {
            throw new CostException("本月已存在对应责任中心和分摊参数中心，请直接修改");
        }
        // 校验两个code是否存在
        checkCodeIsExist(shareParamValueSaveDTO.getShareParamCode(), shareParamValueSaveDTO.getResponsibilityCode(), hospId);
        paramValue.setCreateTime(System.currentTimeMillis())
                .setDateYear(DateUtil.year(dateTime))
                .setDateMonth(DateUtil.month(dateTime) + 1)
                .setHospId(hospId)
                .setDataSourceType(0);
        this.save(paramValue);
    }

    /**
     * 校验两个code是否存在
     *
     * @param shareParamCode     成本分摊参数code
     * @param responsibilityCode 责任中心代码
     * @param hospId             医院id
     */
    @Transactional(rollbackFor = Throwable.class, propagation = Propagation.REQUIRED)
    public void checkCodeIsExist(String shareParamCode, String responsibilityCode, Long hospId) {
        List<CostShareParam> shareParams = costShareParamService.list(new LambdaQueryWrapper<CostShareParam>().select(CostShareParam::getId)
                .eq(CostShareParam::getShareParamCode, shareParamCode).eq(CostShareParam::getHospId, hospId));
        if (shareParams.isEmpty()) {
            throw new CostException("分摊参数code非法");
        }
        List<Responsibility> responsibilities = responsibilityService.list(
                new LambdaQueryWrapper<Responsibility>().select(Responsibility::getId)
                        .eq(Responsibility::getResponsibilityCode, responsibilityCode).eq(Responsibility::getHospId, hospId)
        );
        if (responsibilities.isEmpty()) {
            throw new CostException("责任中心code非法");
        }
    }

    /**
     * 校验是否已存在
     *
     * @param shareParamCode     分摊参数代码
     * @param responsibilityCode 责任中心代码
     * @param hospId             医院id
     * @param date               时间
     */
    private Boolean checkUniq(String shareParamCode, String responsibilityCode, Long hospId, DateTime date) {
        List<ShareParamValue> list = this.list(
                new LambdaQueryWrapper<ShareParamValue>()
                        .eq(ShareParamValue::getHospId, hospId)
                        .eq(ShareParamValue::getShareParamCode, shareParamCode)
                        .eq(ShareParamValue::getResponsibilityCode, responsibilityCode)
                        .eq(ShareParamValue::getDateYear, DateUtil.year(date))
                        .eq(ShareParamValue::getDateMonth, DateUtil.month(date) + 1)
        );
        return !list.isEmpty();
    }

    /**
     * 分摊参数值列表
     *
     * @param current            当前页
     * @param pageSize           每页数据量大小
     * @param date               日期
     * @param shareParamCode     分摊参数代码
     * @param responsibilityCode 责任中心代码
     * @param hospId             医院id
     * @return {@link PageUtils} 分页对象
     */
    @Override
    public PageUtils selectList(Integer current, Integer pageSize, String date, String shareParamCode, String responsibilityCode, Long hospId) {
        Integer startIndex = (current - 1) * pageSize;
        if (startIndex < 0) {
            startIndex = 0;
        }
        List<ShareParamValueVO> list = baseMapper.getValues(startIndex, pageSize, date, shareParamCode, responsibilityCode, hospId);
        int count = baseMapper.getValuesCount(startIndex, pageSize, date, shareParamCode, responsibilityCode, hospId);
        return new PageUtils(list, count, pageSize, current);
    }

    /**
     * 复制数据从某年月到某年月
     *
     * @param copyShareParamValueDTO {@link CopyShareParamValueDTO}
     * @param hospId                 医院id
     */
    @Override
    @Transactional(propagation = Propagation.REQUIRED, rollbackFor = Throwable.class)
    public void copyData(CopyShareParamValueDTO copyShareParamValueDTO, Long hospId) {
        // TODO 复制数据 多个医院同时使用的情况下启用线程池处理
        String fromDate = copyShareParamValueDTO.getFromDate();
        String toDate = copyShareParamValueDTO.getToDate();
        Integer dataSourceType = copyShareParamValueDTO.getDataSourceType();

        LambdaQueryWrapper<ShareParamValue> wrapper = new LambdaQueryWrapper<ShareParamValue>()
                .eq(ShareParamValue::getDateYear, DateUtil.year(DateUtil.parse(fromDate, PURE_DATE_PATTERN)))
                .eq(ShareParamValue::getDateMonth, DateUtil.month(DateUtil.parse(fromDate, PURE_DATE_PATTERN)) + 1);
        if (dataSourceType == 0) {
            wrapper.eq(ShareParamValue::getDataSourceType, 0);
        } else if (dataSourceType == 1) {
            wrapper.eq(ShareParamValue::getDataSourceType, 1);
        } else {
            wrapper.in(ShareParamValue::getDataSourceType, CollUtil.newArrayList(0, 1));
        }
        // 拿到复制的数据
        List<ShareParamValue> list = this.list(wrapper);
        if (list.isEmpty()) {
            throw new CostException("复制日期" + fromDate + "数据不存在");
        }
        DateTime toDateTime = DateUtil.parse(toDate, PURE_DATE_PATTERN);
        int year = DateUtil.year(toDateTime);
        int month = DateUtil.month(toDateTime) + 1;
        // 删除之前的数据
        wrapper.clear();
        this.remove(
                wrapper.eq(ShareParamValue::getDateYear, year)
                        .eq(ShareParamValue::getDateMonth, month)
                        .eq(ShareParamValue::getHospId, hospId)
        );
        list.forEach(i -> {
            i.setId(null);
            i.setCreateTime(System.currentTimeMillis());
            i.setHospId(hospId);
            i.setDateYear(year);
            i.setDateMonth(month);
        });
        this.saveBatch(list);
    }

    /**
     * 编辑某条数据
     *
     * @param shareParamValueEditDTO {@link ShareParamValueEditDTO}
     * @param hospId                 医院id
     */
    @Override
    @Transactional(rollbackFor = Throwable.class, propagation = Propagation.REQUIRED)
    public void editParamValue(ShareParamValueEditDTO shareParamValueEditDTO, Long hospId) {
        Long id = shareParamValueEditDTO.getId();
        this.removeById(id);

        DateTime dateTime = DateUtil.parse(shareParamValueEditDTO.getDate(), PURE_DATE_PATTERN);

        Boolean flag = checkUniq(shareParamValueEditDTO.getShareParamCode(), shareParamValueEditDTO.getResponsibilityCode(), hospId, dateTime);
        if (flag) {
            throw new CostException("本月已存在对应责任中心和分摊参数中心，请直接修改");
        }
        checkCodeIsExist(shareParamValueEditDTO.getShareParamCode(), shareParamValueEditDTO.getResponsibilityCode(), hospId);

        ShareParamValue shareParamValue = BeanUtil.convertObj(shareParamValueEditDTO, ShareParamValue.class);
        shareParamValue
                .setId(null)
                .setCreateTime(System.currentTimeMillis())
                .setDateYear(DateUtil.year(DateUtil.parse(shareParamValueEditDTO.getDate(), PURE_DATE_PATTERN)))
                .setDateMonth(DateUtil.month(DateUtil.parse(shareParamValueEditDTO.getDate(), PURE_DATE_PATTERN)) + 1)
                .setHospId(hospId)
                .setDataSourceType(0);

        this.save(shareParamValue);
    }

    /**
     * 成本分摊参数导入
     *
     * @param list
     * @param user
     * @param file
     * @param dateTime
     * @param fileType
     * @return
     */
    @Override
    @Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
    public Result importShareParamGroup(List<List<Object>> list, User user, MultipartFile file, String dateTime, Integer fileType) {
        // 先检验当前年月是否存在数据
        int year = 0;
        int month = 0;
        Date date = DateUtils.StringToDate(dateTime, DateStyleEnum.YYYY_MM);
        if (StrUtil.isNotBlank(dateTime)) {
            year = DateUtil.year(date);
            month = DateUtil.month(date) + 1;
        }
        Long hospId = user.getHospId();
        List<ShareParamValue> groups = this.list(new QueryWrapper<ShareParamValue>().lambda().eq(ShareParamValue::getHospId, hospId)
                .eq(ShareParamValue::getDateYear, year).eq(ShareParamValue::getDateMonth, month));
        if (!CollectionUtils.isEmpty(groups)) {
            throw new CostException(500, year + "年" + month + "月数据已存在");
        }
        for (int i = list.size() - 1; i >= 0; i--) {
            if (i == NumberConstant.ZERO || i == NumberConstant.ONE || i == NumberConstant.TWO || i == NumberConstant.THREE) {
                list.remove(list.get(i));
            }
        }
        List<IncomeErrorMessage> incomeErrorMessageList = new ArrayList<>();
        Map<String, Department> departmentMap = costIncomeGroupService.getDepartmentByCodeNameMap(hospId);
        Map<Long, Responsibility> responsibilityMap = costIncomeGroupService.getResponsibilityIdResponsibilityMap(hospId);
        List<ResponsibilityDepartment> responsibilityDepartmentList = costIncomeGroupService.getResponsibilityDepartments(hospId);
        if (CollectionUtils.isEmpty(responsibilityDepartmentList)) {
            throw new CostException(500, "没有科室责任中心对照数据");
        }
        Map<Long, Long> responsibilityDepMap = costIncomeGroupService.getDepartmentIdResponsibilityIdMap(responsibilityDepartmentList);
        Map<String, CostShareParam> shareParamMap = costShareParamService.list(new QueryWrapper<CostShareParam>().lambda().eq(CostShareParam::getHospId, hospId)).stream().collect(Collectors.toMap(k -> k.getShareParamCode() + k.getShareParamName(), synOe -> synOe));
        // 要保存的数据
        List<ShareParamValue> shareParamValues = new LinkedList<>();
//        List<CostShareParamGroup> costShareParamGroupList = new LinkedList<>();
        List<Object> departmentCodes = list.get(0);
        List<Object> departmentNames = list.get(1);

        //检验数据是否准确  list包含了科室Code行与Name行 所以要从list的第三行开始
        checkShareParamGroupData(list, year, month, incomeErrorMessageList, departmentMap, responsibilityMap, responsibilityDepMap, shareParamMap, shareParamValues, departmentCodes, departmentNames);
        // 文件上传
        String uploadFile = costIncomeGroupService.uploadFile(file, UserContext.getCurrentUser());
        // 上传记录保存
        if (StrUtil.isBlank(uploadFile)) {
            throw new CostException(500, "文件上传异常");
        }
        // 记录文件上传记录
        CostIncomeFile costIncomeFile = costIncomeFileService.saveCostIncomeFile(list, user, file, hospId, incomeErrorMessageList, uploadFile, fileType, year, month);
        Long id = costIncomeFile.getId();
        shareParamValues.forEach(i -> {
            // 设置文件Id
            i.setFileId(id);
            i.setDataSourceType(1);
        });
        // 保存数据的唯一性
        Map<String, List<ShareParamValue>> map = shareParamValues.stream().collect(Collectors.groupingBy(k -> k.getShareParamCode().trim() + "code" + k.getResponsibilityCode().trim() + k.getDateYear().toString().trim() + "code" + k.getDateMonth().toString().trim()
        ));
        LinkedList<ShareParamValue> realList = new LinkedList<>();
        Set<String> strings = map.keySet();
        for (String str : strings) {
            List<ShareParamValue> shareParamValuesList = map.get(str);
            if (CollUtil.isNotEmpty(shareParamValuesList)) {
                BigDecimal reduce = shareParamValuesList.stream().map(ShareParamValue::getValueNum).reduce(BigDecimal.ZERO, BigDecimal::add);
                ShareParamValue shareParamValue = shareParamValuesList.get(0);
                shareParamValue.setValueNum(reduce);
                realList.add(shareParamValue);
            }
        }
        if (CollectionUtils.isEmpty(incomeErrorMessageList)) {
            this.saveBatch(realList);
            return Result.build(200, "数据导入成功", null);
        } else {
            return Result.build(200, "数据未成功导入", null);
        }
    }

    private void checkShareParamGroupData(List<List<Object>> list, int year, int month,
                                          List<IncomeErrorMessage> incomeErrorMessageList, Map<String, Department> departmentMap,
                                          Map<Long, Responsibility> responsibilityMap, Map<Long, Long> responsibilityDepMap,
                                          Map<String, CostShareParam> shareParamMap, List<ShareParamValue> shareParamValues,
                                          List<Object> departmentCodes, List<Object> departmentNames) {
        for (int i = 2; i < list.size(); i++) {
            int row = i + 4;
            List<Object> data = list.get(i);
            // 检验成本分摊参数是否正确
            ShareParamValue shareParamValueRequest = new ShareParamValue();
            String shareParamCode = data.get(0).toString().trim();
            String shareParamName = data.get(1).toString().trim();
            CostShareParam costShareParam = shareParamMap.get(shareParamCode + shareParamName);
            if (Objects.nonNull(costShareParam)) {
                shareParamValueRequest.setShareParamCode(shareParamCode);
            } else {
                IncomeErrorMessage incomeErrorMessage = new IncomeErrorMessage();
                incomeErrorMessage.setTotal(i);
                incomeErrorMessage.setErrMessage("成本分摊代码:" + shareParamCode + " 名称" + shareParamName + "不存在");
                incomeErrorMessageList.add(incomeErrorMessage);
            }
            for (int j = 2; j < departmentCodes.size(); j++) {
                int column = j + 1;
                if (data.size() > j) {
                    if (Objects.isNull(data.get(j))) {
                        data.set(j, NumberConstant.ZERO);
                    } else if (data.get(j).toString().contains("-") || !StringUtils.isNumber(data.get(j).toString())) {
                        IncomeErrorMessage incomeErrorMessage = new IncomeErrorMessage();
                        incomeErrorMessage.setTotal(row);
                        incomeErrorMessage.setErrMessage("第" + row + "行 第" + column + "列数据不符合规范");
                        incomeErrorMessageList.add(incomeErrorMessage);
                    } else {
                        data.set(j, Double.parseDouble(data.get(j).toString()));
                    }
                } else {
                    data.add(NumberConstant.ZERO);
                }
            }
            if (CollectionUtils.isEmpty(incomeErrorMessageList)) {
                incomeErrorMessageList = new ArrayList<>();
                // 检验数据是否全为0
                int zeroStatus = 0;
                for (int j = 2; j < data.size(); j++) {
                    if (!NumberConstant.ZERO.equals(data.get(j))) {
                        zeroStatus = 1;
                        break;
                    }
                }
                if (NumberConstant.ONE.equals(zeroStatus)) {
                    for (int j = 2; j < data.size(); j++) {
                        ShareParamValue shareParamValue = BeanUtil.convertObj(shareParamValueRequest, ShareParamValue.class);
                        // 检验科室信息是否准确
                        String departmentCode = departmentCodes.get(j).toString();
                        String departmentName = departmentNames.get(j).toString();
                        Department department = departmentMap.get(departmentCode + departmentName);
                        if (Objects.nonNull(department)) {
                            // 检测责任中心是否存在
                            Long id = department.getId();
                            Long responsibilityId = responsibilityDepMap.get(id);
                            if (Objects.nonNull(responsibilityId)) {
                                Responsibility responsibility = responsibilityMap.get(responsibilityId);
                                if (Objects.nonNull(responsibility)) {
                                    shareParamValue.setResponsibilityCode(responsibility.getResponsibilityCode());

                                } else {
                                    IncomeErrorMessage incomeErrorMessage = new IncomeErrorMessage();
                                    incomeErrorMessage.setTotal(j);
                                    incomeErrorMessage.setErrMessage("第" + j + "列科室信息对应的责任中心不存在");
                                    incomeErrorMessageList.add(incomeErrorMessage);
                                }
                            } else {
                                IncomeErrorMessage incomeErrorMessage = new IncomeErrorMessage();
                                incomeErrorMessage.setTotal(j);
                                incomeErrorMessage.setErrMessage("第" + j + "列科室信息不存在对应的责任中心");
                                incomeErrorMessageList.add(incomeErrorMessage);
                            }
                        } else {
                            IncomeErrorMessage incomeErrorMessage = new IncomeErrorMessage();
                            incomeErrorMessage.setTotal(j);
                            incomeErrorMessage.setErrMessage("第" + j + "列科室信息不存在");
                            incomeErrorMessageList.add(incomeErrorMessage);
                        }
                        shareParamValue.setValueNum(BigDecimal.valueOf(Double.parseDouble(("0".equals(data.get(j).toString()) || StrUtil.isBlank(data.get(j).toString())) ? "0.00" : data.get(j).toString())));
                        shareParamValue.setHospId(UserContext.getHospId());
                        shareParamValue.setCreateTime(System.currentTimeMillis());
                        shareParamValue.setDateYear(year);
                        shareParamValue.setDateMonth(month);
                        shareParamValues.add(shareParamValue);
                    }
                }

            }

        }
    }


    /**
     * 计算数值
     *
     * @param date   日期 yyyyMM
     * @param hospId 医院id
     */
    @Override
    @Transactional(propagation = Propagation.REQUIRED, rollbackFor = Throwable.class)
    public void calcData(String date, Long hospId) {
        DateTime parse = DateUtil.parse(date, PURE_DATE_PATTERN);
        int year = DateUtil.year(parse);
        int month = DateUtil.month(parse) + 1;
        // 拿到这些年月的数据
        List<ShareParamValue> list = this.list(
                new LambdaQueryWrapper<ShareParamValue>()
                        .eq(ShareParamValue::getDateYear, year).eq(ShareParamValue::getDateMonth, month).eq(ShareParamValue::getHospId, hospId)
        );
        if (list.isEmpty()){
            throw new CostException(500,parse+"数据未录入");
        }
        // 得到这些责任中心代码
        List<String> responsibilityCodes = list.stream().map(ShareParamValue::getResponsibilityCode).distinct().collect(Collectors.toList());
        List<IncomeCollection> incomeCollections = incomeCollectionService.list(
                new LambdaQueryWrapper<IncomeCollection>()
                        .eq(IncomeCollection::getYear, year)
                        .eq(IncomeCollection::getMonth, month)
                        .eq(IncomeCollection::getHospId, hospId)
                        .in(IncomeCollection::getResponsibilityCode, responsibilityCodes)
        );
        if (incomeCollections.isEmpty()) {
            throw new CostException("本月未归集数据，请先归集数据");
        }

        Map<String, List<IncomeCollection>> collectMap = incomeCollections.stream().collect(Collectors.groupingBy(IncomeCollection::getResponsibilityCode));
        Set<String> strings = collectMap.keySet();
        Map<String,BigDecimal> map = new ConcurrentHashMap<>();
        for (String str : strings) {
            List<IncomeCollection> collections = collectMap.get(str);
            if (CollUtil.isNotEmpty(collections)) {
                BigDecimal reduce = incomeCollections.stream().map(IncomeCollection::getAmount).reduce(BigDecimal.ZERO, BigDecimal::add);
                IncomeCollection incomeCollection = incomeCollections.get(0);
                incomeCollection.setAmount(reduce);
                map.put(str, reduce);
            }
        }
        list.forEach(i -> {
            // TODO: 2021/8/19
            BigDecimal bigDecimal = map.get(i.getResponsibilityCode());
            if (Objects.nonNull(bigDecimal)) {
                i.setValueNum(bigDecimal);
            }
        });
        this.updateBatchById(list);
    }
}