[开发测试]EasyExcel导出工具类

/**
 * 导出设备工具类
 *
 * @author: 杨阳
 * @date: 2022/2/25
 */
@Slf4j
public class ExcelExportUtil {
    private ExcelExportUtil() {
    }

    /**
     * @param response  响应体
     * @param data      数据
     * @param fileName  导出文件名
     * @param sheetName 导出文件的sheet名
     * @param object    对应的excel模板类
     */
    public static void writeExcel(HttpServletResponse response, List<ExportEquipmentVO> data, String fileName, String sheetName, Object object) throws Exception {
        if (CollUtil.isEmpty(data)) {
            log.warn("Excel导出数据为空!");
        }
        EasyExcelFactory.write(getOutputStream(fileName, response), object.getClass())
                .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy())
                .sheet(sheetName)
                .doWrite(data);
    }

    private static OutputStream getOutputStream(String fileName, HttpServletResponse response) throws Exception {
        try {
            fileName = URLEncoder.encode(fileName, "UTF-8");
            response.setContentType("application/vnd.ms-excel");
            response.setCharacterEncoding("utf-8");
            response.setHeader("Content-disposition", "attachment;filename=" + fileName + ";filename*=utf-8''" + fileName + ".xlsx");
            response.setHeader("Pragma", "public");
            response.setHeader("Cache-Control", "no-store");
            response.addHeader("Cache-Control", "max-age=0");
            return response.getOutputStream();
        } catch (IOException e) {
            log.error("导出excel表格失败,失败原因{}", e.getMessage());
            throw new Exception("导出excel表格失败!", e);
        }
    }

}