Alibaba EasyExcel初体验

简介

EasyExcel是一个基于Java的简单、省内存的读写Excel的开源项目。在尽可能节约内存的情况下支持读写百M的Excel。相对于Apache POI来说,EasyExcel是从磁盘上一行行的读取数据,然后逐个解析,避免将大量数据加载到内存从而导致OOM。

相关依赖:

<dependency>
	<groupId>com.alibaba</groupId>
	<artifactId>easyexcel</artifactId>
	<version>2.2.6</version>
</dependency>

<!-- 可选 -->
<dependency>
	<groupId>org.projectlombok</groupId>
	<artifactId>lombok</artifactId>
	<version>1.18.12</version>
	<scope>provided</scope>
</dependency>

Excel导入

需要导入的数据:

Model类:


@Data
public class SchoolData {

    @ExcelProperty(value = "地址")
    private String address;

    @ExcelProperty(value = "学校名称")
    private String schoolName;

    @ExcelProperty(value = "邮政编码")
    private String postcode;

    @ExcelProperty(value = "收集时间")
    private Date collectDate;
}

使用@ExcelProperty注解时,可以使用两种方式将数据列和属性进行绑定:

  1. 通过value指定名称,将名称对应的excel列的数据绑定到属性。
  2. 通过index指定下标,将下标对应的excel列的数据绑定到属性。

在不使用@ExcelProperty注解时,默认Model类属性和excel列数据按照先后顺序进行绑定的。

监听器:

import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class SchoolDataListener extends AnalysisEventListener<SchoolData> {

    List<SchoolData> schoolDataList = new ArrayList<>();

    //需要传参时,通过构造方法传进来
    public SchoolDataListener() {
    }

    /**
     * 每解析一行数据都会进行调用
     */
    @Override
    public void invoke(SchoolData data, AnalysisContext context) {
        //一般数据量大的时候应该分批处理,防止数据全部加载到内存,导致OOM。
        schoolDataList.add(data);
    }

    /**
     * 数据全部解析完成时调用
     */
    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
        System.out.println("数据读取完毕:");
        schoolDataList.stream().forEach(System.out::println);
    }

    /**
     * 每解析一行表头都会进行调用
     */
    @Override
    public void invokeHeadMap(Map<Integer, String> headMap, AnalysisContext context) {
        System.out.println("表头:");
        headMap.forEach((k, v) -> System.out.println(v));
        System.out.println();
    }
}

导入测试:

导入有以下两种写法,一般使用简写的方式就行,当需要实现复杂的功能时可能会需要另外一种复杂的写法。

@Test
public void importExcel() {
	ExcelReader excelReader = null;
	try {
		excelReader = EasyExcel.read("F:\\测试数据.xlsx", SchoolData.class, new SchoolDataListener()).build();
		ReadSheet readSheet = EasyExcel.readSheet(0).build();
		excelReader.read(readSheet);
	} finally {
		if (excelReader != null) {
			// 这里千万别忘记关闭,读的时候会创建临时文件,到时磁盘会崩的
			excelReader.finish();
		}
	}
}

@Test
public void importExcelSimple() {
	//sheet() 默认读取第一个sheet
	EasyExcel.read("F:\\测试数据.xlsx", SchoolData.class, new SchoolDataListener()).sheet().doRead();
}

输出结果:

表头:
学校名称
地址
邮政编码
收集时间

数据读取完毕:
SchoolData(address=晋宁县古城镇旧寨村  , schoolName=晋宁县古城镇古城中学, postcode=650604, collectDate=Sun Jan 05 00:00:00 CST 2020)
SchoolData(address=云南省昆明市晋宁县二街乡老高村  , schoolName=晋宁县二街中学, postcode=650608, collectDate=Mon Jan 06 00:00:00 CST 2020)
SchoolData(address=云南省昆明市晋宁县晋城镇寺林街  , schoolName=晋宁县晋城中学, postcode=650605, collectDate=Tue Jan 07 00:00:00 CST 2020)
SchoolData(address=云南省晋宁县六街乡中学  , schoolName=晋宁县六街中学, postcode=650609, collectDate=Wed Jan 08 00:00:00 CST 2020)
SchoolData(address=云南省昆明市晋宁县新街乡文河村  , schoolName=晋宁县新街中学, postcode=650606, collectDate=Thu Jan 09 00:00:00 CST 2020)
SchoolData(address=晋宁县中和乡普照路  , schoolName=晋宁县中和中学, postcode=650600, collectDate=Fri Jan 10 00:00:00 CST 2020)
SchoolData(address=云南省昆明市晋宁县上蒜乡上蒜中学  , schoolName=晋宁县上蒜乡上蒜中学, postcode=650607, collectDate=Sat Jan 11 00:00:00 CST 2020)
SchoolData(address=云南省昆明市晋宁县化乐乡化乐村  , schoolName=晋宁县化乐乡中学, postcode=650611, collectDate=Sun Jan 12 00:00:00 CST 2020)
SchoolData(address=晋宁县夕阳乡夕阳街夕阳民族中学  , schoolName=晋宁县夕阳民族中学, postcode=650603, collectDate=Mon Jan 13 00:00:00 CST 2020)
SchoolData(address=云南省昆明市晋宁县宝峰镇古城村委会小古城村  , schoolName=晋宁县宝峰镇宝峰中学, postcode=650601, collectDate=Tue Jan 14 00:00:00 CST 2020)
SchoolData(address=晋宁县双河乡椿树营  , schoolName=晋宁县双河民族中学, postcode=650602, collectDate=Wed Jan 15 00:00:00 CST 2020)
SchoolData(address=昆明市富民县永定镇环城南路7号永定中学  , schoolName=富民县永定中学, postcode=650400, collectDate=Thu Jan 16 00:00:00 CST 2020)
SchoolData(address=富民县永定镇大西山村  , schoolName=富民县勤劳中学, postcode=650400, collectDate=Fri Jan 17 00:00:00 CST 2020)
SchoolData(address=富民县大营镇大营街16号  , schoolName=富民县大营中学, postcode=650400, collectDate=Sat Jan 18 00:00:00 CST 2020)
SchoolData(address=云南省昆明市罗免乡者北村委会者北街  , schoolName=富民县第二中学, postcode=650401, collectDate=Sun Jan 19 00:00:00 CST 2020)

Excel导出

模拟生成数据:

public class DataGenerate {

    /**
     * 数据生成
     */
    public static List<SchoolData> data() {
        List<SchoolData> schoolDataList = new ArrayList<>();
        SchoolData schoolData;
        for (int i = 0; i < 15; i++) {
            schoolData = new SchoolData();
            schoolData.setSchoolName("第" + i + "XXX中学");
            schoolData.setAddress("云南省昆明市XXXXX");
            schoolData.setPostcode("123456");
            schoolData.setCollectDate(new Date());
            schoolDataList.add(schoolData);
        }

        return schoolDataList;
    }
}

测试导出:

同导入,导出也有以下两种写法。LongestMatchColumnWidthStyleStrategy用来设置自动列宽,如果需要设置列宽、行高等属性,可以在Model类上添加对应注解进行设置,如@ColumnWidth、@ContentRowHeight、@HeadRowHeight等等。

@Test
public void exportExcel() {
	ExcelWriter excelWriter = null;
	try {
		excelWriter = EasyExcel.write("F:\\测试数据_副本.xlsx", SchoolData.class).build();
		WriteSheet writeSheet = EasyExcel.writerSheet("学校数据").registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).build();
		excelWriter.write(DataGenerate.data(), writeSheet);
	} finally {
		// 千万别忘记finish 会帮忙关闭流
		if (excelWriter != null) {
			excelWriter.finish();
		}
	}
}

@Test
public void exportExcelSimple() {
	EasyExcel.write("F:\\测试数据_副本.xlsx", SchoolData.class).registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).sheet("学校数据").doWrite(DataGenerate.data());
}

导出结果:

基于Excel模板的填充

模板:

{} 代表普通变量 {.} 代表是list的变量。

模板填充测试:

@Test
public void excelTemplateFill() {
	ExcelWriter excelWriter = EasyExcel.write("F:\\测试数据_副本3.xlsx").withTemplate("F:\\模板.xlsx").build();
	WriteSheet writeSheet = EasyExcel.writerSheet().build();
	FillConfig fillConfig = FillConfig.builder().forceNewRow(Boolean.TRUE).build();

	//填充列表数据
	excelWriter.fill(DataGenerate.data(), fillConfig, writeSheet);

	//填充普通数据
	Map<String, Object> map = new HashMap<String, Object>();
	map.put("name", "狗子");
	map.put("date", "2021-02-19");
	excelWriter.fill(map, writeSheet);
	excelWriter.finish();
}

填充结果:

Web中的导入和导出

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;

@RestController
public class SchoolDataController {

    /**
     * 导入Excel
     */
    @PostMapping("/import")
    public String importExcel(MultipartFile file) throws IOException {
        EasyExcel.read(file.getInputStream(), SchoolData.class, new SchoolDataListener()).sheet().doRead();
        return "success";
    }

    /**
     * 导出Excel
     */
    @GetMapping("/export")
    public void exportExcel(HttpServletResponse response) throws IOException {
        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("utf-8");
        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
        String fileName = URLEncoder.encode("测试", "UTF-8").replaceAll("\\+", "%20");
        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
        EasyExcel.write(response.getOutputStream(), SchoolData.class).registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).sheet("学校数据").doWrite(DataGenerate.data());
    }
}

参考:官方文档

posted @ 2021-02-22 20:21  布禾  阅读(1061)  评论(2编辑  收藏  举报