Java常用工具类封装——封装对excel的操作

封装对excel的操作，包括本地读写excel和流中输出excel

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.text.DecimalFormat;
import java.util.LinkedList;
import java.util.List;
 
import javax.servlet.http.HttpServletResponse;
 
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
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.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.DateUtil;
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.ss.usermodel.WorkbookFactory;
 
/**
* 封装对excel的操作，包括本地读写excel和流中输出excel,支持office 2007。<br/>
* 依赖于poi-3.9-20121203.jar,poi-ooxml-3.9-20121203.jar,poi-ooxml-schemas-3.9-
* 20121203.jar,dom4j-1.6.1.jar<br/>
* 有参构造函数参数为excel的全路径<br/>
*
* @author jiangshuai
* @date 2017年04月24日 
*/
public class ExcelUtil {
 
    // excel文件路径
    private String path = "";
 
    // 写入excel时，是否自动扩展列宽度来符合内容。
    private boolean autoColumnWidth = false;
 
    /**
     * 无参构造函数 默认
     */
    public ExcelUtil() {
    }
 
    /**
     * 有参构造函数
     *
     * @param path
     *            excel路径
     */
    public ExcelUtil(String path) {
        this.path = path;
    }
 
    /**
     * 读取某个工作簿上的所有单元格的值。
     *
     * @param sheetOrder
     *            工作簿序号，从0开始。
     * @return List<Object[]> 所有单元格的值。
     * @throws IOException
     *             加载excel文件IO异常。
     * @throws FileNotFoundException
     *             excel文件没有找到异常。
     * @throws InvalidFormatException
     * @author jiangshuai
     * @date 2017年04月24日 
     */
    public List<Object[]> read(int sheetOrder) throws FileNotFoundException,
            IOException, InvalidFormatException {
        FileInputStream fis = new FileInputStream(path);
        Workbook workbook = WorkbookFactory.create(fis);
        if (fis != null) {
            fis.close();
        }
        Sheet sheet = workbook.getSheetAt(sheetOrder);
        // 用来记录excel值
        List<Object[]> valueList = new LinkedList<Object[]>();
        // 循环遍历每一行、每一列。
        for (Row row : sheet) {
            // 每一行
            Object[] rowObject = null;
            for (Cell cell : row) {
                // cell.getCellType是获得cell里面保存的值的type
                switch (cell.getCellType()) {
                case Cell.CELL_TYPE_BOOLEAN:
                    // 得到Boolean对象的方法
                    rowObject = CollectionUtil.addObjectToArray(rowObject,
                            cell.getBooleanCellValue());
                    break;
                case Cell.CELL_TYPE_NUMERIC:
                    // 先看是否是日期格式
                    if (DateUtil.isCellDateFormatted(cell)) {
                        // 读取日期格式
                        rowObject = CollectionUtil.addObjectToArray(rowObject,
                                cell.getDateCellValue());
                    } else {
                        DecimalFormat df = new DecimalFormat();
                        // 单元格的值,替换掉,
                        String value = df.format(cell.getNumericCellValue())
                                .replace(",", "");
                        // 读取数字
                        rowObject = CollectionUtil.addObjectToArray(rowObject,
                                value);
                    }
                    break;
                case Cell.CELL_TYPE_FORMULA:
                    // 读取公式
                    rowObject = CollectionUtil.addObjectToArray(rowObject,
                            cell.getCellFormula());
                    break;
                case Cell.CELL_TYPE_STRING:
                    // 读取String
                    rowObject = CollectionUtil.addObjectToArray(rowObject, cell
                            .getRichStringCellValue().toString());
                    break;
                }
            }
            // 将这行添加到list。
            valueList.add(rowObject);
        }
        return valueList;
    }
 
    /**
     * 读取某个工作簿上的某个单元格的值。
     *
     * @param sheetOrder
     *            工作簿序号,从0开始。
     * @param colum
     *            列数 从1开始
     * @param row
     *            行数 从1开始
     * @return 单元格的值。
     * @throws Exception
     *             加载excel异常。
     * @author jiangshuai
     * @date 2017年04月24日 
     */
    public String read(int sheetOrder, int colum, int row) throws Exception {
        FileInputStream fis = new FileInputStream(path);
        Workbook workbook = WorkbookFactory.create(fis);
        if (fis != null) {
            fis.close();
        }
        Sheet sheet = workbook.getSheetAt(sheetOrder);
        Row rows = sheet.getRow(row - 1);
        Cell cell = rows.getCell(colum - 1);
        String content = cell.getStringCellValue();
        return content;
    }
 
    /**
     * 在指定的工作簿、行、列书写值。
     *
     * @param sheetOrder
     *            工作簿序号，基于0.
     * @param colum
     *            列 基于1
     * @param row
     *            行 基于1
     * @param content
     *            将要被书写的内容。
     * @throws Exception
     *             书写后保存异常。
     * @author jiangshuai
     * @date 2017年04月24日 
     */
    public void write(int sheetOrder, int colum, int row, String content)
            throws Exception {
        FileInputStream fis = new FileInputStream(path);
        Workbook workbook = WorkbookFactory.create(fis);
        if (fis != null) {
            fis.close();
        }
        Sheet sheet = workbook.getSheetAt(sheetOrder);
        Row rows = sheet.createRow(row - 1);
        Cell cell = rows.createCell(colum - 1);
        cell.setCellValue(content);
        FileOutputStream fileOut = new FileOutputStream(path);
        workbook.write(fileOut);
        fileOut.close();
 
    }
 
    /**
     * 得到一个工作区最后一条记录的序号，相当于这个工作簿共多少行数据。
     *
     * @param sheetOrder
     *            工作区序号
     * @return int 序号。
     * @throws IOException
     *             根据excel路径加载excel异常。
     * @throws InvalidFormatException
     * @author jiangshuai
     * @date 2017年04月24日 
     */
    public int getSheetLastRowNum(int sheetOrder) throws IOException,
            InvalidFormatException {
        FileInputStream fis = new FileInputStream(path);
        Workbook workbook = WorkbookFactory.create(fis);
        if (fis != null) {
            fis.close();
        }
        Sheet sheet = workbook.getSheetAt(sheetOrder);
        return sheet.getLastRowNum();
    }
 
    /**
     * 在磁盘生成一个含有内容的excel,路径为path属性
     *
     * @param sheetName
     *            导出的sheet名称
     * @param fieldName
     *            列名数组
     * @param data
     *            数据组
     * @throws IOException
     * @author jiangshuai
     * @date 2017年04月24日 
     */
    public void makeExcel(String sheetName, String[] fieldName,
            List<Object[]> data) throws IOException {
        // 在内存中生成工作薄
        HSSFWorkbook workbook = makeWorkBook(sheetName, fieldName, data);
        // 截取文件夹路径
        String filePath = path.substring(0, path.lastIndexOf("\\"));
        // 如果路径不存在，创建路径
        File file = new File(filePath);
        // System.out.println(path+"-----------"+file.exists());
        if (!file.exists())
            file.mkdirs();
        FileOutputStream fileOut = new FileOutputStream(path);
        workbook.write(fileOut);
        fileOut.close();
    }
 
    /**
     * 在输出流中导出excel。
     *
     * @param excelName
     *            导出的excel名称 包括扩展名
     * @param sheetName
     *            导出的sheet名称
     * @param fieldName
     *            列名数组
     * @param data
     *            数据组
     * @param response
     *            response
     * @throws IOException
     *             转换流时IO错误
     * @author jiangshuai
     * @date 2017年04月24日 
     */
    public void makeStreamExcel(String excelName, String sheetName,
            String[] fieldName, List<Object[]> data,
            HttpServletResponse response) throws IOException {
        OutputStream os = null;
        response.reset(); // 清空输出流
        os = response.getOutputStream(); // 取得输出流
        response.setHeader("Content-disposition", "attachment; filename="
                + new String(excelName.getBytes(), "ISO-8859-1")); // 设定输出文件头
        response.setContentType("application/msexcel"); // 定义输出类型
        // 在内存中生成工作薄
        HSSFWorkbook workbook = makeWorkBook(sheetName, fieldName, data);
        os.flush();
        workbook.write(os);
    }
 
    /**
     * 根据条件，生成工作薄对象到内存。
     *
     * @param sheetName
     *            工作表对象名称
     * @param fieldName
     *            首列列名称
     * @param data
     *            数据
     * @return HSSFWorkbook
     * @author jiangshuai
     * @date 2017年04月24日 
     */
    private HSSFWorkbook makeWorkBook(String sheetName, String[] fieldName,
            List<Object[]> data) {
        // 用来记录最大列宽,自动调整列宽。
        Integer collength[] = new Integer[fieldName.length];
 
        // 产生工作薄对象
        HSSFWorkbook workbook = new HSSFWorkbook();
        // 产生工作表对象
        HSSFSheet sheet = workbook.createSheet();
        // 为了工作表能支持中文,设置字符集为UTF_16
        workbook.setSheetName(0, sheetName);
        // 产生一行
        HSSFRow row = sheet.createRow(0);
        // 产生单元格
        HSSFCell cell;
        // 写入各个字段的名称
        for (int i = 0; i < fieldName.length; i++) {
            // 创建第一行各个字段名称的单元格
            cell = row.createCell((short) i);
            // 设置单元格内容为字符串型
            cell.setCellType(HSSFCell.CELL_TYPE_STRING);
            // 为了能在单元格中输入中文,设置字符集为UTF_16
            // cell.setEncoding(HSSFCell.ENCODING_UTF_16);
            // 给单元格内容赋值
            cell.setCellValue(new HSSFRichTextString(fieldName[i]));
            // 初始化列宽
            collength[i] = fieldName[i].getBytes().length;
        }
        // 临时单元格内容
        String tempCellContent = "";
        // 写入各条记录,每条记录对应excel表中的一行
        for (int i = 0; i < data.size(); i++) {
            Object[] tmp = data.get(i);
            // 生成一行
            row = sheet.createRow(i + 1);
            for (int j = 0; j < tmp.length; j++) {
                cell = row.createCell((short) j);
                // 设置单元格字符类型为String
                cell.setCellType(HSSFCell.CELL_TYPE_STRING);
                tempCellContent = (tmp[j] == null) ? "" : tmp[j].toString();
                cell.setCellValue(new HSSFRichTextString(tempCellContent));
 
                // 如果自动调整列宽度。
                if (autoColumnWidth) {
                    if (j >= collength.length) { // 标题列数小于数据列数时。
                        collength = CollectionUtil.addObjectToArray(collength,
                                tempCellContent.getBytes().length);
                    } else {
                        // 如果这个内容的宽度大于之前最大的，就按照这个设置宽度。
                        if (collength[j] < tempCellContent.getBytes().length) {
                            collength[j] = tempCellContent.getBytes().length;
                        }
                    }
                }
            }
        }
 
        // 自动调整列宽度。
        if (autoColumnWidth) {
            // 调整列为这列文字对应的最大宽度。
            for (int i = 0; i < fieldName.length; i++) {
                sheet.setColumnWidth(i, collength[i] * 2 * 256);
            }
        }
        return workbook;
    }
 
    /**
     * 功能：设置写入excel时，是否自动扩展列宽度来符合内容，默认为false。
     *
     * @author jiangshuai
     * @date 2017年04月24日 
     * @param autoColumnWidth
     *            true或者false
     */
    public void setAutoColumnWidth(boolean autoColumnWidth) {
        this.autoColumnWidth = autoColumnWidth;
    }
}