文档章节

ExcleTransfer

Curry_M
 Curry_M
发布于 2017/04/07 15:30
字数 706
阅读 2
收藏 0
package com.pingan.common.transfer;

import java.io.IOException;
import java.io.InputStream;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.lang.ArrayUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
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.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import com.pingan.common.model.BaseEntity;

/**
 * @author chinwer
 * @created 2014-5-20
 */
public abstract class ExcelTransfer<T extends BaseEntity> {

    public static final String OFFICE_EXCEL_2003_POSTFIX = "xls";
    public static final String OFFICE_EXCEL_2010_POSTFIX = "xlsx";
    public static final String NOT_EXCEL_FILE = " : Not the Excel file!";
    public static final String PROCESSING = "Processing...";
    public static final String EMPTY = "";
    public static final String POINT = ".";

    protected abstract T getModel();

    private List<String> parseErrorList = new ArrayList<String>();

    public String getParseErrorInfo() {
        if (!parseErrorList.isEmpty()) {
            return ArrayUtils.toString(parseErrorList, "\n");
        }
        return "";
    }

    /**
     * @return the parseErrorList
     */
    public List<String> getParseErrorList() {
        return parseErrorList;
    }

    /**
     * 读取xls或者xlsx文件
     * @param is excel文件输入流
     * @param startRow 开始行 传null值默认取1
     * @param endRow 结束行  传null值默认取现有数据行
     * @param startColumn 开始列
     * @param endColumn 结束列
     * @return 每行数据以list返回,每个元素对应一列
     * @throws Exception 
     */
    public List<T> readExcel(String fileName, InputStream is, Integer startRow, Integer endRow, Integer startColumn,
            Integer endColumn) throws Exception {
        if (is == null) {
            throw new Exception("is参数为必传");
        }
        if (startColumn == null || endColumn == null) {
            throw new Exception("startColumn和endColumn参数为必传");
        } else {
            String postfix = getPostfix(fileName);
            if (StringUtils.isNotBlank(postfix)) {
                if (OFFICE_EXCEL_2003_POSTFIX.equals(postfix)) {
                    return readXls(is, startRow, endRow, startColumn, endColumn);
                } else if (OFFICE_EXCEL_2010_POSTFIX.equals(postfix)) {
                    return readXlsx(is, startRow, endRow, startColumn, endColumn);
                }
            } else {
                System.out.println(fileName + NOT_EXCEL_FILE);
            }
        }
        return null;
    }

    /**
     * Read the Excel 2010
     * @param is of the excel file
     * @return
     * @throws IOException
     */
    public List<T> readXlsx(InputStream is, Integer startRow, Integer endRow, Integer startColumn, Integer endColumn)
            throws IOException {
        XSSFWorkbook xssfWorkbook = new XSSFWorkbook(is);
        List<T> retlist = new ArrayList<T>();
        for (int numSheet = 0; numSheet < xssfWorkbook.getNumberOfSheets(); numSheet++) {
            XSSFSheet xssfSheet = xssfWorkbook.getSheetAt(numSheet);
            if (xssfSheet == null) {
                continue;
            }
            if (startRow == null)
                startRow = 1;
            if (endRow == null)
                endRow = xssfSheet.getLastRowNum() + 1;
            for (int rowNum = startRow - 1; rowNum < endRow; rowNum++) {
                XSSFRow xssfRow = xssfSheet.getRow(rowNum);
                if (xssfRow != null) {
                    List<String> errorList = new ArrayList<String>();
                    T model = getModel();
                    for (int colNum = startColumn; colNum <= endColumn; colNum++) {
                        try {
                            setCellValue(model, xssfRow, colNum);
                        } catch (Exception e) {
                            errorList.add(String.valueOf(colNum));
                        }
                    }
                    if (!errorList.isEmpty()) {
                        parseErrorList.add("第" + String.valueOf(xssfRow.getRowNum() + 1) + "行的第"
                                + ArrayUtils.toString(errorList, ",") + "列存在格式错误!");
                    } else if (retlist.contains(model)) {
                        parseErrorList.add("第" + String.valueOf(xssfRow.getRowNum() + 1) + "行的数据重复!");
                    } else {
                        model.setRowNum(String.valueOf(xssfRow.getRowNum() + 1));
                        retlist.add(model);
                    }
                }
            }
            break;
        }
        return retlist;
    }

    /**
     * Read the Excel 2003-2007
     * @param is of the Excel
     * @return
     * @throws IOException
     */
    public List<T> readXls(InputStream is, Integer startRow, Integer endRow, Integer startColumn, Integer endColumn)
            throws IOException {
        HSSFWorkbook hssfWorkbook = new HSSFWorkbook(is);
        List<T> retlist = new ArrayList<T>();
        for (int numSheet = 0; numSheet < hssfWorkbook.getNumberOfSheets(); numSheet++) {
            System.out.println(numSheet);
            HSSFSheet hssfSheet = hssfWorkbook.getSheetAt(numSheet);
            if (hssfSheet == null) {
                continue;
            }
            if (startRow == null)
                startRow = 1;
            if (endRow == null)
                endRow = hssfSheet.getLastRowNum() + 1;
            for (int rowNum = startRow - 1; rowNum < endRow; rowNum++) {
                HSSFRow hssfRow = hssfSheet.getRow(rowNum);
                if (hssfRow != null) {
                    List<String> errorList = new ArrayList<String>();
                    T model = getModel();
                    for (int colNum = startColumn; colNum <= endColumn; colNum++) {
                        try {
                            setCellValue(model, hssfRow, colNum);
                        } catch (Exception e) {
                            errorList.add(String.valueOf(colNum));
                        }
                    }
                    if (!errorList.isEmpty()) {
                        parseErrorList.add("第" + String.valueOf(hssfRow.getRowNum() + 1) + "行的第"
                                + ArrayUtils.toString(errorList, ",") + "列存在格式错误!");
                    } else if (retlist.contains(model)) {
                        parseErrorList.add("第" + String.valueOf(hssfRow.getRowNum() + 1) + "行的数据重复!");
                    } else {
                        model.setRowNum(String.valueOf(hssfRow.getRowNum() + 1));
                        retlist.add(model);
                    }
                }
            }
            break;
        }
        return retlist;
    }

    @SuppressWarnings("static-access")
    protected String getValue(Cell cell, DecimalFormat df) {
        if (cell == null) {
            return EMPTY;
        }
        if (cell.getCellType() == cell.CELL_TYPE_BOOLEAN) {
            return String.valueOf(cell.getBooleanCellValue());
        } else if (cell.getCellType() == cell.CELL_TYPE_NUMERIC) {
            return df.format(cell.getNumericCellValue());
        } else {
            return String.valueOf(cell.getStringCellValue());
        }
    }

    private static String getPostfix(String path) {
        if (path == null || StringUtils.isEmpty(path.trim())) {
            return EMPTY;
        }
        if (path.contains(POINT)) {
            return path.substring(path.lastIndexOf(POINT) + 1, path.length());
        }
        return EMPTY;
    }

    /**
     * 设置每个单元格对应模型的属性<br>
     * (主要注意一下对存数字的单元格和即可能存字符又可能存数字的单元格的格式化)
     * @param model
     * @param row
     * @param colNum
     */
    protected abstract void setCellValue(T model, Row row, int colNum) throws Exception;
}

 

© 著作权归作者所有

共有 人打赏支持
Curry_M
粉丝 1
博文 88
码字总数 84728
作品 0
浦东
程序员

暂无文章

大数据框架对比:Hadoop、Storm、Samza、Spark和Flink

简介 大数据是收集、整理、处理大容量数据集,并从中获得见解所需的非传统战略和技术的总称。虽然处理数据所需的计算能力或存储容量早已超过一台计算机的上限,但这种计算类型的普遍性、规模...

hblt-j
39分钟前
3
0
正则介绍及grep/egrep用法

10月16日任务 9.1 正则介绍_grep上 9.2 grep中 9.3 grep下 扩展 把一个目录下,过滤所有*.php文档中含有eval的行 grep -r --include="*.php" 'eval' /data 9.1 正则介绍_grep上 什么是正则 ...

zgxlinux
54分钟前
2
0
想用Unity3D引擎软件赚点钱的看过来

前言: 你可以不拥有很多钱 但你一定要有赚钱的能力 目前手上有项目, 需要熟练Unity3D引擎软件的伙伴 有意向的给我发私信

猿神出窍
56分钟前
2
0
Spring Boot全局异常处理

Spring Boot默认的异常处理机制 默认情况下,Spring Boot为两种情况提供了不同的响应方式。 一种是浏览器客户端请求一个不存在的页面或服务端处理发生异常时,一般情况下浏览器默认发送的请求...

狼王黄师傅
今天
8
0
Thinkphp5 优雅配置两个数据库

工作需要需要配置两个数据库,框架5.0的,步骤如下: 1、在database.php同级创建一个database2.php文件 在里面配置第二个数据库信息, 2、在config中配置这个数据库信息: 3、创建第二个表的...

wqzbxh
今天
5
1

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部