问题描述
现需要批量导入数据,数据以 Excel 形式导入。
POI 介绍
我选择使用的是 apache POI。这是有 Apache 软件基金会开放的函数库,他会提供 API 给 java,使其可以对 office 文件进行读写。
我这里只需要使用其中的 Excel 部分。
实现
首先,Excel 有两种格式,一种是.xls(03 版), 另一种是.xlsx(07 版)。针对两种不同的表格格式,POI 对应提供了两种接口。HSSFWorkbook 和 XSSFWorkbook
导入依赖
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>RELEASE</version>
</dependency>
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>RELEASE</version>
</dependency>
处理版本
Workbook workbook = null;
try {
if (file.getPath().endsWith(“xls”)) {
System.out.println(“ 这是 2003 版本 ”);
workbook = new XSSFWorkbook(new FileInputStream(file));
} else if (file.getPath().endsWith(“xlsx”)){
workbook = new HSSFWorkbook(new FileInputStream(file));
System.out.println(“ 这是 2007 版本 ”);
}
} catch (IOException e) {
e.printStackTrace();
}
这里需要判断一下 Excel 的版本,根据扩展名,用不同的类来处理文件。
获取表格数据
获取表格中的数据分为以下几步:
1. 获取表格 2. 获取某一行 3. 获取这一行中的某个单元格
代码实现:
// 获取第一个张表
Sheet sheet = workbook.getSheetAt(0);
// 获取每行中的字段
for (int i = 0; i <= sheet.getLastRowNum(); i++) {
Row row = sheet.getRow(i); // 获取行
// 获取单元格中的值
String studentNum = row.getCell(0).getStringCellValue();
String name = row.getCell(1).getStringCellValue();
String phone = row.getCell(2).getStringCellValue();
}
持久化
获取出单元格中的数据后,最后就是用数据建立对象了。
List<Student> studentList = new ArrayList<>();
for (int i = 0; i <= sheet.getLastRowNum(); i++) {
Row row = sheet.getRow(i); // 获取行
// 获取单元格中的值
String studentNum = row.getCell(0).getStringCellValue();
String name = row.getCell(1).getStringCellValue();
String phone = row.getCell(2).getStringCellValue();
Student student = new Student();
student.setStudentNumber(studentNum);
student.setName(name);
student.setPhoneNumber(phone);
studentList.add(student);
}
// 持久化
studentRepository.saveAll(studentList);
相关参考:https://poi.apache.org/compon…https://blog.csdn.net/daihuim…