导出调整

16639036659

2024-05-15 3ea61dd8477c0e652ff660aabbd11a11cdf63a45

 src/main/java/com/gkhy/labRiskManage/api/controller/basic/OldRiskAssessController.java

@@ -20,6 +20,7 @@
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.FileInputStream;

/**
@@ -134,20 +135,21 @@
    }

    @PostMapping("/select/exportOldRiskAssess")
    public Result exportOldRiskAssess(@RequestBody OldRiskAssessQueryReqBO queryReqDO){
    public Result exportOldRiskAssess(@RequestBody OldRiskAssessQueryReqBO queryReqDO, HttpServletResponse response, HttpServletRequest request){

        Result result = new Result();
        result.setCode(ResultCode.OK);
        result.setMsg("导出成功");


        int importResult = oldRiskAssessService.exportOldRiskAssess(getCurrentUserId(), queryReqDO);
        Result<OldRiskAssessExportRespDTO>  importResult = oldRiskAssessService.exportOldRiskAssess(getCurrentUserId(), queryReqDO, response, request);

        if (importResult < 1){
            result.setCode(ResultCode.NOT_OK);
            result.setMsg("导出失败");
        }
        result.setCount(importResult);
//        if (importResult < 1){
//            result.setCode(ResultCode.NOT_OK);
//            result.setMsg("导出失败");
//        }
        result.setCount(importResult.getCount());
        result.setData(importResult.getData());

        return result;
    }