POI工具类

package com.ysq.ssm.poi;

import org.apache.poi.hssf.usermodel.HSSFCellStyle;import org.apache.poi.hssf.usermodel.HSSFFont;import org.apache.poi.hssf.usermodel.HSSFPalette;import org.apache.poi.hssf.usermodel.HSSFWorkbook;import org.apache.poi.ss.usermodel.Cell;import org.apache.poi.ss.usermodel.CellStyle;import org.apache.poi.ss.usermodel.Row;import org.apache.poi.ss.usermodel.Sheet;

import javax.servlet.http.HttpServletResponse;import java.io.File;import java.io.FileOutputStream;import java.io.IOException;import java.io.OutputStream;import java.lang.reflect.Method;import java.net.URLEncoder;import java.text.DecimalFormat;import java.util.List;

/** * Created by yangshuqi on 17-5-3. */public class PoiExcelExport {    HttpServletResponse response;    // 文件名    private String fileName;    //文件保存路径    private String fileDir;    //sheet名    private String sheetName;    //表头字体    private String titleFontType = "Arial Unicode MS";    //表头背景色    private String titleBackColor = "C1FBEE";    //表头字号    private short titleFontSize = 12;    //添加自动筛选的列 如 A:M    private String address = "";    //正文字体    private String contentFontType = "Arial Unicode MS";    //正文字号    private short contentFontSize = 12;    //Float类型数据小数位    private String floatDecimal = ".00";    //Double类型数据小数位    private String doubleDecimal = ".00";    //设置列的公式    private String colFormula[] = null;

DecimalFormat floatDecimalFormat = new DecimalFormat(floatDecimal);    DecimalFormat doubleDecimalFormat = new DecimalFormat(doubleDecimal);

private HSSFWorkbook workbook = null;

public PoiExcelExport(String fileDir, String sheetName) {        this.fileDir = fileDir;        this.sheetName = sheetName;        workbook = new HSSFWorkbook();    }

public PoiExcelExport(HttpServletResponse response, String fileName, String sheetName) {        this.response = response;        this.sheetName = sheetName;        workbook = new HSSFWorkbook();    }

/**     * 设置表头字体.     *     * @param titleFontType     */    public void setTitleFontType(String titleFontType) {        this.titleFontType = titleFontType;    }

/**     * 设置表头背景色.     *     * @param titleBackColor 十六进制     */    public void setTitleBackColor(String titleBackColor) {        this.titleBackColor = titleBackColor;    }

/**     * 设置表头字体大小.     *     * @param titleFontSize     */    public void setTitleFontSize(short titleFontSize) {        this.titleFontSize = titleFontSize;    }

/**     * 设置表头自动筛选栏位,如A:AC.     *     * @param address     */    public void setAddress(String address) {        this.address = address;    }

/**     * 设置正文字体.     *     * @param contentFontType     */    public void setContentFontType(String contentFontType) {        this.contentFontType = contentFontType;    }

/**     * 设置正文字号.     *     * @param contentFontSize     */    public void setContentFontSize(short contentFontSize) {        this.contentFontSize = contentFontSize;    }

/**     * 设置float类型数据小数位 默认.00     *     * @param doubleDecimal 如 ".00"     */    public void setDoubleDecimal(String doubleDecimal) {        this.doubleDecimal = doubleDecimal;    }

/**     * 设置doubel类型数据小数位 默认.00     *     * @param floatDecimalFormat 如 ".00     */    public void setFloatDecimalFormat(DecimalFormat floatDecimalFormat) {        this.floatDecimalFormat = floatDecimalFormat;    }

/**     * 设置列的公式     *     * @param colFormula 存储i-1列的公式 涉及到的行号使用@替换 如[email protected][email protected]     */    public void setColFormula(String[] colFormula) {        this.colFormula = colFormula;    }

/**     * 写excel.     *     * @param titleColumn 对应bean的属性名     * @param titleName   excel要导出的表名     * @param dataList    数据     */    public void wirteExcel(String titleColumn[], String titleName[], List<?> dataList) {        //添加Worksheet（不添加sheet时生成的xls文件打开时会报错)        Sheet sheet = workbook.createSheet(this.sheetName);        //新建文件        OutputStream out = null;        try {            if (fileDir != null) {                //有文件路径                out = new FileOutputStream(fileDir);            } else {                //否则，直接写到输出流中                out = response.getOutputStream();                fileName = fileName + ".xls";                response.setContentType("application/x-msdownload");                response.setHeader("Content-Disposition", "attachment; filename="                        + URLEncoder.encode(fileName, "UTF-8"));            }

//写入excel的表头            Row titleNameRow = workbook.getSheet(sheetName).createRow(0);            //设置样式            HSSFCellStyle titleStyle = workbook.createCellStyle();            titleStyle = (HSSFCellStyle) setFontAndBorder(titleStyle, titleFontType, (short) titleFontSize);            titleStyle = (HSSFCellStyle) setColor(titleStyle, titleBackColor, (short) 10);

for (int i = 0; i < titleName.length; i++) {                Cell cell = titleNameRow.createCell(i);                cell.setCellStyle(titleStyle);                cell.setCellValue(titleName[i].toString());            }

//通过反射获取数据并写入到excel中            if (dataList != null && dataList.size() > 0) {                //设置样式                HSSFCellStyle dataStyle = workbook.createCellStyle();                titleStyle = (HSSFCellStyle) setFontAndBorder(titleStyle, contentFontType, (short) contentFontSize);

if (titleColumn.length > 0) {                    for (int rowIndex = 1; rowIndex <= dataList.size(); rowIndex++) {                        Object obj = dataList.get(rowIndex - 1);     //获得该对象                        Class clsss = obj.getClass();     //获得该对对象的class实例                        Row dataRow = workbook.getSheet(sheetName).createRow(rowIndex);                        for (int columnIndex = 0; columnIndex < titleColumn.length; columnIndex++) {                            String title = titleColumn[columnIndex].toString().trim();                            if (!"".equals(title)) {  //字段不为空                                //使首字母大写//                                String UTitle = Character.toUpperCase(title.charAt(0)) + title.substring(1, title.length()); // 使其首字母大写;//                                String methodName = "get" + UTitle;                                String getMethodName = "get" + title.substring(0, 1).toUpperCase() + title.substring(1);

// 设置要执行的方法                                Method method = clsss.getDeclaredMethod(getMethodName);

//获取返回类型                                String returnType = method.getReturnType().getName();

String data = method.invoke(obj) == null ? "" : method.invoke(obj).toString();                                Cell cell = dataRow.createCell(columnIndex);                                if (data != null && !"".equals(data)) {                                    if ("int".equals(returnType)) {                                        cell.setCellValue(Integer.parseInt(data));                                    } else if ("long".equals(returnType)) {                                        cell.setCellValue(Long.parseLong(data));                                    } else if ("float".equals(returnType)) {                                        cell.setCellValue(floatDecimalFormat.format(Float.parseFloat(data)));                                    } else if ("double".equals(returnType)) {                                        cell.setCellValue(doubleDecimalFormat.format(Double.parseDouble(data)));                                    } else {                                        cell.setCellValue(data);                                    }                                }                            } else {   //字段为空 检查该列是否是公式                                if (colFormula != null) {                                    String sixBuf = colFormula[columnIndex].replace("@", (rowIndex + 1) + "");                                    Cell cell = dataRow.createCell(columnIndex);                                    cell.setCellFormula(sixBuf.toString());                                }                            }                        }                    }

}            }            workbook.write(out);        } catch (Exception e) {            e.printStackTrace();        } finally {            try {                out.close();            } catch (IOException e) {                e.printStackTrace();            }        }    }

/**     * 将16进制的颜色代码写入样式中来设置颜色     *     * @param style 保证style统一     * @param color 颜色：66FFDD     * @param index 索引 8-64 使用时不可重复     * @return     */    public CellStyle setColor(CellStyle style, String color, short index) {        if (color != "" && color != null) {            //转为RGB码            int r = Integer.parseInt((color.substring(0, 2)), 16);   //转为16进制            int g = Integer.parseInt((color.substring(2, 4)), 16);            int b = Integer.parseInt((color.substring(4, 6)), 16);            //自定义cell颜色            HSSFPalette palette = workbook.getCustomPalette();            palette.setColorAtIndex((short) index, (byte) r, (byte) g, (byte) b);

style.setFillPattern(CellStyle.SOLID_FOREGROUND);            style.setFillForegroundColor(index);        }        return style;    }

/**     * 设置字体并加外边框     *     * @param style 样式     * @param style 字体名     * @param style 大小     * @return     */    public CellStyle setFontAndBorder(CellStyle style, String fontName, short size) {        HSSFFont font = workbook.createFont();        font.setFontHeightInPoints(size);        font.setFontName(fontName);        style.setFont(font);        style.setBorderBottom(CellStyle.BORDER_THIN); //下边框        style.setBorderLeft(CellStyle.BORDER_THIN);//左边框        style.setBorderTop(CellStyle.BORDER_THIN);//上边框        style.setBorderRight(CellStyle.BORDER_THIN);//右边框        return style;    }

/**     * 删除文件     *     * @param fileDir     * @return     */    public boolean deleteExcel() {        boolean flag = false;        File file = new File(this.fileDir);        // 判断目录或文件是否存在        if (!file.exists()) {  // 不存在返回 false            return flag;        } else {            // 判断是否为文件            if (file.isFile()) {  // 为文件时调用删除文件方法                file.delete();                flag = true;            }        }        return flag;    }

/**     * 删除文件     *     * @param fileDir     * @return     */    public boolean deleteExcel(String path) {        boolean flag = false;        File file = new File(path);        // 判断目录或文件是否存在        if (!file.exists()) {  // 不存在返回 false            return flag;        } else {            // 判断是否为文件            if (file.isFile()) {  // 为文件时调用删除文件方法                file.delete();                flag = true;            }        }        return flag;    }}