Java使用POI从Excel读取数据,解决读取到空行问题,并存入数据库。

项目场景:需要从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;