test read excel

Author: netbuffer

pom.xml

@@ -22,6 +22,7 @@
 		<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
 		<jfinal.version>4.8</jfinal.version>
 		<freemarker.version>2.3.30</freemarker.version>
+		<poi.version>4.1.2</poi.version>
 	</properties>
 
 	<dependencies>
@@ -101,7 +102,7 @@
 		<dependency>
 			<groupId>org.apache.poi</groupId>
 			<artifactId>poi</artifactId>
-			<version>3.13</version>
+			<version>${poi.version}</version>
 		</dependency>
 		<dependency>
 			<groupId>junit</groupId>

src/main/java/cn/netbuffer/jfinal_bootstrap_table/controller/IndexController.java

@@ -103,7 +103,7 @@ public void export() {
         titles.add("phone");
         titles.add("sex");
         String file = projectPath + format.format(new Date()) + "." + ConfigConstant.EXCELSTR;
-        POIExcelUtil.export(titles, mps, file);
+        POIExcelUtil.write(titles, mps, file);
         renderFile(new File(file));
     }
 

src/main/java/cn/netbuffer/jfinal_bootstrap_table/util/POIExcelUtil.java

@@ -1,18 +1,56 @@
 package cn.netbuffer.jfinal_bootstrap_table.util;
 
+import lombok.extern.slf4j.Slf4j;
 import org.apache.poi.hssf.usermodel.HSSFWorkbook;
 import org.apache.poi.ss.usermodel.*;
 
+import java.io.File;
+import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.IOException;
-import java.util.Iterator;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
+import java.util.*;
 
+@Slf4j
 public class POIExcelUtil {
 
-    public static void export(List<String> titles, List<Map<String, Object>> datas, String path) {
+    public static List<List<String>> read(String path) {
+        Workbook workbook = null;
+        try {
+            workbook = new HSSFWorkbook(new FileInputStream(new File(path)));
+        } catch (IOException e) {
+            log.error(e.getMessage());
+        }
+        if (workbook == null) {
+            return null;
+        }
+//        for (int sheetNum = 0; sheetNum < workbook.getNumberOfSheets(); sheetNum++) {
+//        }
+        Sheet sheet = workbook.getSheetAt(0);
+        int firstRowNum = sheet.getFirstRowNum();
+        Row firstRow = sheet.getRow(firstRowNum);
+        if (null == firstRow) {
+            log.warn("解析Excel失败，在第一行没有读取到任何数据！");
+        }
+        // 解析每一行的数据，构造数据对象
+        int rowStart = firstRowNum + 1;
+        int rowEnd = sheet.getPhysicalNumberOfRows();
+        List<List<String>> data = new ArrayList<>();
+        for (int rowNum = rowStart; rowNum < rowEnd; rowNum++) {
+            Row row = sheet.getRow(rowNum);
+            if (null == row) {
+                continue;
+            }
+            int cells = row.getPhysicalNumberOfCells();
+            List<String> arr = new ArrayList<>(cells);
+            for (int i = 0; i < cells; i++) {
+                arr.add(row.getCell(i).getStringCellValue());
+            }
+            data.add(arr);
+        }
+        return data;
+    }
+
+    public static void write(List<String> titles, List<Map<String, Object>> datas, String path) {
         Workbook wb = new HSSFWorkbook();
         Sheet sheet = wb.createSheet(titles.get(0));
         // 冻结该行，使其无法移动
@@ -30,17 +68,17 @@ public static void export(List<String> titles, List<Map<String, Object>> datas,
         titleFont.setBold(true);
         titleFont.setColor(IndexedColors.AQUA.getIndex());
         CellStyle cellStyle = wb.createCellStyle();
-        cellStyle.setAlignment(CellStyle.ALIGN_CENTER);
-        cellStyle.setVerticalAlignment(CellStyle.VERTICAL_CENTER);
+        cellStyle.setAlignment(HorizontalAlignment.CENTER);
+        cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
         cellStyle.setFont(titleFont);
         for (int i = 0; i < titleCount; i++) {
             Cell cell = row.createCell(i);
             cell.setCellValue(titles.get(i));
             cell.setCellStyle(cellStyle);
         }
         CellStyle contentStyle = wb.createCellStyle();
-        cellStyle.setAlignment(CellStyle.ALIGN_CENTER);
-        cellStyle.setVerticalAlignment(CellStyle.VERTICAL_CENTER);
+        cellStyle.setAlignment(HorizontalAlignment.CENTER);
+        cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
         int dataCount = datas.size();
         for (int i = 1; i < dataCount + 1; i++) {
             Row rowIndex = sheet.createRow((short) i);

src/test/java/cn/netbuffer/jfinal_bootstrap_table/util/TestPOI.java

@@ -1,5 +1,6 @@
 package cn.netbuffer.jfinal_bootstrap_table.util;
 
+import lombok.extern.slf4j.Slf4j;
 import org.junit.Test;
 
 import java.io.File;
@@ -8,11 +9,18 @@
 import java.util.List;
 import java.util.Map;
 
+@Slf4j
 public class TestPOI {
 
+    String path = System.getProperty("user.dir") + File.separator + "test.xlsx";
+
+    @Test
+    public void testReadFile() {
+        log.info("read {}:\n{}", path, POIExcelUtil.read(path));
+    }
+
     @Test
     public void testExportFile() {
-        String dir = System.getProperty("user.dir");
         List<String> titles = new ArrayList<>();
         titles.add("昵称");
         titles.add("姓名");
@@ -21,12 +29,12 @@ public void testExportFile() {
         for (int i = 0; i < 4; i++) {
             //保障顺序
             Map<String, Object> mp = new LinkedHashMap<String, Object>();
-            mp.put("n", "tt" + i);
-            mp.put("x", "行ing" + i);
-            mp.put("b", "男" + i);
+            mp.put("nick", "nick" + i);
+            mp.put("name", "name" + i);
+            mp.put("sex", "sex" + i);
             datas.add(mp);
         }
-        POIExcelUtil.export(titles, datas, dir + File.separator + "test.xls");
+        POIExcelUtil.write(titles, datas, path);
     }
 
 }