Merge pull request #1 from mrdear/master

💥 modify workbook

Author: hellokaton

src/main/java/io/github/biezhi/excel/plus/writer/ExcelWriter.java

@@ -1,14 +1,17 @@
 package io.github.biezhi.excel.plus.writer;
 
 import io.github.biezhi.excel.plus.Constant;
-import io.github.biezhi.excel.plus.enums.ExcelType;
 import io.github.biezhi.excel.plus.exception.ExcelException;
 import io.github.biezhi.excel.plus.utils.ExcelUtils;
 import io.github.biezhi.excel.plus.utils.Pair;
-import org.apache.poi.hssf.usermodel.HSSFWorkbook;
-import org.apache.poi.ss.usermodel.*;
+
+import org.apache.poi.ss.usermodel.Cell;
+import org.apache.poi.ss.usermodel.CellStyle;
+import org.apache.poi.ss.usermodel.Row;
+import org.apache.poi.ss.usermodel.Sheet;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.ss.usermodel.WorkbookFactory;
 import org.apache.poi.ss.util.CellRangeAddress;
-import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 
 import java.io.InputStream;
 import java.io.OutputStream;

src/main/java/io/github/biezhi/excel/plus/writer/Exporter.java

@@ -1,8 +1,12 @@
 package io.github.biezhi.excel.plus.writer;
 
 import io.github.biezhi.excel.plus.enums.ExcelType;
+
+import org.apache.poi.hssf.usermodel.HSSFWorkbook;
 import org.apache.poi.ss.usermodel.CellStyle;
 import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.xssf.streaming.SXSSFWorkbook;
+import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 
 import java.util.Collection;
 import java.util.function.Function;
@@ -15,6 +19,8 @@
  */
 public class Exporter<T> {
 
+    private static final Integer BIG_DATA_EXCEL_BOUNDARY = 10000;
+
     private String        headerTitle;
     private String        templatePath;
     private ExcelType     excelType;
@@ -100,4 +106,12 @@ public ExcelType getExcelType() {
         return excelType;
     }
 
+
+    Workbook createWorkbook() {
+        if (getExcelType() == ExcelType.XLSX) {
+            return getData().size() > BIG_DATA_EXCEL_BOUNDARY ? new SXSSFWorkbook(): new XSSFWorkbook();
+        }
+        return new HSSFWorkbook();
+    }
+
 }