修改设备

cs-device/cs-device-api/src/main/java/com/njcn/csdevice/utils/ExcelStyleUtil.java

@@ -3,9 +3,15 @@ package com.njcn.csdevice.utils;
 import cn.afterturn.easypoi.excel.entity.params.ExcelExportEntity;
 import cn.afterturn.easypoi.excel.entity.params.ExcelForEachParams;
 import cn.afterturn.easypoi.excel.export.styler.IExcelExportStyler;
+import com.njcn.common.pojo.enums.response.CommonResponseEnum;
+import com.njcn.common.pojo.exception.BusinessException;
 import org.apache.poi.hssf.util.HSSFColor;
 import org.apache.poi.ss.usermodel.*;
 
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletResponse;
+import java.net.URLEncoder;
+
 /**
  * pqs
  *
@@ -155,4 +161,40 @@ public class ExcelStyleUtil implements IExcelExportStyler {
         style.setVerticalAlignment(VerticalAlignment.CENTER);
         return style;
     }
+
+    public static void exportFileByWorkbook(Workbook workbook, String fileName, HttpServletResponse response) {
+        response.reset();
+
+        try {
+            ServletOutputStream outputStream = response.getOutputStream();
+            Throwable var4 = null;
+
+            try {
+                fileName = URLEncoder.encode(fileName, "UTF-8");
+                response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");
+                response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=UTF-8");
+                workbook.write(outputStream);
+            } catch (Throwable var14) {
+                var4 = var14;
+                throw var14;
+            } finally {
+                if (outputStream != null) {
+                    if (var4 != null) {
+                        try {
+                            outputStream.close();
+                        } catch (Throwable var13) {
+                            var4.addSuppressed(var13);
+                        }
+                    } else {
+                        outputStream.close();
+                    }
+                }
+
+            }
+
+        } catch (Exception var16) {
+            var16.printStackTrace();
+            throw new BusinessException(CommonResponseEnum.EXPORT_FILE_ERROR);
+        }
+    }
 }