Java, 对Excel解析的封装

最终的调用方法

List<ExcelPOJO> result = new ArrayList<>();
ResolveExcel<ExcelPOJO> resolveExcel = new ResolveExcel<ExcelPOJO>(file, ExcelPOJO.class);
resolveExcel.forEach((item)->{
	result.add((ExcelPOJO) item);
});

源代码

Excel对应的实体类

package com.guoz.excel;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import java.io.Serializable;

/**
* @Author guozhi
* @Date 2019/7/30 15:28
* @Description TODO
*/

public class ExcelPOJO implements Serializable, IConvertRowToEntity<ExcelPOJO> {

   private int first;
   private String two;

   public ExcelPOJO() {
   }

   public int getFirst() {
       return first;
   }

   public void setFirst(int first) {
       this.first = first;
   }

   public String getTwo() {
       return two;
   }

   public void setTwo(String two) {
       this.two = two;
   }

   /**
    * Excel表数据行操作
    *
    * @param row Row类
    * @return 实体类
    */
   @Override
   public ExcelPOJO getEntity(Row row) {
       if (row == null)
           return null;
       Cell cell;
       if ( (cell =  row.getCell(0)) != null) {
           // todo 设置成数值类型
           cell.setCellType(Cell.CELL_TYPE_NUMERIC);
           setFirst((int) cell.getNumericCellValue());
       }
       if ((cell = row.getCell(1)) != null){
           // todo 设置成字符串类型
           cell.setCellType(Cell.CELL_TYPE_STRING);
           setTwo(cell.getStringCellValue());
       }
       return this;
   }

   @Override
   public String toString() {
       return "ExcelPOJO{" +
               "first=" + first +
               ", two='" + two + '\'' +
               '}';
   }
}

实体类要实现的接口

package com.guoz.excel;
import org.apache.poi.ss.usermodel.Row;
/**
* 根据excel表的数据行 Row 获取实体类
* @param <T>
*/
public interface IConvertRowToEntity<T> {
   /**
    * Excel表数据行操作
    * @param row Row类
    * @return 实体类
    */
    T getEntity(Row row);
}

核心封装代码

package com.guoz.excel;


import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

import java.lang.reflect.Array;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;

/**
* 解析Excel文件
* gz
*
* @param <T>
*/
public class ResolveExcel<T> implements Iterable {

  private static final String SUFFIX_2003 = ".xls";
  private static final String SUFFIX_2007 = ".xlsx";

  private Workbook workbook;

  // 表格总数
  private int numOfSheet = 0;
  // 当前遍历的表格数目
  private int positionOfSheet = 0;
  // 当前的表格
  private Sheet sheet;

  // 当前表的总行数
  private int lastNumOfRow = 0;
  // 当前表遍历的行数
  private int positionOfRow = 1;

  private Class clazz;

  /**
   * @param file excel文件对象
   * @param clazz 实现了IConvertRowToEntity 接口的实体类
   */
  public ResolveExcel(MultipartFile file, Class clazz ) throws Exception {
      // todo 判断目标实体类是否实现了 IConvertRowToEntity 接口
      if (! (Arrays.asList(clazz.getInterfaces()).contains(IConvertRowToEntity.class))) {
          throw new Exception("实体类未实现 IConvertRowToEntity 接口 ");
      }
      this.clazz = clazz;
      String originalFilename = file.getOriginalFilename();
      // todo 检验文件格式
      if (originalFilename.endsWith(SUFFIX_2003)) {
          workbook = new HSSFWorkbook(file.getInputStream());
      } else if (originalFilename.endsWith(SUFFIX_2007)) {
          workbook = new XSSFWorkbook(file.getInputStream());
      }
      if (workbook == null)
          throw new Exception("文件格式错误");
      // todo 获取所有的工作表的的数量
      numOfSheet = workbook.getNumberOfSheets();
      if (numOfSheet <= 0)
          throw new Exception("文件不能为空！");
      // todo 获取第一张表格信息
      sheet = workbook.getSheetAt(positionOfSheet++);
      lastNumOfRow = sheet.getLastRowNum();
  }

  @Override
  public Iterator iterator() {
      return new Inner();
  }

  class Inner implements Iterator<T> {
      @Override
      public boolean hasNext() {
          return (positionOfRow <= lastNumOfRow || positionOfSheet < numOfSheet);
      }

      @Override
      public T next() {
          if (!hasNext())
              return null;
          Row row = null;
          if (positionOfRow <= lastNumOfRow) {
              row = sheet.getRow(positionOfRow++);
          } else if (positionOfSheet < numOfSheet) {
              positionOfRow = 1;
              sheet = workbook.getSheetAt(positionOfSheet++);
              lastNumOfRow = sheet.getLastRowNum();
              row = sheet.getRow(positionOfRow++);
          }
          try {
              return (T) ((IConvertRowToEntity) clazz.newInstance()).getEntity(row);
          } catch (InstantiationException | IllegalAccessException e) {
              e.printStackTrace();
          }
          return null;
      }
  }
}

依赖的jar包

        <!--解析Excel-->
       <dependency>
           <groupId>org.apache.poi</groupId>
           <artifactId>poi</artifactId>
           <version>3.14</version>
       </dependency>
       <dependency>
           <groupId>org.apache.poi</groupId>
           <artifactId>poi-ooxml</artifactId>
           <version>3.14</version>
       </dependency>
       <dependency>
           <groupId>org.apache.poi</groupId>
           <artifactId>poi-ooxml-schemas</artifactId>
           <version>3.14</version>
       </dependency>

完整Demo项目

项目的git地址