weaver-hrm-salary/src/com/engine/salary/web/SIExportController.java

package com.engine.salary.web;

import com.engine.common.util.ServiceUtil;
import com.engine.salary.biz.SIAccountBiz;
import com.engine.salary.entity.siaccount.param.InspectAccountParam;
import com.engine.salary.entity.siaccount.po.InsuranceAccountInspectPO;
import com.engine.salary.entity.siarchives.param.InsuranceArchivesListParam;
import com.engine.salary.entity.siexport.param.InsuranceExportParam;
import com.engine.salary.enums.siaccount.PaymentStatusEnum;
import com.engine.salary.service.SIAccountService;
import com.engine.salary.service.impl.SIAccountServiceImpl;
import com.engine.salary.wrapper.SIExportWrapper;
import io.swagger.v3.oas.annotations.parameters.RequestBody;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import weaver.hrm.HrmUserVarify;
import weaver.hrm.User;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.StreamingOutput;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.time.LocalDate;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @Author weaver_cl
 * @Description: TODO
 * @Date 2022/4/18
 * @Version V1.0
 **/
public class SIExportController {

    public SIAccountService getService(User user) {
        return ServiceUtil.getService(SIAccountServiceImpl.class, user);
    }

    public SIExportWrapper getSIExportWrapper(User user) {
        return ServiceUtil.getService(SIExportWrapper.class,user);
    }

    @POST
    @Path("/archives/export")
    @Produces(MediaType.APPLICATION_OCTET_STREAM)
    public Response export(@Context HttpServletRequest request, @Context HttpServletResponse response,
                                  @RequestBody InsuranceArchivesListParam param) {
        User user = HrmUserVarify.getUser(request, response);
        XSSFWorkbook workbook = getSIExportWrapper(user).export(param);
        String time = LocalDate.now().toString();
        String fileName = "福利档案报表" + time;
        try {
            fileName = URLEncoder.encode(fileName + ".xlsx", "UTF-8");
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        StreamingOutput output = outputStream -> {
            workbook.write(outputStream);
            outputStream.flush();
        };
        response.setContentType("application/octet-stream");
        return Response.ok(output).header("Content-disposition", "attachment;filename=" + fileName).header("Cache-Control", "no-cache").build();
    }

    @POST
    @Path("/archives/exportInspect")
    @Produces(MediaType.APPLICATION_OCTET_STREAM)
    public Response exportInspect(@Context HttpServletRequest request, @Context HttpServletResponse response,
                           @RequestBody InspectAccountParam param) {
        User user = HrmUserVarify.getUser(request, response);
        SIAccountBiz siAccountBiz = new SIAccountBiz();
        List<InsuranceAccountInspectPO> insuranceAccountInspectPOS = siAccountBiz.allInspects(param.getIds(), param.getBillMonth());
        InsuranceArchivesListParam req = new InsuranceArchivesListParam();
        req.setEmployeeIds(insuranceAccountInspectPOS.stream().map(InsuranceAccountInspectPO::getEmployeeId).distinct().collect(Collectors.toList()));
        XSSFWorkbook workbook = getSIExportWrapper(user).export(req);
        String time = LocalDate.now().toString();
        String fileName = "核算异常人员档案报表" + time;
        try {
            fileName = URLEncoder.encode(fileName + ".xlsx", "UTF-8");
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        StreamingOutput output = outputStream -> {
            workbook.write(outputStream);
            outputStream.flush();
        };
        response.setContentType("application/octet-stream");
        return Response.ok(output).header("Content-disposition", "attachment;filename=" + fileName).header("Cache-Control", "no-cache").build();
    }




    @POST
    @Path("/common/export")
    @Produces(MediaType.APPLICATION_OCTET_STREAM)
    public Response exportAccount(@Context HttpServletRequest request, @Context HttpServletResponse response,
                                        @RequestBody InsuranceExportParam param) {
        User user = HrmUserVarify.getUser(request, response);
        XSSFWorkbook workbook = getSIExportWrapper(user).exportAccount(PaymentStatusEnum.COMMON.getValue(),param);
        String time = LocalDate.now().toString();
        String fileName = "正常缴纳核算报表" + time;
        try {
            fileName = URLEncoder.encode(fileName + ".xlsx", "UTF-8");
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        StreamingOutput output = outputStream -> {
            workbook.write(outputStream);
            outputStream.flush();
        };
        response.setContentType("application/octet-stream");
        return Response.ok(output).header("Content-disposition", "attachment;filename=" + fileName).header("Cache-Control", "no-cache").build();
    }


    @POST
    @Path("/supplementary/export")
    @Produces(MediaType.APPLICATION_OCTET_STREAM)
    public Response exportSupplementary(@Context HttpServletRequest request, @Context HttpServletResponse response,
                                   @RequestBody InsuranceExportParam param) {
        User user = HrmUserVarify.getUser(request, response);
        XSSFWorkbook workbook = getSIExportWrapper(user).exportAccount(PaymentStatusEnum.REPAIR.getValue(),param);
        String time = LocalDate.now().toString();
        String fileName = "补缴核算报表" + time;
        try {
            fileName = URLEncoder.encode(fileName + ".xlsx", "UTF-8");
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        StreamingOutput output = outputStream -> {
            workbook.write(outputStream);
            outputStream.flush();
        };
        response.setContentType("application/octet-stream");
        return Response.ok(output).header("Content-disposition", "attachment;filename=" + fileName).header("Cache-Control", "no-cache").build();
    }

    @POST
    @Path("/overView/export")
    @Produces(MediaType.APPLICATION_OCTET_STREAM)
    public Response exportOverView(@Context HttpServletRequest request, @Context HttpServletResponse response,
                                   @RequestBody InsuranceExportParam param) {
        User user = HrmUserVarify.getUser(request, response);
        XSSFWorkbook workbook = getSIExportWrapper(user).exportOverView(param);
        String time = LocalDate.now().toString();
        String fileName = "总览报表" + time;
        try {
            fileName = URLEncoder.encode(fileName + ".xlsx", "UTF-8");
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        StreamingOutput output = outputStream -> {
            workbook.write(outputStream);
            outputStream.flush();
        };
        response.setContentType("application/octet-stream");
        return Response.ok(output).header("Content-disposition", "attachment;filename=" + fileName).header("Cache-Control", "no-cache").build();
    }


}