123

ace-common/pom.xml

@@ -24,6 +24,7 @@
     </build>
     <properties>
         <mapper.version>3.4.0</mapper.version>
+        <poi.version>3.15</poi.version>
     </properties>
     <dependencies>
 
@@ -125,6 +126,50 @@
             <scope>compile</scope>
         </dependency>
 
+        <!-- excel 组件 -->
+        <dependency>
+            <groupId>org.apache.poi</groupId>
+            <artifactId>poi</artifactId>
+            <version>${poi.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.poi</groupId>
+            <artifactId>poi-ooxml</artifactId>
+            <version>${poi.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.poi</groupId>
+            <artifactId>poi-ooxml-schemas</artifactId>
+            <version>${poi.version}</version>
+            <exclusions>
+                <exclusion>
+                    <artifactId>stax-api</artifactId>
+                    <groupId>stax</groupId>
+                </exclusion>
+            </exclusions>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.poi</groupId>
+            <artifactId>poi-ooxml</artifactId>
+            <version>${poi.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.xmlbeans</groupId>
+            <artifactId>xmlbeans</artifactId>
+            <version>2.6.0</version>
+            <exclusions>
+                <exclusion>
+                    <artifactId>stax-api</artifactId>
+                    <groupId>stax</groupId>
+                </exclusion>
+            </exclusions>
+        </dependency>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <optional>true</optional>
+        </dependency>
+
     </dependencies>
 
 

ace-common/src/main/java/com/github/wxiaoqi/security/common/util/excel/Excel2003Reader.java

+package com.github.wxiaoqi.security.common.util.excel;
+
+
+import org.apache.poi.hssf.eventusermodel.EventWorkbookBuilder.SheetRecordCollectingListener;
+import org.apache.poi.hssf.eventusermodel.dummyrecord.LastCellOfRowDummyRecord;
+import org.apache.poi.hssf.eventusermodel.dummyrecord.MissingCellDummyRecord;
+import org.apache.poi.hssf.model.HSSFFormulaParser;
+import org.apache.poi.hssf.usermodel.HSSFWorkbook;
+import org.apache.poi.poifs.filesystem.POIFSFileSystem;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * Excel2003格式解析
+ *
+ * @author lipengjun
+ * @email 939961241@qq.com
+ * @date 2017年10月28日 13:11:27
+ */
+public class Excel2003Reader implements HSSFListener {
+    private int minColumns = -1;
+
+    private POIFSFileSystem fs;
+
+    /**
+     * 最后一行行号
+     */
+    private int lastRowNumber;
+
+    /**
+     * 最后一列列号
+     */
+    private int lastColumnNumber;
+
+    /**
+     * Should we output the formula, or the value it has?
+     */
+    private boolean outputFormulaValues = true;
+
+    /**
+     * For parsing Formulas
+     */
+    private SheetRecordCollectingListener workbookBuildingListener;
+
+    // 工作薄
+    private HSSFWorkbook stubWorkbook;
+
+    // Records we pick up as we process
+    private SSTRecord sstRecord;
+
+    private FormatTrackingHSSFListener formatListener;
+
+    // 表索引
+    private int sheetIndex = -1;
+
+    private BoundSheetRecord[] orderedBSRs;
+
+    @SuppressWarnings("rawtypes")
+    private ArrayList boundSheetRecords = new ArrayList();
+
+    // For handling formulas with string results
+    private int nextRow;
+
+    private int nextColumn;
+
+    private boolean outputNextStringRecord;
+
+    // 存储行记录的容器
+    private List<String> rowlist = new ArrayList<String>();
+    ;
+
+    // 单Sheet数据
+    private List<String[]> sheetData = new ArrayList<String[]>();
+
+    // 多Sheet数据
+    private Map<Integer, List<String[]>> workData = new HashMap<Integer, List<String[]>>();
+
+    /**
+     * 遍历excel下所有的sheet
+     *
+     * @param fileStream 处理文件流
+     * @throws IOException 抛出IO异常
+     */
+    public void process(InputStream fileStream)
+            throws IOException {
+        this.fs = new POIFSFileSystem(fileStream);
+        MissingRecordAwareHSSFListener listener = new MissingRecordAwareHSSFListener(
+                this);
+        formatListener = new FormatTrackingHSSFListener(listener);
+        HSSFEventFactory factory = new HSSFEventFactory();
+        HSSFRequest request = new HSSFRequest();
+        if (outputFormulaValues) {
+            request.addListenerForAllRecords(formatListener);
+        } else {
+            workbookBuildingListener = new SheetRecordCollectingListener(
+                    formatListener);
+            request.addListenerForAllRecords(workbookBuildingListener);
+        }
+        factory.processWorkbookEvents(request, fs);
+    }
+
+    /**
+     * HSSFListener 监听方法，处理 Record
+     *
+     * @param record 行记录
+     */
+    @SuppressWarnings("unchecked")
+    @Override
+    public void processRecord(Record record) {
+        int thisRow = -1;
+        int thisColumn = -1;
+        String thisStr = null;
+        String value = null;
+        switch (record.getSid()) {
+            case BoundSheetRecord.sid:
+                boundSheetRecords.add(record);
+                break;
+            case BOFRecord.sid:
+                BOFRecord br = (BOFRecord) record;
+                if (br.getType() == BOFRecord.TYPE_WORKSHEET) {
+                    // 如果有需要，则建立子工作薄
+                    if (workbookBuildingListener != null && stubWorkbook == null) {
+                        stubWorkbook = workbookBuildingListener.getStubHSSFWorkbook();
+                    }
+                    if (sheetIndex >= 0) {
+                        List<String[]> data = new ArrayList<String[]>();
+                        data.addAll(sheetData);
+                        workData.put(sheetIndex, data);
+                        sheetData.clear();
+                    }
+                    sheetIndex++;
+                    if (orderedBSRs == null) {
+                        orderedBSRs = BoundSheetRecord.orderByBofPosition(boundSheetRecords);
+                    }
+                }
+                break;
+            case SSTRecord.sid:
+                sstRecord = (SSTRecord) record;
+                break;
+            case BlankRecord.sid:
+                BlankRecord brec = (BlankRecord) record;
+                thisRow = brec.getRow();
+                thisColumn = brec.getColumn();
+                thisStr = "";
+                rowlist.add(thisColumn, thisStr);
+                break;
+            case BoolErrRecord.sid: // 单元格为布尔类型
+                BoolErrRecord berec = (BoolErrRecord) record;
+                thisRow = berec.getRow();
+                thisColumn = berec.getColumn();
+                thisStr = berec.getBooleanValue() + "";
+                rowlist.add(thisColumn, thisStr);
+                break;
+            case FormulaRecord.sid: // 单元格为公式类型
+                FormulaRecord frec = (FormulaRecord) record;
+                thisRow = frec.getRow();
+                thisColumn = frec.getColumn();
+                if (outputFormulaValues) {
+                    if (Double.isNaN(frec.getValue())) {
+                        // Formula result is a string
+                        // This is stored in the next record
+                        outputNextStringRecord = true;
+                        nextRow = frec.getRow();
+                        nextColumn = frec.getColumn();
+                    } else {
+                        thisStr = formatListener.formatNumberDateCell(frec);
+                    }
+                } else {
+                    thisStr = '"' + HSSFFormulaParser.toFormulaString(
+                            stubWorkbook, frec.getParsedExpression()) + '"';
+                }
+                rowlist.add(thisColumn, thisStr);
+                break;
+            case StringRecord.sid:// 单元格中公式的字符串
+                if (outputNextStringRecord) {
+                    // String for formula
+                    StringRecord srec = (StringRecord) record;
+                    thisStr = srec.getString();
+                    thisRow = nextRow;
+                    thisColumn = nextColumn;
+                    outputNextStringRecord = false;
+                }
+                break;
+            case LabelRecord.sid:
+                LabelRecord lrec = (LabelRecord) record;
+                thisColumn = lrec.getColumn();
+                value = lrec.getValue().trim();
+                value = value.equals("") ? " " : value;
+                this.rowlist.add(thisColumn, value);
+                break;
+            case LabelSSTRecord.sid: // 单元格为字符串类型
+                LabelSSTRecord lsrec = (LabelSSTRecord) record;
+                thisColumn = lsrec.getColumn();
+                if (sstRecord == null) {
+                    rowlist.add(thisColumn, " ");
+                } else {
+                    value = sstRecord.getString(lsrec.getSSTIndex()).toString().trim();
+                    value = value.equals("") ? " " : value;
+                    rowlist.add(thisColumn, value);
+                }
+                break;
+            case NumberRecord.sid: // 单元格为数字类型
+                NumberRecord numrec = (NumberRecord) record;
+                thisColumn = numrec.getColumn();
+                value = formatListener.formatNumberDateCell(numrec).trim();
+                value = value.equals("") ? " " : value;
+                // 向容器加入列值
+                rowlist.add(thisColumn, value);
+                break;
+            default:
+                break;
+        }
+
+        // 遇到新行的操作
+        if (thisRow != -1 && thisRow != lastRowNumber) {
+            lastColumnNumber = -1;
+        }
+
+        // 空值的操作
+        if (record instanceof MissingCellDummyRecord) {
+            MissingCellDummyRecord mc = (MissingCellDummyRecord) record;
+            thisColumn = mc.getColumn();
+            rowlist.add(thisColumn, "");
+        }
+
+        // 更新行和列的值
+        if (thisRow > -1) {
+            lastRowNumber = thisRow;
+        }
+        if (thisColumn > -1) {
+            lastColumnNumber = thisColumn;
+        }
+
+        // 行结束时的操作
+        if (record instanceof LastCellOfRowDummyRecord) {
+            if (minColumns > 0) {
+                // 列值重新置空
+                if (lastColumnNumber == -1) {
+                    lastColumnNumber = 0;
+                }
+            }
+            lastColumnNumber = -1;
+            // 每行结束时， 数据写入集合
+            sheetData.add(rowlist.toArray(new String[]{}));
+            // 清空容器
+            rowlist.clear();
+        }
+    }
+
+    /**
+     * 获取数据(单Sheet)
+     *
+     * @param sheetIndex sheet下标
+     * @return List<String[]> 数据
+     */
+    public List<String[]> getSheetData(Integer sheetIndex) {
+        return workData.get(sheetIndex);
+    }
+
+    /**
+     * 获取数据(多Sheet)
+     *
+     * @return Map<Integer, List<String[]>> 多sheet的数据
+     */
+    public Map<Integer, List<String[]>> getSheetData() {
+        return workData;
+    }
+}
\ No newline at end of file

ace-common/src/main/java/com/github/wxiaoqi/security/common/util/excel/ExcelImport.java

+package com.github.wxiaoqi.security.common.util.excel;
+
+
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
+import java.util.List;
+import java.util.Map;
+
+
+/**
+ * Excel文件导入的基本功能类
+ * 可导入EXCEL2003 和 EXCEL2007格式。
+ *
+ * @author lipengjun
+ * @email 939961241@qq.com
+ * @date 2017年10月28日 13:11:27
+ */
+public class ExcelImport {
+    /**
+     * excel2003扩展名
+     */
+    public static final String EXCEL03_EXTENSION = ".xls";
+
+    /**
+     * excel2007扩展名
+     */
+    public static final String EXCEL07_EXTENSION = ".xlsx";
+
+    private ExcelImport() {
+    }
+
+    /**
+     * 解析EXCEL数据为 List<String[]>
+     *
+     * @param excelFile 要解析的上传EXCEL文件
+     * @return List<String[]) 行（列）
+     */
+    public static List<String[]> getExcelData07(MultipartFile excelFile) {
+        List<String[]> resultList = null;
+
+        if (null == excelFile || excelFile.isEmpty()) {
+            throw new RuntimeException("文件内容为空！");
+
+        }
+
+        Excel2007Reader excel07 = new Excel2007Reader();
+        try {
+            excel07.process(excelFile.getInputStream(), false);
+        } catch (Exception e) {
+            throw new RuntimeException("excel解析失败！");
+        }
+        resultList = excel07.getSheetData(0);
+
+        return resultList;
+    }
+
+    /**
+     * 解析EXCEL数据为 List<String[]>
+     *
+     * @param excelFile 要解析的上传EXCEL文件
+     * @return List<String[]) 行（列）
+     */
+    public static List<String[]> getExcelData03(MultipartFile excelFile) {
+        List<String[]> resultList = null;
+
+        if (null == excelFile || excelFile.isEmpty()) {
+            throw new RuntimeException("文件内容为空！");
+
+        }
+
+        Excel2003Reader excel03 = new Excel2003Reader();// 实例化excel处理对象
+        try {
+            excel03.process(excelFile.getInputStream());
+        } catch (IOException e) {
+            throw new RuntimeException("excel解析失败！");
+        }
+        resultList = excel03.getSheetData(0);
+
+        return resultList;
+    }
+
+    /**
+     * 通过解析MultipartFile对象获取excel内容，并且将其拼装为List<String[]>对象返回
+     *
+     * @param excelFile
+     * @return
+     * @throws Exception
+     */
+    public static List<String[]> getExcelData(MultipartFile excelFile)
+            throws RuntimeException {
+        List<String[]> resultList = null;
+
+        if (!excelFile.isEmpty()) {// 上传的文件不能为空
+            String excelFileName = excelFile.getOriginalFilename();// 文件名（带后缀）
+            if (excelFileName.toLowerCase().endsWith(EXCEL03_EXTENSION)) {// 如果文件是以.xls为后缀
+                Excel2003Reader excel03 = new Excel2003Reader();// 实例化excel处理对象
+                try {
+                    excel03.process(excelFile.getInputStream());
+                } catch (IOException e) {
+                    throw new RuntimeException("excel解析失败！");
+                }
+                resultList = excel03.getSheetData(0);
+            } else if (excelFileName.toLowerCase().endsWith(EXCEL07_EXTENSION)) {// 如果文件是以.xlsx为后缀
+                Excel2007Reader excel07 = new Excel2007Reader();
+                try {
+                    excel07.process(excelFile.getInputStream(), false);
+                } catch (Exception e) {
+                    throw new RuntimeException("excel解析失败！");
+                }
+                resultList = excel07.getSheetData(0);
+            }
+        }
+
+        return resultList;
+    }
+
+    /**
+     * 通过解析MultipartFile对象获取excel内容，并且将其拼装为Map<Integer, List<String[]>>对象返回
+     *
+     * @param excelFile
+     * @return
+     * @throws Exception
+     */
+    public static Map<Integer, List<String[]>> getExcelDataAll(MultipartFile excelFile)
+            throws RuntimeException {
+        Map<Integer, List<String[]>> result = null;
+
+        if (!excelFile.isEmpty()) {// 上传的文件不能为空
+            String excelFileName = excelFile.getOriginalFilename();// 文件名（带后缀）
+            if (excelFileName.toLowerCase().endsWith(EXCEL03_EXTENSION)) {// 如果文件是以.xls为后缀
+                Excel2003Reader excel03 = new Excel2003Reader();// 实例化excel处理对象
+                try {
+                    excel03.process(excelFile.getInputStream());
+                } catch (IOException e) {
+                    throw new RuntimeException("excel解析失败！");
+                }
+                result = excel03.getSheetData();
+            } else if (excelFileName.toLowerCase().endsWith(EXCEL07_EXTENSION)) {// 如果文件是以.xlsx为后缀
+                Excel2007Reader excel07 = new Excel2007Reader();
+                try {
+                    excel07.process(excelFile.getInputStream(), true);
+                } catch (Exception e) {
+                    throw new RuntimeException("excel解析失败！");
+                }
+                result = excel07.getSheetData();
+            }
+        }
+
+        return result;
+    }
+}

ace-common/src/main/java/com/github/wxiaoqi/security/common/util/excel/XssfDataType.java

+package com.github.wxiaoqi.security.common.util.excel;
+
+/**
+ * XSSFDataType
+ *
+ * @author lipengjun
+ * @email 939961241@qq.com
+ * @date 2017年10月28日 13:11:27
+ */
+public enum XssfDataType {
+    BOOL, ERROR, FORMULA, INLINESTR, SSTINDEX, NUMBER, DATE, DATETIME, TIME,
+}