Apache POI & EasyExcel。
文章目录
Apache POI。
Apache POI 是 Apache 软件基金会的开放源码函式库,POI 提供 API 给 Java 程序对 Microsoft Office 格式档案读和写的功能。
HSSF - 提供读写 Microsoft Excel 格式档案的功能。(2003 版,最多 65535 行)。
XSSF - 提供读写 Microsoft Excel OOXML 格式档案的功能。(2007 版)
HWPF - 提供读写 Microsoft Word 格式档案的功能。
HSLF - 提供读写 Microsoft PowerPoint 格式档案的功能。
HDGF - 提供读写 Microsoft Visio 格式档案的功能。
EasyExcel。
https://github.com/alibaba/easyexcel
JAVA 解析 Excel 工具 ~ EasyExcel。
Java 解析、生成 Excel 比较有名的框架有 Apache poi、jxl。但他们都存在一个严重的问题就是非常的耗内存,poi 有一套 SAX 模式的 API 可以一定程度的解决一些内存溢出的问题,但 POI 还是有一些缺陷,比如 07 版 Excel 解压缩以及解压后存储都是在内存中完成的,内存消耗依然很大。easyexcel 重写了 poi 对 07 版 Excel 的解析,能够原本一个 3M 的 excel 用 POI sax 依然需要 100M 左右内存降低到几 M,并且再大的 excel 不会出现内存溢出,03 版依赖 POI 的 sax 模式。在上层做了模型转换的封装,让使用者更加简单方便。
https://www.yuque.com/easyexcel/doc/easyexcel
Excel xls 和 xlsx 最主要区别。
- 2007 版本的 xlsx 行数没有限制。
- 而 2003 版本的 xls 最多 65536 行。
万物皆对象。
区别。
- XSSFWorkbook~xlsx。
- HSSFWorkbook~xls。
package com.geek;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
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.xssf.usermodel.XSSFWorkbook;
import org.joda.time.DateTime;
import org.junit.Test;
import java.io.FileOutputStream;
import java.io.IOException;
public class ExcelWriteTest {
private final String PATH = "G:\\lyfGeek\\IdeaProjects\\poi_easyExcel_geek\\geek-poi";
@Test
public void testWrite07() throws IOException {
// 创建一个工作薄。
Workbook workbook = new XSSFWorkbook();
// 创建一个工作表。
Sheet sheet = workbook.createSheet("Geek 统计表");
// 创建一行。
Row row1 = sheet.createRow(0);
// 创建一个单元格。
Cell cell11 = row1.createCell(0);
cell11.setCellValue("今日新增观众");
// (1, 2)
Cell cell12 = row1.createCell(1);
cell12.setCellValue(666);
// 第二行。(2, 1)
Row row2 = sheet.createRow(1);
Cell cell21 = row2.createCell(0);
cell21.setCellValue("统计时间");
// (2, 2)
Cell cell22 = row2.createCell(1);
String date = new DateTime().toString("yyyy-MM-dd HH:mm:ss");
cell22.setCellValue(date);
// 生成一张表。IO 流。03 版本 .xls。
FileOutputStream fileOutputStream = new FileOutputStream(PATH + "Geek 观众统计表07.xlsx");
workbook.write(fileOutputStream);
// 关闭流。
fileOutputStream.close();
System.out.println("Excel 生成完毕。");
}
@Test
public void testWrite03() throws IOException {
// 创建一个工作薄。
Workbook workbook = new HSSFWorkbook();
// 创建一个工作表。
Sheet sheet = workbook.createSheet("Geek 统计表");
// 创建一行。
Row row1 = sheet.createRow(0);
// 创建一个单元格。
Cell cell11 = row1.createCell(0);
cell11.setCellValue("今日新增观众");
// (1, 2)
Cell cell12 = row1.createCell(1);
cell12.setCellValue(666);
// 第二行。(2, 1)
Row row2 = sheet.createRow(1);
Cell cell21 = row2.createCell(0);
cell21.setCellValue("统计时间");
// (2, 2)
Cell cell22 = row2.createCell(1);
String date = new DateTime().toString("yyyy-MM-dd HH:mm:ss");
cell22.setCellValue(date);
// 生成一张表。IO 流。03 版本 .xls。
FileOutputStream fileOutputStream = new FileOutputStream(PATH + "Geek 观众统计表03.xls");
workbook.write(fileOutputStream);
// 关闭流。
fileOutputStream.close();
System.out.println("Excel 生成完毕。");
}
}
大文件写入。
package com.geek;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
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.xssf.streaming.SXSSFWorkbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.joda.time.DateTime;
import org.junit.Test;
import java.io.FileOutputStream;
import java.io.IOException;
public class ExcelWriteTest {
private final String PATH = "G:\\lyfGeek\\IdeaProjects\\poi_easyExcel_geek\\geek-poi";
// SXSSF 过程中会产生临时文件。
// 默认由 100 条记录被保存在内存中,超过则写入临时文件。
// new SXSSFWorkbook(数量); 自定义内在中的数量。
@Test
public void testWrite07BigDataS() throws IOException {
long start = System.currentTimeMillis();
// 创建一个工作薄。
Workbook workbook = new SXSSFWorkbook();
// 创建表。
Sheet sheet = workbook.createSheet();
// 写入数据。
for (int rowNum = 0; rowNum < 100000; rowNum++) {
Row row = sheet.createRow(rowNum);
for (int cellNum = 0; cellNum < 10; cellNum++) {
Cell cell = row.createCell(cellNum);
cell.setCellValue(cellNum);
}
}
System.out.println("ok");
FileOutputStream fileOutputStream = new FileOutputStream(PATH + "testWrite07BigDataS.xlsx");
workbook.write(fileOutputStream);
fileOutputStream.close();
// 清除临时文件。
((SXSSFWorkbook) workbook).dispose();
long end = System.currentTimeMillis();
System.out.println((double) (end + start) / 1000 + " s。");
// java.lang.IllegalArgumentException: Invalid row number (65536) outside allowable range (0..65535)
/*
ok
1.836 s。
Process finished with exit code 0
*/
}
// 耗时较长。行数没有限制。
// 优化:缓存。~SXSSF。
// SXSSF 过程中会产生临时文件。
// 默认由 100 条记录被保存在内存中,超过则写入临时文件。
// new SXSSFWorkbook(数量); 自定义内在中的数量。
@Test
public void testWrite07BigData() throws IOException {
long start = System.currentTimeMillis();
// 创建一个工作薄。
Workbook workbook = new XSSFWorkbook();
// 创建表。
Sheet sheet = workbook.createSheet();
// 写入数据。
for (int rowNum = 0; rowNum < 100000; rowNum++) {
Row row = sheet.createRow(rowNum);
for (int cellNum = 0; cellNum < 10; cellNum++) {
Cell cell = row.createCell(cellNum);
cell.setCellValue(cellNum);
}
}
System.out.println("ok");
FileOutputStream fileOutputStream = new FileOutputStream(PATH + "testWrite07BigData.xlsx");
workbook.write(fileOutputStream);
fileOutputStream.close();
long end = System.currentTimeMillis();
System.out.println((double) (end + start) / 1000 + " s。");
// java.lang.IllegalArgumentException: Invalid row number (65536) outside allowable range (0..65535)
/*
ok
11.518 s。
Process finished with exit code 0
*/
}
// HSSFWorkbook()
// 速度快,最多 65536 行。
@Test
public void testWrite03BigData() throws IOException {
long start = System.currentTimeMillis();
// 创建一个工作薄。
Workbook workbook = new HSSFWorkbook();
// 创建表。
Sheet sheet = workbook.createSheet();
// 写入数据。
for (int rowNum = 0; rowNum < 65536; rowNum++) {
Row row = sheet.createRow(rowNum);
for (int cellNum = 0; cellNum < 10; cellNum++) {
Cell cell = row.createCell(cellNum);
cell.setCellValue(cellNum);
}
}
System.out.println("ok");
FileOutputStream fileOutputStream = new FileOutputStream(PATH + "testWrite03BigData.xls");
workbook.write(fileOutputStream);
fileOutputStream.close();
long end = System.currentTimeMillis();
System.out.println((double) (end + start) / 1000 + " s。");
// java.lang.IllegalArgumentException: Invalid row number (65536) outside allowable range (0..65535)
/*
ok
1.651 s。
Process finished with exit code 0
*/
}
@Test
public void testWrite07() throws IOException {
// 创建一个工作薄。
Workbook workbook = new XSSFWorkbook();
// 创建一个工作表。
Sheet sheet = workbook.createSheet("Geek 统计表");
// 创建一行。
Row row1 = sheet.createRow(0);
// 创建一个单元格。
Cell cell11 = row1.createCell(0);
cell11.setCellValue("今日新增观众");
// (1, 2)
Cell cell12 = row1.createCell(1);
cell12.setCellValue(666);
// 第二行。(2, 1)
Row row2 = sheet.createRow(1);
Cell cell21 = row2.createCell(0);
cell21.setCellValue("统计时间");
// (2, 2)
Cell cell22 = row2.createCell(1);
String date = new DateTime().toString("yyyy-MM-dd HH:mm:ss");
cell22.setCellValue(date);
// 生成一张表。IO 流。03 版本 .xls。
FileOutputStream fileOutputStream = new FileOutputStream(PATH + "Geek 观众统计表07.xlsx");
workbook.write(fileOutputStream);
// 关闭流。
fileOutputStream.close();
System.out.println("Excel 生成完毕。");
}
@Test
public void testWrite03() throws IOException {
// 创建一个工作薄。
Workbook workbook = new HSSFWorkbook();
// 创建一个工作表。
Sheet sheet = workbook.createSheet("Geek 统计表");
// 创建一行。
Row row1 = sheet.createRow(0);
// 创建一个单元格。
Cell cell11 = row1.createCell(0);
cell11.setCellValue("今日新增观众");
// (1, 2)
Cell cell12 = row1.createCell(1);
cell12.setCellValue(666);
// 第二行。(2, 1)
Row row2 = sheet.createRow(1);
Cell cell21 = row2.createCell(0);
cell21.setCellValue("统计时间");
// (2, 2)
Cell cell22 = row2.createCell(1);
String date = new DateTime().toString("yyyy-MM-dd HH:mm:ss");
cell22.setCellValue(date);
// 生成一张表。IO 流。03 版本 .xls。
FileOutputStream fileOutputStream = new FileOutputStream(PATH + "Geek 观众统计表03.xls");
workbook.write(fileOutputStream);
// 关闭流。
fileOutputStream.close();
System.out.println("Excel 生成完毕。");
}
}
Excel 基本读取。
注意获取值的类型。
package com.geek;
import org.apache.poi.ss.usermodel.Cell;
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.xssf.usermodel.XSSFWorkbook;
import org.junit.Test;
import java.io.FileInputStream;
import java.io.IOException;
public class ExcelReadTest {
private final String PATH = "G:\\lyfGeek\\IdeaProjects\\poi_easyExcel_geek\\";
@Test
public void testRead07() throws IOException {
// 获取文件流。
FileInputStream fileInputStream = new FileInputStream(PATH + "geek-poiGeek 观众统计表07.xlsx");
// 创建一个工作薄。
Workbook workbook = new XSSFWorkbook(fileInputStream);
// 得到表。
Sheet sheet1 = workbook.getSheetAt(0);
// 得到行。
Row row = sheet1.getRow(0);
// 得到格。
Cell cell = row.getCell(1);
System.out.println(cell.getNumericCellValue());
// 666.0
// Cell cell = row.getCell(0);
// 字符串。
// System.out.println(cell.getStringCellValue());
// java.lang.IllegalStateException: Cannot get a text value from a numeric cell
fileInputStream.close();
}
@Test
public void testRead03() throws IOException {
// 获取文件流。
FileInputStream fileInputStream = new FileInputStream(PATH + "geek-poiGeek 观众统计表03.xls");
// 创建一个工作薄。
Workbook workbook = new XSSFWorkbook(fileInputStream);
// 得到表。
Sheet sheet1 = workbook.getSheetAt(0);
// 得到行。
Row row = sheet1.getRow(0);
// 得到格。
Cell cell = row.getCell(1);
System.out.println(cell.getNumericCellValue());
// 666.0
// Cell cell = row.getCell(0);
// 字符串。
// System.out.println(cell.getStringCellValue());
// java.lang.IllegalStateException: Cannot get a text value from a numeric cell
fileInputStream.close();
}
}
Cell 类型判断。
package com.geek;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.ss.usermodel.Cell;
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.xssf.usermodel.XSSFWorkbook;
import org.joda.time.DateTime;
import org.junit.Test;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Date;
public class ExcelReadTest {
private final String PATH = "G:\\lyfGeek\\IdeaProjects\\poi_easyExcel_geek\\geek-poi\\";
@Test
public void testCellType() throws IOException {
// 获取文件流。
FileInputStream fileInputStream = new FileInputStream(PATH + "明细表.xlsx");
// 创建一个工作薄。
Workbook workbook = new XSSFWorkbook(fileInputStream);
Sheet sheet = workbook.getSheetAt(0);
// 获取标题内容。
Row rowTitle = sheet.getRow(0);
if (rowTitle != null) {
int cellCount = rowTitle.getPhysicalNumberOfCells();
for (int cellNum = 0; cellNum < cellCount; cellNum++) {
Cell cell = rowTitle.getCell(cellNum);
if (cell != null) {
int cellType = cell.getCellType();
// System.out.println("cellType = " + cellType);//1
String stringCellValue = cell.getStringCellValue();
System.out.print(stringCellValue + " | ");
}
}
System.out.println();
}
// 获取表中的内容。
int rowCount = sheet.getPhysicalNumberOfRows();
for (int rowNum = 1; rowNum < rowCount; rowNum++) {
Row rowData = sheet.getRow(rowNum);
if (rowData != null) {
// 读取列。
int cellCount = rowData.getPhysicalNumberOfCells();
for (int cellNum = 0; cellNum < cellCount; cellNum++) {
System.out.print("[" + (rowNum + 1) + "-" + (cellNum + 1) + "]");
Cell cell = rowData.getCell(cellNum);
// 匹配类的类型。
if (cell != null) {
int cellType = cell.getCellType();
String cellValue = "";
switch (cellType) {
case HSSFCell.CELL_TYPE_NUMERIC:// = 0; 数字(日期、普通数字等)。
System.out.print("【~ ~ ~ CELL_TYPE ~ numeric ~ 0 ~ ~ ~】");
if (HSSFDateUtil.isCellDateFormatted(cell)) {
// 日期。
System.out.print("【~ ~ ~ CELL_TYPE ~ numeric ~ 0 ~ 日期。~ ~ ~】");
Date date = cell.getDateCellValue();
cellValue = new DateTime(date).toString("yyyy-MM-dd");
} else {
System.out.print("【~ ~ ~ CELL_TYPE ~ numeric ~ 0 ~ 转换为字符串输出。~ ~ ~】");
// 如果不是日期格式,防止数字过长。
cell.setCellType(HSSFCell.CELL_TYPE_STRING);
cellValue = cell.toString();
}
break;
case HSSFCell.CELL_TYPE_STRING:// = 1;
System.out.print("【~ ~ ~ CELL_TYPE ~ String ~ 1 ~ ~ ~】");
cellValue = cell.getStringCellValue();
break;
// case HSSFCell.CELL_TYPE_FORMULA:// = 2;
// System.out.println("~ ~ ~ CELL_TYPE ~ formula ~ 2 ~ ~ ~");
// cellValue = cell.getStringCellValue();
// break;
case HSSFCell.CELL_TYPE_BLANK:// = 3;
System.out.print("【~ ~ ~ CELL_TYPE ~ blank ~ 3 ~ ~ ~】");
break;
case HSSFCell.CELL_TYPE_BOOLEAN:// = 4;
System.out.print("【~ ~ ~ CELL_TYPE ~ boolean ~ 4 ~ ~ ~】");
cellValue = String.valueOf(cell.getBooleanCellValue());
break;
case HSSFCell.CELL_TYPE_ERROR:// = 5;
System.out.print("【~ ~ ~ CELL_TYPE ~ error ~ 5 数据类型错误。~ ~ ~】");
cellValue = String.valueOf(cell.getErrorCellValue());
break;
}
System.out.println("cellValue = " + cellValue);
}
}
}
fileInputStream.close();
}
}
@Test
public void testRead07() throws IOException {
// 获取文件流。
FileInputStream fileInputStream = new FileInputStream(PATH + "geek-poiGeek 观众统计表07.xlsx");
// 创建一个工作薄。
Workbook workbook = new XSSFWorkbook(fileInputStream);
// 得到表。
Sheet sheet1 = workbook.getSheetAt(0);
// 得到行。
Row row = sheet1.getRow(0);
// 得到格。
Cell cell = row.getCell(1);
System.out.println(cell.getNumericCellValue());
// 666.0
// Cell cell = row.getCell(0);
// 字符串。
// System.out.println(cell.getStringCellValue());
// java.lang.IllegalStateException: Cannot get a text value from a numeric cell
fileInputStream.close();
}
@Test
public void testRead03() throws IOException {
// 获取文件流。
FileInputStream fileInputStream = new FileInputStream(PATH + "geek-poiGeek 观众统计表03.xls");
// 创建一个工作薄。
Workbook workbook = new XSSFWorkbook(fileInputStream);
// 得到表。
Sheet sheet1 = workbook.getSheetAt(0);
// 得到行。
Row row = sheet1.getRow(0);
// 得到格。
Cell cell = row.getCell(1);
System.out.println(cell.getNumericCellValue());
// 666.0
// Cell cell = row.getCell(0);
// 字符串。
// System.out.println(cell.getStringCellValue());
// java.lang.IllegalStateException: Cannot get a text value from a numeric cell
fileInputStream.close();
}
}
计算公式。
package com.geek;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFFormulaEvaluator;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.joda.time.DateTime;
import org.junit.Test;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Date;
public class ExcelReadTest {
private final String PATH = "G:\\lyfGeek\\IdeaProjects\\poi_easyExcel_geek\\geek-poi\\";
@Test
public void testFormula() throws IOException {
FileInputStream fileInputStream = new FileInputStream(PATH + "公式.xlsx");
Workbook workbook = new XSSFWorkbook(fileInputStream);
Sheet sheet = workbook.getSheetAt(0);
Row row = sheet.getRow(4);
Cell cell = row.getCell(0);
System.out.println("cell = " + cell);
// SUM(A2:A4)
// 拿到计算公式。
XSSFFormulaEvaluator formulaEvaluator = (XSSFFormulaEvaluator) new XSSFFormulaEvaluator((XSSFWorkbook) workbook);
int cellType = cell.getCellType();
switch (cellType) {
case Cell.CELL_TYPE_FORMULA:
String cellFormula = cell.getCellFormula();
System.out.println("cellFormula = " + cellFormula);
// cellFormula = SUM(A2:A4)
// 计算。
CellValue evaluate = formulaEvaluator.evaluate(cell);
String formatAsString = evaluate.formatAsString();
System.out.println("formatAsString = " + formatAsString);
break;
}
}
}
EasyExcel。
https://github.com/alibaba/easyexcel
快速、简单避免 OOM 的 java 处理 Excel 工具。
- pom.xml。
<!-- https://mvnrepository.com/artifact/com.alibaba/easyexcel -->
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>easyexcel</artifactId>
<version>2.2.3</version>
</dependency>
- 实体类。
package com.geek.easyexcel;
import com.alibaba.excel.annotation.ExcelIgnore;
import com.alibaba.excel.annotation.ExcelProperty;
import lombok.Data;
import java.util.Date;
@Data
public class DemoData {
@ExcelProperty("字符串标题")
private String string;
@ExcelProperty("日期标题")
private Date date;
@ExcelProperty("数字标题")
private Double doubleData;
/**
* 忽略这个字段。
*/
@ExcelIgnore
private String ignore;
}
注意删去 POI 依赖,否则报错。
<!-- xsl 2003-->
<!-- https://mvnrepository.com/artifact/org.apache.poi/poi -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>3.9</version>
</dependency>
<!-- xlsx 2007-->
<!-- https://mvnrepository.com/artifact/org.apache.poi/poi-ooxml -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>3.9</version>
</dependency>
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>fastjson</artifactId>
<version>1.2.67</version>
</dependency>
写数据。
package com.geek.easyexcel;
import com.alibaba.excel.EasyExcel;
import org.junit.Test;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
public class EasyTest {
private final String PATH = "G:\\lyfGeek\\IdeaProjects\\poi_easyExcel_geek\\geek-poi";
private List<DemoData> data() {
List<DemoData> list = new ArrayList<DemoData>();
for (int i = 0; i < 10; i++) {
DemoData data = new DemoData();
data.setString("字符串" + i);
data.setDate(new Date());
data.setDoubleData(0.56);
list.add(data);
}
return list;
}
/**
* 最简单的写。
* <p>1. 创建 excel 对应的实体对象 参照{@link DemoData}
* <p>2. 直接写即可。
*/
@Test
public void simpleWrite() {
// ~~~~~~~写法 1。
// String fileName = TestFileUtil.getPath() + "simpleWrite" + System.currentTimeMillis() + ".xlsx";
String fileName = PATH + "simpleWrite" + System.currentTimeMillis() + ".xlsx";
// 这里需要指定写用哪个 class 去写,然后写到第一个 sheet,名字为模板。然后文件流会自动关闭。
// 如果这里想使用 03,则传入 excelType 参数即可。
EasyExcel.write(fileName, DemoData.class).sheet("模板").doWrite(data());
// ~~~~~~~写法 2。
// fileName = TestFileUtil.getPath() + "simpleWrite" + System.currentTimeMillis() + ".xlsx";
// String fileName = PATH + "simpleWrite" + System.currentTimeMillis() + ".xlsx";
// 这里 需要指定写用哪个 class 去写。
/*
ExcelWriter excelWriter = EasyExcel.write(fileName, DemoData.class).build();
WriteSheet writeSheet = EasyExcel.writerSheet("模板").build();
excelWriter.write(data(), writeSheet);
*/
// 千万别忘记 finish 会帮忙关闭流。
// excelWriter.finish();
}
}
读数据。
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>fastjson</artifactId>
<version>1.2.67</version>
</dependency>
package com.geek.easyexcel;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.fastjson.JSON;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.ArrayList;
import java.util.List;
// 有个很重要的点。DemoDataListener 不能被 spring 管理,要每次读取 excel 都要 new,然后里面用到 spring 可以构造方法传进去。
public class DemoDataListener extends AnalysisEventListener<DemoData> {
private static final Logger LOGGER = LoggerFactory.getLogger(DemoDataListener.class);
/**
* 每隔 5 条存储数据库,实际使用中可以 3000 条,然后清理 list,方便内存回收。
*/
private static final int BATCH_COUNT = 5;
List<DemoData> list = new ArrayList<DemoData>();
/**
* 假设这个是一个 DAO,当然有业务逻辑这个也可以是一个 service。当然如果不用存储这个对象没用。
*/
private DemoDAO demoDAO;
public DemoDataListener() {
// 这里是 demo,所以随便 new 一个。实际使用如果到了 spring,请使用下面的有参构造函数。
demoDAO = new DemoDAO();
}
/**
* 如果使用了 spring,请使用这个构造方法。每次创建 Listener 的时候需要把 spring 管理的类传进来。
*
* @param demoDAO
*/
public DemoDataListener(DemoDAO demoDAO) {
this.demoDAO = demoDAO;
}
/**
* 这个每一条数据解析都会来调用。
*
* @param data one row value. Is is same as {@link AnalysisContext#readRowHolder()}
* @param context
*/
// 读取数据时执行此方法。
@Override
public void invoke(DemoData data, AnalysisContext context) {
LOGGER.info("解析到一条数据:{}", JSON.toJSONString(data));
list.add(data);
// 达到 BATCH_COUNT 了,需要去存储一次数据库,防止数据几万条数据在内存,容易 OOM。
if (list.size() >= BATCH_COUNT) {
saveData();
// 存储完成清理 list。
list.clear();
}
}
/**
* 所有数据解析完成了 都会来调用。
*
* @param context
*/
@Override
public void doAfterAllAnalysed(AnalysisContext context) {
// 这里也要保存数据,确保最后遗留的数据也存储到数据库
saveData();
LOGGER.info("所有数据解析完成!");
}
/**
* 加上存储数据库。
*/
private void saveData() {
LOGGER.info("{}条数据,开始存储数据库!", list.size());
demoDAO.save(list);
LOGGER.info("存储数据库成功!");
}
}
package com.geek.easyexcel;
import com.alibaba.excel.EasyExcel;
import org.junit.Test;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
public class EasyTest {
private final String PATH = "G:\\lyfGeek\\IdeaProjects\\poi_easyExcel_geek\\geek-poi";
// 读取。
/**
* 最简单的读
* <p>1. 创建excel对应的实体对象 参照{@link DemoData}
* <p>2. 由于默认一行行的读取excel,所以需要创建excel一行一行的回调监听器,参照{@link DemoDataListener}
* <p>3. 直接读即可
*/
@Test
public void simpleRead() {
// 有个很重要的点 DemoDataListener 不能被spring管理,要每次读取excel都要new,然后里面用到spring可以构造方法传进去
// 写法 1。
// String fileName = TestFileUtil.getPath() + "demo" + File.separator + "demo.xlsx";
// String fileName = PATH + "demo.xlsx";
String fileName = "G:\\lyfGeek\\IdeaProjects\\poi_easyExcel_geek\\geek-poisimpleWrite1588424469263.xlsx";
// 这里 需要指定读用哪个 class 去读,然后读取第一个 sheet 文件流会自动关闭。
EasyExcel.read(fileName, DemoData.class, new DemoDataListener()).sheet().doRead();
// 写法2。
// fileName = TestFileUtil.getPath() + "demo" + File.separator + "demo.xlsx";
// ExcelReader excelReader = EasyExcel.read(fileName, DemoData.class, new DemoDataListener()).build();
// ReadSheet readSheet = EasyExcel.readSheet(0).build();
// excelReader.read(readSheet);
// 这里千万别忘记关闭,读的时候会创建临时文件,到时磁盘会崩的
// excelReader.finish();
}
private List<DemoData> data() {
List<DemoData> list = new ArrayList<DemoData>();
for (int i = 0; i < 10; i++) {
DemoData data = new DemoData();
data.setString("字符串" + i);
data.setDate(new Date());
data.setDoubleData(0.56);
list.add(data);
}
return list;
}
/**
* 最简单的写。
* <p>1. 创建 excel 对应的实体对象 参照{@link DemoData}
* <p>2. 直接写即可。
*/
@Test
public void simpleWrite() {
// ~~~~~~~写法 1。
// String fileName = TestFileUtil.getPath() + "simpleWrite" + System.currentTimeMillis() + ".xlsx";
String fileName = PATH + "simpleWrite" + System.currentTimeMillis() + ".xlsx";
// 这里需要指定写用哪个 class 去写,然后写到第一个 sheet,名字为模板。然后文件流会自动关闭。
// 如果这里想使用 03,则传入 excelType 参数即可。
// EasyExcel.write(fileName, 格式类).sheet("模板").doWrite(数据);
EasyExcel.write(fileName, DemoData.class).sheet("模板").doWrite(data());
// ~~~~~~~写法 2。
// fileName = TestFileUtil.getPath() + "simpleWrite" + System.currentTimeMillis() + ".xlsx";
// String fileName = PATH + "simpleWrite" + System.currentTimeMillis() + ".xlsx";
// 这里 需要指定写用哪个 class 去写。
/*
ExcelWriter excelWriter = EasyExcel.write(fileName, DemoData.class).build();
WriteSheet writeSheet = EasyExcel.writerSheet("模板").build();
excelWriter.write(data(), writeSheet);
*/
// 千万别忘记 finish 会帮忙关闭流。
// excelWriter.finish();
}
}