Java使用POI从Excel读取数据并存入数据库(解决读取到空行问题)
作者:大龙Java
项目场景:
需要从Excel读取多行数据,映射到java对象,并存入到数据库当中(单Sheet版)
问题分析
例如:读取数据映射到java实体类的时候,发现了有一条数据的属性都为null,导致一些方法的空指针异常。通过debug我们发现,实际上Excel当中数据有4条(包括header),但是在使用poi读取并映射到java实体类的时候却映射了5条。
解决方案:
既然最后一条数据读到的是没有数据的一行,我们就可以在遍历excel数据的时候加一条判断,每一行当中每一个单元格的通过poi读取后,在java中岗位cell类型的对象,只需判断是否为null,若为null直接结束遍历即可。
其中最外层循环,为遍历每行数据。内层循环为遍历每行数据当中的单元格Cell,内层循环的数量是根据第一行(header头部标题)的数量来决定的。
完整代码:
maven先导入poi相关jar包,本人使用的是 4.1.2 版本。
<dependency> <groupId>org.apache.poi</groupId> <artifactId>poi-ooxml</artifactId> <version>${poi.version}</version> </dependency> <dependency> <groupId>org.apache.poi</groupId> <artifactId>poi</artifactId> <version>${poi.version}</version> </dependency> <dependency> <groupId>org.apache.poi</groupId> <artifactId>poi-scratchpad</artifactId> <version>${poi.version}</version> </dependency>
该代码为单sheet版本,若为多sheet版本只需在外层加一层for循环遍历sheet即可。
若Excel表中数据为日期,可以使用cell.getDateCellValue()来获取,cell提供的方法很多,可以根据具体情况来进行获取。
import org.apache.poi.ss.usermodel.*; import org.apache.poi.xssf.usermodel.XSSFWorkbook; import org.springblade.busi.entity.PersonalInfo; import org.springblade.busi.framework.annotation.InfoChangeAnnotation; import org.springframework.web.multipart.MultipartFile; import java.io.IOException; import java.io.InputStream; import java.lang.reflect.Field; import java.util.ArrayList; import java.util.Iterator; import java.util.List; public class ExcelToObject1 { /** * 读取Excel文件,版本:只读取基本信息一个Sheet,若有需要多个sheet在增加for循环 * * @param file * @return */ public List<PersonalInfo> toPersonalInfo(MultipartFile file) { //创建返回对象 List<PersonalInfo> persons = new ArrayList<>(); try (InputStream fis = file.getInputStream(); Workbook workbook = new XSSFWorkbook(fis)) { Class<?> clazz = PersonalInfo.class; //只读取基本信息一个Sheet Sheet sheet = workbook.getSheetAt(0); Row headerRow = sheet.getRow(0); List<String> columnNames = new ArrayList<>(); Iterator<Cell> cellIterator = headerRow.cellIterator(); while (cellIterator.hasNext()) { Cell cell = cellIterator.next(); columnNames.add(cell.getStringCellValue()); } for (int rowIndex = 1; rowIndex <= sheet.getLastRowNum(); rowIndex++) { Row row = sheet.getRow(rowIndex); PersonalInfo person = (PersonalInfo) clazz.newInstance(); //获取Excel当前行的第一位,若未null则不处理 ---> 防止poi读取Excel多读取没有数据的一行,导致非空字段为空。 Cell cellNow = row.getCell(0); if (cellNow == null) { break; } for (int columnIndex = 0; columnIndex < columnNames.size(); columnIndex++) { Cell cell = row.getCell(columnIndex); String columnName = columnNames.get(columnIndex); //通过实体类注解与Excel头部中文进行映射 Field field = getFieldByColumnName(clazz, columnName); if (field != null) { field.setAccessible(true); field.set(person, getCellValueAsString(cell)); } } //处理每一行Excel加到返回对象当中 persons.add(person); } } catch (IOException | IllegalAccessException | InstantiationException e) { e.printStackTrace(); System.out.println(e.getMessage()); } return persons; } //通过实体类注解与Excel头部中文进行映射 private static Field getFieldByColumnName(Class<?> clazz, String columnName) { for (Field field : clazz.getDeclaredFields()) { InfoChangeAnnotation annotation = field.getAnnotation(InfoChangeAnnotation.class); if (annotation != null && annotation.value().equals(columnName)) { return field; } } return null; } //获取每个单元格数据并转为String,其他类型在上方做了ifelse private static String getCellValueAsString(Cell cell) { if (cell == null) { return null; } cell.setCellType(CellType.STRING); String cellValue = cell.getStringCellValue(); System.out.println("getCellValueAsString:" + cellValue); return cellValue; }
在映射java实体类的时候需要在其实体类属性添加注解,注解可以自行创建,也可以引用其他带有value的注解,本文章使用的是 @InfoChangeAnnotation("Excel表中第一行的中文")。直接可以与Excel表头进行映射。 上述代码中PersonalInfo为需要映射到java的实体类,必须根据实际情况进行修改。
@InfoChangeAnnotation("姓名") private String personalName;
总结
到此这篇关于Java使用POI从Excel读取数据并存入数据库的文章就介绍到这了,更多相关Java读取Excel数据存入数据库内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!