Poi实现Excel导出工具类封装
工具类代码PoiExcelExport如下：
package com.myssm.util.poi; 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; import javax.servlet.http.HttpServletResponse; 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 org.apache.poi.ss.util.CellRangeAddress; 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列的公式 涉及到的行号使用@替换 如A@+B@ */ public void setColFormula(String[] colFormula) { this.colFormula = colFormula; } /** * 写excel. * @param titleColumn 对应bean的属性名 * @param titleName excel要导出的表名 * @param titleSize 列宽 * @param dataList 数据 */ public void wirteExcel(String titleColumn[],String titleName[],int titleSize[],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++){ sheet.setColumnWidth(i, titleSize[i]*256); //设置宽度 Cell cell = titleNameRow.createCell(i); cell.setCellStyle(titleStyle); cell.setCellValue(titleName[i].toString()); } //为表头添加自动筛选 if(!"".equals(address)){ CellRangeAddress c = (CellRangeAddress) CellRangeAddress.valueOf(address); sheet.setAutoFilter(c); } //通过反射获取数据并写入到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; // 设置要执行的方法 Method method = clsss.getDeclaredMethod(methodName); //获取返回类型 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); font.setBold(true); 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; } } 测试如下：
实体bean:
package com.myssm.util.poi; public class Man { private String name; private int sex; private String idCard; private float salary; public Man(String name, int sex, String idCard, float salary) { super(); this.name = name; this.sex = sex; this.idCard = idCard; this.salary = salary; } public Man() { super(); } public String getName() { return name; } public void setName(String name) { this.name = name; } public int getSex() { return sex; } public void setSex(int sex) { this.sex = sex; } public String getIdCard() { return idCard; } public void setIdCard(String idCard) { this.idCard = idCard; } public float getSalary() { return salary; } public void setSalary(float salary) { this.salary = salary; } } 测试类： package com.myssm.util.poi; import java.util.ArrayList; import java.util.List; public class Test { public static void main(String[] args) { PoiExcelExport pee = new PoiExcelExport("E:/test.xls","sheet1"); //数据 List<Man> dataList = new ArrayList(); Man man1 = new Man("张三",20,"男",(float)10000.8); Man man2 = new Man("李四",21,"男",(float)11000.8); Man man3 = new Man("王五",22,"女",(float)1200.8); Man man4 = new Man("赵六",23,"男",(float)13000.8); Man man5 = new Man("田七",24,"男",(float)14000.8); Man man6 = new Man(); man6.setName("老八"); dataList.add(man1);dataList.add(man2);dataList.add(man3);dataList.add(man4);dataList.add(man5); dataList.add(man6); //调用 String titleColumn[] = {"name","sex","idCard","salary",""}; String titleName[] = {"姓名","性别","身份证号","月薪","年薪"}; int titleSize[] = {13,13,13,13,13}; //其他设置 set方法可全不调用 String colFormula[] = new String[5]; colFormula[4] = "D@*12"; //设置第5列的公式 pee.setColFormula(colFormula); pee.setAddress("A:D"); //自动筛选 pee.wirteExcel(titleColumn, titleName, titleSize, dataList); } }