1. 增加支持多个sheet 对应map

unknown
1 parent a661d13e
Showing 4 changed files with 69 additions and 7 deletions Show diff stats
src/main/java/com/taover/easyexcel/event/SyncReadAllListener.java
src/main/java/com/taover/easyexcel/read/builder/ExcelReaderBuilder.java
src/main/java/com/taover/easyexcel/read/builder/ExcelReaderSheetBuilder.java
src/test/java/com/taover/easyexcel/test/demo/read/ReadTest.java
@@ -0,0 +1,37 @@
+package com.taover.easyexcel.event;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import com.taover.easyexcel.context.AnalysisContext;
+
+/**
+ * Synchronous data reading
+ *
+ * @author wangbin
+ */
+public class SyncReadAllListener extends AnalysisEventListener<Map<Integer, Object>> {
+	private int currSheetIndex = 0;
+    private Map<Integer, List<Map<Integer, Object>>> sheetMap = new HashMap<Integer, List<Map<Integer, Object>>>();
+
+    @Override
+    public void invoke(Map<Integer, Object> data, AnalysisContext context) {
+    	List<Map<Integer, Object>> sheetData = sheetMap.get(currSheetIndex);
+    	if(sheetData == null) {
+    		sheetData = new ArrayList<Map<Integer, Object>>();
+    	}
+    	sheetData.add((Map<Integer, Object>)data);
+    	this.sheetMap.put(this.currSheetIndex, sheetData);
+    }
+
+    @Override
+    public void doAfterAllAnalysed(AnalysisContext context) {
+    	++this.currSheetIndex;    	
+    }
+
+    public Map<Integer, List<Map<Integer, Object>>> getSheetMap() {
+        return this.sheetMap;
+    }
+}
@@ -4,6 +4,7 @@ import java.io.File;
 import java.io.InputStream;
 import java.util.HashSet;
 import java.util.List;
+import java.util.Map;
 import javax.xml.parsers.SAXParserFactory;
@@ -13,6 +14,7 @@ import com.taover.easyexcel.cache.selector.ReadCacheSelector;
 import com.taover.easyexcel.context.AnalysisContext;
 import com.taover.easyexcel.enums.CellExtraTypeEnum;
 import com.taover.easyexcel.event.AnalysisEventListener;
+import com.taover.easyexcel.event.SyncReadAllListener;
 import com.taover.easyexcel.event.SyncReadListener;
 import com.taover.easyexcel.read.listener.ModelBuildEventListener;
 import com.taover.easyexcel.read.metadata.ReadWorkbook;
@@ -198,13 +200,13 @@ public class ExcelReaderBuilder extends AbstractExcelReaderParameterBuilder&lt;Exce
         excelReader.readAll();
         excelReader.finish();
     }
-
+    
     /**
      * Synchronous reads return results
      *
      * @return
      */
-    public <T> List<T> doReadAllSync() {
+    public <T> List<T>  doReadAllSync() {
         SyncReadListener syncReadListener = new SyncReadListener();
         registerReadListener(syncReadListener);
         ExcelReader excelReader = build();
@@ -212,6 +214,20 @@ public class ExcelReaderBuilder extends AbstractExcelReaderParameterBuilder&lt;Exce
         excelReader.finish();
         return (List<T>)syncReadListener.getList();
     }
+    
+    /**
+     * Synchronous reads return results
+     *
+     * @return
+     */
+    public Map<Integer, List<Map<Integer, Object>>> doReadAllSyncForMap() {
+        SyncReadAllListener syncReadListener = new SyncReadAllListener();
+        registerReadListener(syncReadListener);
+        ExcelReader excelReader = build();
+        excelReader.readAll();
+        excelReader.finish();
+        return (Map<Integer, List<Map<Integer, Object>>>)syncReadListener.getSheetMap();
+    }
     public ExcelReaderSheetBuilder sheet() {
         return sheet(null, null);
 package com.taover.easyexcel.read.builder;
 import java.util.List;
+import java.util.Map;
 import com.taover.easyexcel.ExcelReader;
+import com.taover.easyexcel.event.SyncReadAllListener;
 import com.taover.easyexcel.event.SyncReadListener;
 import com.taover.easyexcel.exception.ExcelAnalysisException;
 import com.taover.easyexcel.exception.ExcelGenerateException;
@@ -3,6 +3,7 @@ package com.taover.easyexcel.test.demo.read;
 import java.io.File;
 import java.util.List;
 import java.util.Map;
+import java.util.Map.Entry;
 import org.junit.Ignore;
 import org.junit.Test;
@@ -244,12 +245,12 @@ public class ReadTest {
      */
     @Test
     public void synchronousRead() {
-        String fileName = TestFileUtil.getPath() + "demo" + File.separator + "demo.xls";
+        String fileName = TestFileUtil.getPath() + "demo" + File.separator + "demo.xlsx";
         // 这里 需要指定读用哪个class去读，然后读取第一个sheet 同步读取会自动finish
-        List<DemoData> list = EasyExcel.read(fileName).head(DemoData.class).sheet().doReadSync();
-        for (DemoData data : list) {
-            LOGGER.info("读取到数据:{}", JSON.toJSONString(data));
-        }
+//        List<DemoData> list = EasyExcel.read(fileName).head(DemoData.class).sheet().doReadSync();
+//        for (DemoData data : list) {
+//            LOGGER.info("读取到数据:{}", JSON.toJSONString(data));
+//        }
         // 这里 也可以不指定class，返回一个list，然后读取第一个sheet 同步读取会自动finish
 //        List<Map<Integer, String>> listMap = EasyExcel.read(fileName).sheet().doReadSync();
@@ -257,6 +258,12 @@ public class ReadTest {
 //            // 返回每条数据的键值对 表示所在的列 和所在列的值
 //            LOGGER.info("读取到数据:{}", JSON.toJSONString(data));
 //        }
+        
+        Map<Integer, List<Map<Integer, Object>>> mapListMap = EasyExcel.read(fileName).doReadAllSyncForMap();
+        for (Entry<Integer, List<Map<Integer, Object>>> item: mapListMap.entrySet()) {
+            // 返回每条数据的键值对 表示所在的列 和所在列的值
+            LOGGER.info("读取到数据:{}\n{}", JSON.toJSONString(item.getKey()), JSON.toJSONString(item.getValue()));
+        }
     }
     /**
@@ -0,0 +1,37 @@		@@ -0,0 +1,37 @@
	1	+package com.taover.easyexcel.event;
	2	+
	3	+import java.util.ArrayList;
	4	+import java.util.HashMap;
	5	+import java.util.List;
	6	+import java.util.Map;
	7	+
	8	+import com.taover.easyexcel.context.AnalysisContext;
	9	+
	10	+/**
	11	+ * Synchronous data reading
	12	+ *
	13	+ * @author wangbin
	14	+ */
	15	+public class SyncReadAllListener extends AnalysisEventListener<Map<Integer, Object>> {
	16	+ private int currSheetIndex = 0;
	17	+ private Map<Integer, List<Map<Integer, Object>>> sheetMap = new HashMap<Integer, List<Map<Integer, Object>>>();
	18	+
	19	+ @Override
	20	+ public void invoke(Map<Integer, Object> data, AnalysisContext context) {
	21	+ List<Map<Integer, Object>> sheetData = sheetMap.get(currSheetIndex);
	22	+ if(sheetData == null) {
	23	+ sheetData = new ArrayList<Map<Integer, Object>>();
	24	+ }
	25	+ sheetData.add((Map<Integer, Object>)data);
	26	+ this.sheetMap.put(this.currSheetIndex, sheetData);
	27	+ }
	28	+
	29	+ @Override
	30	+ public void doAfterAllAnalysed(AnalysisContext context) {
	31	+ ++this.currSheetIndex;
	32	+ }
	33	+
	34	+ public Map<Integer, List<Map<Integer, Object>>> getSheetMap() {
	35	+ return this.sheetMap;
	36	+ }
	37	+}