blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
xiangpei
2024-09-30 a4d54f771794a8ea36319c679d489180b595edfd 
 ycl-server/src/main/java/com/ycl/platform/service/impl/CalculateReportServiceImpl.java


@@ -1,10 +1,15 @@


package com.ycl.platform.service.impl;




import com.alibaba.excel.EasyExcel;


import com.alibaba.excel.ExcelWriter;


import com.alibaba.excel.write.metadata.WriteSheet;


import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;


import com.baomidou.mybatisplus.core.metadata.IPage;


import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;


import com.ycl.platform.domain.entity.CalculateRecord;


import com.ycl.platform.domain.entity.CalculateReport;


import com.ycl.platform.domain.entity.CalculateRule;


import com.ycl.platform.domain.excel.CalculateExport;


import com.ycl.platform.domain.form.CalculateReportBackfillForm;


import com.ycl.platform.domain.form.CalculateReportForm;


@@ -14,8 +19,10 @@


import com.ycl.platform.mapper.CalculateRecordMapper;


import com.ycl.platform.mapper.CalculateReportMapper;


import com.ycl.platform.service.CalculateReportService;


import com.ycl.platform.service.ICalculateRuleService;


import com.ycl.system.Result;


import com.ycl.system.page.PageUtil;


import com.ycl.utils.SecurityUtils;


import enumeration.general.CalculateReportStatusEnum;


import jakarta.servlet.http.HttpServletResponse;


import lombok.RequiredArgsConstructor;


@@ -29,6 +36,7 @@


import java.math.BigDecimal;


import java.net.URLEncoder;


import java.nio.charset.StandardCharsets;


import java.util.Date;


import java.util.List;


import java.util.Objects;


import java.util.stream.Collectors;


@@ -43,7 +51,7 @@


@RequiredArgsConstructor


public class CalculateReportServiceImpl extends ServiceImpl<CalculateReportMapper, CalculateReport> implements CalculateReportService {




    private final CalculateReportMapper calculateReportMapper;


    private final ICalculateRuleService calculateRuleService;


    private final CalculateRecordMapper calculateRecordMapper;




    /**


@@ -54,6 +62,9 @@


    @Override


    public Result add(CalculateReportForm form) {


        CalculateReport entity = CalculateReportForm.getEntityByForm(form, null);


        Date now = new Date();


        entity.setCreateTime(now);


        entity.setUpdateTime(now);


        if(baseMapper.insert(entity) > 0) {


            return Result.ok("添加成功");


        }


@@ -69,10 +80,11 @@


    public Result update(CalculateReportForm form) {




        CalculateReport entity = baseMapper.selectById(form.getId());




        // 为空抛IllegalArgumentException，做全局异常处理


        Assert.notNull(entity, "记录不存在");


        BeanUtils.copyProperties(form, entity);


        Date now = new Date();


        entity.setUpdateTime(now);


        if (baseMapper.updateById(entity) > 0) {


            return Result.ok("修改成功");


        }


@@ -113,6 +125,7 @@


    @Override


    public Result page(CalculateReportQuery query) {


        IPage<CalculateReportVO> page = PageUtil.getPage(query, CalculateReportVO.class);


        query.setUnitId(SecurityUtils.getUnitId());


        baseMapper.page(query, page);


        page.getRecords().stream().forEach(item -> {


            if (Objects.isNull(item)) {


@@ -199,12 +212,26 @@


    @Override


    @SneakyThrows


    public void export(Integer contractId, HttpServletResponse response) {


        // 获取数据


        List<CalculateExport> list = baseMapper.exportData(contractId);


        CalculateExport calculateExport = new CalculateExport();


        calculateExport.setRuleName("合计");


        calculateExport.setNum(list.stream().mapToInt(CalculateExport::getNum).sum());


        calculateExport.setScore(100 + list.stream().mapToInt(CalculateExport::getScore).sum());


        list.add(calculateExport);


        // 获取规则


        List<CalculateRule> ruleList = calculateRuleService.list(new LambdaQueryWrapper<CalculateRule>().eq(CalculateRule::getContractId, contractId));


        // 输出文件


        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");


        response.setCharacterEncoding("utf-8");


        String fileName = URLEncoder.encode("核算报告", StandardCharsets.UTF_8).replace("\\+", "%20");


        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");


        EasyExcel.write(response.getOutputStream(), CalculateExport.class)


                .sheet("核算报告")


                .doWrite(baseMapper.exportData(contractId));


        // 增加sheet


        try (ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()).build()){


            WriteSheet sheet = EasyExcel.writerSheet(0, "核算报告").head(CalculateExport.class).build();


            excelWriter.write(list, sheet);


            WriteSheet sheet2 = EasyExcel.writerSheet(1, "核算规则").head(CalculateRule.class).registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).build();


            excelWriter.write(ruleList, sheet2);


        }


    }


}