weaver-hrm-salary/src/com/engine/salary/biz/AddUpDeductionBiz.java

package com.engine.salary.biz;

import com.engine.salary.entity.datacollection.AddUpDeduction;
import com.engine.salary.entity.datacollection.dto.AddUpDeductionDTO;
import com.engine.salary.entity.datacollection.dto.AddUpDeductionRecordDTO;
import com.engine.salary.entity.datacollection.dto.AddUpDeductionRecordStrDTO;
import com.engine.salary.entity.datacollection.dto.AddUpDeductionStrDTO;
import com.engine.salary.entity.datacollection.param.AddUpDeductionQueryParam;
import com.engine.salary.mapper.datacollection.AddUpDeductionMapper;
import com.engine.salary.util.excel.ExcelUtil;
import com.google.common.collect.Lists;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.ibatis.session.SqlSession;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import weaver.conn.mybatis.MyBatisFactory;
import weaver.general.BaseBean;
import weaver.general.Util;

import java.text.SimpleDateFormat;
import java.util.*;
import java.util.stream.Collectors;

public class AddUpDeductionBiz extends BaseBean {


    /**
     * 关联查询查询列表
     *
     * @param param
     * @return
     */
    public List<AddUpDeductionDTO> list(AddUpDeductionQueryParam param) {
        SqlSession sqlSession = MyBatisFactory.sqlSessionFactory.openSession();
        try {
            AddUpDeductionMapper mapper = sqlSession.getMapper(AddUpDeductionMapper.class);
            List<AddUpDeductionStrDTO> list = mapper.list(param);
            return AddUpDeductionStrDTOEncrypt.decryptAddUpDeductionList(list);
        } finally {
            sqlSession.close();
        }
    }

    /**
     * 条件查询
     *
     * @param param
     * @return
     */
    public List<AddUpDeduction> listSome(AddUpDeduction param) {
        SqlSession sqlSession = MyBatisFactory.sqlSessionFactory.openSession();
        try {
            AddUpDeductionMapper mapper = sqlSession.getMapper(AddUpDeductionMapper.class);
            List<AddUpDeduction> addUpDeductions = mapper.listSome(param);
            return AddUpDeductionEncrypt.decryptAddUpDeductionList(addUpDeductions);
        } finally {
            sqlSession.close();
        }
    }


    /**
     * 根据id获取
     *
     * @param id
     * @return
     */
    public AddUpDeduction getById(Long id) {
        SqlSession sqlSession = MyBatisFactory.sqlSessionFactory.openSession();
        try {
            AddUpDeductionMapper mapper = sqlSession.getMapper(AddUpDeductionMapper.class);
            AddUpDeduction byId = mapper.getById(id);
            return AddUpDeductionEncrypt.decryptAddUpDeduction(byId);
        } finally {
            sqlSession.close();
        }
    }

    /**
     * 详情列表
     *
     * @param param
     * @return
     */
    public List<AddUpDeductionRecordDTO> recordList(AddUpDeductionQueryParam param) {
        SqlSession sqlSession = MyBatisFactory.sqlSessionFactory.openSession();
        try {
            AddUpDeductionMapper mapper = sqlSession.getMapper(AddUpDeductionMapper.class);
            List<AddUpDeductionRecordStrDTO> addUpDeductionRecordStrDTOS = mapper.recordList(param);
            return AddUpDeductionRecordStrDTOEncrypt.decryptAddUpDeductionRecordDTO(addUpDeductionRecordStrDTOS);
        } finally {
            sqlSession.close();
        }
    }

    /**
     * 批量插入
     *
     * @param param
     * @return
     */
    public void batchSave(List<AddUpDeduction> param) {
        if (CollectionUtils.isEmpty(param)) {
            return;
        }
        SqlSession sqlSession = MyBatisFactory.sqlSessionFactory.openSession();
        try {
            AddUpDeductionMapper mapper = sqlSession.getMapper(AddUpDeductionMapper.class);
            AddUpDeductionEncrypt.encryptAddUpDeductionList(param);
            List<List<AddUpDeduction>> partition = Lists.partition(param, 100);
            partition.forEach(mapper::insertData);
            sqlSession.commit();
        } finally {
            sqlSession.close();
        }
    }

    /**
     * 批量插入
     *
     * @param param
     * @return
     */
    public void batchUpdate(List<AddUpDeduction> param) {
        if (CollectionUtils.isEmpty(param)) {
            return;
        }
        SqlSession sqlSession = MyBatisFactory.sqlSessionFactory.openSession();
        try {
            AddUpDeductionMapper mapper = sqlSession.getMapper(AddUpDeductionMapper.class);
            AddUpDeductionEncrypt.encryptAddUpDeductionList(param);
            List<List<AddUpDeduction>> partition = Lists.partition(param, 100);
            partition.forEach(mapper::updateData);
            sqlSession.commit();
        } finally {
            sqlSession.close();
        }
    }


    /**
     * 导出
     *
     * @param param
     * @return
     */
    public XSSFWorkbook export(AddUpDeductionQueryParam param) {

        //获取操作按钮资源
        List<List<String>> rowList = getExcelRowList(param);

        //获取excel
        return ExcelUtil.genWorkbook(rowList, "累计专项附加扣除");
    }


    /**
     * 获取excel数据行
     *
     * @return 导出数据行集合
     */
    private List<List<String>> getExcelRowList(AddUpDeductionQueryParam param) {
        //excel标题
        List<String> title = Arrays.asList("姓名", "个税扣缴义务人", "部门", "手机号", "工号", "证件号码", "入职日期", "累计子女教育", "累计继续教育", "累计住房贷款利息", "累计住房租金", "累计赡养老人", "累计大病医疗", "累计婴幼儿照护");

        SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd");
        List<AddUpDeductionDTO> list = list(param);
        final List<List<String>> dataRowList = Optional.ofNullable(list)
                .map(List::stream)
                .map(operatorStream -> operatorStream.map(dto -> {
                    List<String> cellList = new ArrayList<>();
                    cellList.add(Util.null2String(dto.getUsername()));
                    cellList.add(Util.null2String(dto.getTaxAgentName()));
                    cellList.add(Util.null2String(dto.getDepartmentName()));
                    cellList.add(Util.null2String(dto.getMobile()));
                    cellList.add(Util.null2String(dto.getJobNum()));
                    cellList.add(Util.null2String(dto.getIdNo()));
                    cellList.add(Util.null2String(dto.getHiredate()));
                    cellList.add(Util.null2String(dto.getAddUpChildEducation()));
                    cellList.add(Util.null2String(dto.getAddUpContinuingEducation()));
                    cellList.add(Util.null2String(dto.getAddUpHousingLoanInterest()));
                    cellList.add(Util.null2String(dto.getAddUpHousingRent()));
                    cellList.add(Util.null2String(dto.getAddUpSupportElderly()));
                    cellList.add(Util.null2String(dto.getAddUpIllnessMedical()));
                    cellList.add(Util.null2String(dto.getAddUpInfantCare()));
                    return cellList;
                }).collect(Collectors.toList()))
                .orElse(Collections.emptyList());

        List<List<String>> rowList = new ArrayList<>();
        rowList.add(title);
        rowList.addAll(dataRowList);
        return rowList;
    }


    /**
     * 导出详情列表
     *
     * @param param
     * @return
     */
    public XSSFWorkbook exportDetail(AddUpDeductionQueryParam param) {

        //获取操作按钮资源
        List<List<String>> rowList = getExcelRowDetailList(param);

        //获取excel
        return ExcelUtil.genWorkbook(rowList, "累计专项附加扣除明细");
    }


    /**
     * 导出详情
     *
     * @param param
     * @return
     */
    private List<List<String>> getExcelRowDetailList(AddUpDeductionQueryParam param) {

        //excel标题
        List<String> title = Arrays.asList("姓名", "申报月份", "个税扣缴义务人", "部门", "工号", "累计子女教育", "累计继续教育", "累计住房贷款利息", "累计住房租金", "累计赡养老人", "累计大病医疗", "累计婴幼儿照护");


        SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM");
        //查询详细信息
        List<AddUpDeductionRecordDTO> list = recordList(param);
        final List<List<String>> dataRowList = Optional.ofNullable(list)
                .map(List::stream)
                .map(operatorStream -> operatorStream.map(dto -> {
                    List<String> cellList = new ArrayList<>();
                    cellList.add(Util.null2String(dto.getUsername()));
                    cellList.add(Util.null2String(dto.getDeclareMonth() == null ? "" : formatter.format(dto.getDeclareMonth())));
                    cellList.add(Util.null2String(dto.getTaxAgentName()));
                    cellList.add(Util.null2String(dto.getDepartmentName()));
                    cellList.add(Util.null2String(dto.getJobNum()));
                    cellList.add(Util.null2String(dto.getAddUpChildEducation()));
                    cellList.add(Util.null2String(dto.getAddUpContinuingEducation()));
                    cellList.add(Util.null2String(dto.getAddUpHousingLoanInterest()));
                    cellList.add(Util.null2String(dto.getAddUpHousingRent()));
                    cellList.add(Util.null2String(dto.getAddUpSupportElderly()));
                    cellList.add(Util.null2String(dto.getAddUpIllnessMedical()));
                    cellList.add(Util.null2String(dto.getAddUpInfantCare()));
                    return cellList;
                }).collect(Collectors.toList()))
                .orElse(Collections.emptyList());

        List<List<String>> rowList = new ArrayList<>();
        rowList.add(title);
        rowList.addAll(dataRowList);
        return rowList;
    }

    /**
     * 处理导入数据
     *
     * @param pos
     */
    public void handleImportData(List<AddUpDeduction> pos) {
        if (CollectionUtils.isEmpty(pos)) {
            return;
        }
        AddUpDeduction po = pos.get(0);
        // 多条相同人的则以第一条为准，如果逆序排列（用于重复的则以最后一条为准）Collections.reverse(pos);
        // 去重（通过记录的唯一条件(申报月份，人员id，个税扣缴义务人id)拼接）
        List<AddUpDeduction> finalPos = pos.stream().collect(Collectors.collectingAndThen(Collectors.toCollection(() -> new TreeSet<>(Comparator.comparing(f -> f.getEmployeeId() + "-" + f.getTaxAgentId()))), ArrayList::new));
        // 查询已有数据
        List<AddUpDeduction> list = listSome(AddUpDeduction.builder().declareMonth(po.getDeclareMonth()).build());
        // 待修改的 本地已存在则更新【交集】
        List<AddUpDeduction> updateList = list.stream().map(m -> {
            Optional<AddUpDeduction> optional = finalPos.stream().filter(p -> (p.getEmployeeId() + "-" + p.getTaxAgentId()).equals(m.getEmployeeId() + "-" + m.getTaxAgentId())).findFirst();
            AddUpDeduction temp = null;
            if (optional.isPresent()) {
                temp = optional.get();
                // 换成本地库的id
                temp.setId(m.getId());
            }
            return temp;
        }).filter(Objects::nonNull).collect(Collectors.toList());
        // 待新增的 导入比本地多，则新增【差集(导入 - local)】
        List<AddUpDeduction> saveList = finalPos.stream().map(m -> {
            Optional<AddUpDeduction> optional = list.stream().filter(p -> (p.getEmployeeId() + "-" + p.getTaxAgentId()).equals(m.getEmployeeId() + "-" + m.getTaxAgentId())).findFirst();
            AddUpDeduction temp = null;
            if (!optional.isPresent()) {
                temp = m;
            }
            return temp;
        }).filter(Objects::nonNull).collect(Collectors.toList());

        // 修改
        if (CollectionUtils.isNotEmpty(updateList)) {
            batchUpdate(updateList);
        }
        // 保存
        if (CollectionUtils.isNotEmpty(saveList)) {
            batchSave(saveList);
        }
        // 记录操作日志
//        saveList.addAll(updateList);
//
//        if (CollectionUtils.isNotEmpty(saveList)) {
//            LoggerContext loggerContext = new LoggerContext();
//            loggerContext.setTargetId(String.valueOf(IdGenerator.generate()));
//            loggerContext.setTargetName(SalaryI18nUtil.getI18nLabel(message.getTenantKey(), message.getUserId(), 100351, "导入累计专项附加扣除"));
//            loggerContext.setOperateType(OperateTypeEnum.ADD.getValue());
//            loggerContext.setOperateTypeName(SalaryI18nUtil.getI18nLabel(message.getTenantKey(), message.getUserId(), 100351, "导入累计专项附加扣除"));
//            loggerContext.setOperatedesc(SalaryI18nUtil.getI18nLabel(message.getTenantKey(), message.getUserId(), 100351, "导入累计专项附加扣除"));
//            loggerContext.setNewValueList(saveList);
//            loggerContext.setTenant_key(message.getTenantKey());
//            loggerContext.setOperator(message.getUserId().toString());
//            loggerContext.setOperatorName(message.getOpreator());
//            loggerContext.setClientIp(message.getClientIp());
//            addUpDeductionLoggerTemplate.write(loggerContext);
//        }
    }


}