微调

njcn-springboot/spingboot2.3.12/src/main/java/com/njcn/web/utils/ExcelUtil.java

@@ -0,0 +1,62 @@
+package com.njcn.web.utils;
+
+import cn.afterturn.easypoi.excel.ExcelExportUtil;
+import cn.afterturn.easypoi.excel.entity.ExportParams;
+import cn.hutool.core.util.CharsetUtil;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.poi.ss.usermodel.Workbook;
+
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.net.URLEncoder;
+import java.util.Collection;
+
+/**
+ * @author caozehui
+ * @version 1.0.0
+ * @date 2024年11月05日
+ */
+@Slf4j
+public class ExcelUtil {
+    /**
+     * 指定名称、数据下载报表
+     *
+     * @param fileName 文件名
+     */
+    public static void exportExcel(String fileName, Class<?> pojoClass, Collection<?> dataSet) {
+        HttpServletResponse response = HttpServletUtil.getResponse();
+        try (ServletOutputStream outputStream = response.getOutputStream()) {
+            fileName = URLEncoder.encode(fileName, CharsetUtil.UTF_8);
+            response.reset();
+            response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");
+            response.setContentType("application/octet-stream;charset=UTF-8");
+            ExportParams exportParams = new ExportParams();
+            Workbook workbook = ExcelExportUtil.exportExcel(exportParams, pojoClass, dataSet);
+            workbook.write(outputStream);
+        } catch (IOException e) {
+            log.error(">>> 导出数据异常：{}", e.getMessage());
+        }
+    }
+
+    /**
+     * 指定名称、sheet名、数据下载报表
+     *
+     * @param fileName 文件名
+     */
+    public static void exportExcel(String fileName, String sheetName, Class<?> pojoClass, Collection<?> dataSet) {
+        HttpServletResponse response = HttpServletUtil.getResponse();
+        try (ServletOutputStream outputStream = response.getOutputStream()) {
+            fileName = URLEncoder.encode(fileName, CharsetUtil.UTF_8);
+            response.reset();
+            response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");
+            response.setContentType("application/octet-stream;charset=UTF-8");
+            ExportParams exportParams = new ExportParams();
+            exportParams.setSheetName(sheetName);
+            Workbook workbook = ExcelExportUtil.exportExcel(exportParams, pojoClass, dataSet);
+            workbook.write(outputStream);
+        } catch (IOException e) {
+            log.error(">>> 导出数据异常：{}", e.getMessage());
+        }
+    }
+}

njcn-springboot/spingboot2.3.12/src/main/java/com/njcn/web/utils/HttpServletUtil.java

@@ -0,0 +1,40 @@
+package com.njcn.web.utils;
+
+import com.njcn.common.pojo.enums.response.CommonResponseEnum;
+import com.njcn.common.pojo.exception.BusinessException;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+/**
+ *
+ * HttpServlet工具类，获取当前request和response
+ *
+ * @author hongawen
+ * @version 1.0.0
+ * @date 2022年04月01日 11:04
+ */
+public class HttpServletUtil {
+
+
+    public static HttpServletRequest getRequest() {
+        ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
+        if (requestAttributes == null) {
+            throw new BusinessException(CommonResponseEnum.REQUEST_EMPTY);
+        } else {
+            return requestAttributes.getRequest();
+        }
+    }
+
+
+    public static HttpServletResponse getResponse() {
+        ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
+        if (requestAttributes == null) {
+            throw new BusinessException(CommonResponseEnum.REQUEST_EMPTY);
+        } else {
+            return requestAttributes.getResponse();
+        }
+    }
+}