时间:2020-11-20 17:33:40 | 栏目:JAVA代码 | 点击:次
问题描述
现需要批量导入数据,数据以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);