poi解析excel文件(通用方法)

import java.util.List;
/**
 * @author jungle
 * @description
 */
public interface BaseExcelParseModel {

    List<String> listColumnName();

    Integer getSheetNumber();

    Integer getBeginRowNumber();

}

import org.apache.commons.lang3.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
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 java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;

/**
 * @author jungle
 * @create 2018/4/26 下午3:42
 * @description excel解析
 */
public class ExcelParseUtil {

    private static final Log log = LogFactory.getLog(ExcelParseUtil.class);

    public static <T extends BaseExcelParseModel> List<T> parseExcelFile(File file, Class<T> clazz) {
        if (null == file) {
            throw new ApplicationException("文件不存在");
        }
        String fileName = file.getName();
        InputStream inputStream;
        try {
            inputStream = new FileInputStream(file);
        } catch (IOException e) {
            log.error("文件转输入流失败,文件路径:{},失败原因:{}", file.getAbsoluteFile(), e.getMessage());
            throw new ApplicationException("文件出错,请尝试重新上传");
        }
        return parseExcelInputStream(inputStream, fileName, clazz);
    }

    public static <T extends BaseExcelParseModel> List<T> parseExcelInputStream(InputStream inputStream, String fileName, Class<T> clazz) {
        Workbook workbook = getWorkbook(inputStream, fileName);
        try {
            T excelParseModel = clazz.newInstance();
            Integer sheetNumber = excelParseModel.getSheetNumber();
            if (sheetNumber == null || sheetNumber < 0) {
                sheetNumber = 0;
            }
            Sheet sheet = workbook.getSheetAt(sheetNumber);
            if (sheet == null) {
                return null;
            }
            List<T> results = new ArrayList<>(sheet.getLastRowNum());
            Integer beginRowNumber = excelParseModel.getBeginRowNumber();
            if (beginRowNumber == null || beginRowNumber < 0) {
                beginRowNumber = 0;
            }
            for (int rowNumber = beginRowNumber; rowNumber <= sheet.getLastRowNum(); rowNumber++) {
                Row row = sheet.getRow(rowNumber);
                if (row == null) {
                    continue;
                }
                T model = convertRow2Model(row, clazz);
                if (model != null) {
                    results.add(model);
                }
            }
            return results;
        } catch (Exception e) {
            log.error("文件解析失败,失败原因:{}", e.getMessage());
        } finally {
            try {
                if (inputStream != null) {
                    inputStream.close();
                }
            } catch (IOException ioException) {
                log.warn("输入流关闭失败,失败原因:{}", ioException.getMessage());
            }
        }
        throw new ApplicationException("文件解析失败,请重新尝试");
    }

    private static Workbook getWorkbook(InputStream inputStream, String fileName) {
        try {
            if (StringUtils.isNotEmpty(fileName)) {
                int length = fileName.length();
                String fileSuffix;
                if (length > 5) {
                    fileSuffix = fileName.substring(length - 5, length);
                } else {
                    fileSuffix = fileName;
                }
                if (fileSuffix.toUpperCase().endsWith(".XLS")) {
                    //2003
                    return new HSSFWorkbook(inputStream);
                } else if (fileSuffix.toUpperCase().endsWith(".XLSX")) {
                    //2007
                    return new XSSFWorkbook(inputStream);
                }
            }
        } catch (Exception e) {
            log.error("创建Workbook工作薄对象失败,失败原因:{}", e.getMessage());
        }
        log.error("文件格式不支持,文件名:{}", fileName);
        throw new ApplicationException("该文件格式不支持,暂支持xls、xlsx文件");
    }

    private static <T extends BaseExcelParseModel> T convertRow2Model(Row row, Class<T> clazz) {
        try {
            T excelParseModel = clazz.newInstance();
            List<String> columnNameList = excelParseModel.listColumnName();
            if (CollectionUtils.isNotEmpty(columnNameList)) {
                for (int i = 0; i < columnNameList.size(); i++) {
                    Cell cell = row.getCell(i);
                    setValue(excelParseModel, columnNameList.get(i), getCellValue(cell));
                }
            }
            return excelParseModel;
        } catch (Exception e) {

        }
        return null;
    }

    /**
     * 给对象设值
     *
     * @param object    对象
     * @param fieldName 属性名
     * @param value     值
     * @return
     */
    private static Object setValue(Object object, String fieldName, String value) {
        String setterMethodName = getSetterMethodNameByFieldName(fieldName);
        try {
            Method setterMethod = object.getClass().getDeclaredMethod(setterMethodName, object.getClass().getField(fieldName).getType());
            return setterMethod.invoke(object, value);
        } catch (Exception e) {
            log.info("{}类中反射{}方法发生异常!异常:{}", object.getClass(), setterMethodName, e.getMessage());
            return "";
        }
    }

    /**
     * 根据 属性名 得到 setter方法名
     *
     * @param fieldName 属性名
     * @return
     */
    private static String getSetterMethodNameByFieldName(String fieldName) {
        StringBuilder methodName = new StringBuilder("set");
        methodName.append(fieldName.substring(0, 1).toUpperCase()).append(fieldName.substring(1));
        return methodName.toString();
    }

    private static String getCellValue(Cell cell) {
        String cellValue = "";
        if (cell == null) {
            return cellValue;
        }
        //把数字当成String来读,避免出现1读成1.0的情况
        if (cell.getCellType() == Cell.CELL_TYPE_NUMERIC) {
            cell.setCellType(Cell.CELL_TYPE_STRING);
        }
        //判断数据的类型
        switch (cell.getCellType()) {
            case Cell.CELL_TYPE_NUMERIC:
                cellValue = String.valueOf(cell.getNumericCellValue());
                break;
            case Cell.CELL_TYPE_STRING:
                cellValue = String.valueOf(cell.getStringCellValue());
                break;
            case Cell.CELL_TYPE_BOOLEAN:
                cellValue = String.valueOf(cell.getBooleanCellValue());
                break;
            case Cell.CELL_TYPE_FORMULA:
                cellValue = String.valueOf(cell.getCellFormula());
                break;
            case Cell.CELL_TYPE_BLANK:
                cellValue = "";
                break;
            case Cell.CELL_TYPE_ERROR:
                cellValue = "非法字符";
                break;
            default:
                cellValue = "未知类型";
                break;
        }
        return cellValue;
    }

}

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值