java-excel工具类

excel工具类

import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.hssf.util.HSSFColor;
import org.apache.poi.ss.util.CellRangeAddress;

import java.io.FileOutputStream;
import java.io.OutputStream;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

public class ExcelUtil {
    //显示的导出表的标题
    private List sheetTitle;
    //导出表的列名
    private String[] rowName;
    private List dataList = new ArrayList();
// String             response;

    //构造方法,传入要导出的数据
    public ExcelUtil(List sheetTitle, String[] rowName, List dataList) {
        this.dataList = dataList;
        this.rowName = rowName;
        this.sheetTitle = sheetTitle;
    }

    /*
     * 导出数据
     * */
    public void export(String outSrc) throws Exception {
        try {
            HSSFWorkbook workbook = new HSSFWorkbook(); // 创建工作簿对象
            for (int i = 0; i < sheetTitle.size(); i++) {
                HSSFSheet sheet = workbook.createSheet(sheetTitle.get(i)); // 创建工作表sheet
                // 产生表格标题行
                 HSSFRow rowm = sheet.createRow(0);
                 HSSFCell cellTitle = rowm.createCell(0);
                 sheet.addMergedRegion(new CellRangeAddress(0, 1, 0, (rowName.length-1)));
                 cellTitle.setCellValue(sheetTitle.get(i));

                //sheet样式定义【getColumnTopStyle()/getStyle()均为自定义方法 - 在下面 - 可扩展】
                HSSFCellStyle columnTopStyle = this.getColumnTopStyle(workbook);//获取列头样式对象
                cellTitle.setCellStyle(columnTopStyle);
                //HSSFCellStyle style = this.getStyle(workbook); //单元格样式对象

                // 定义所需列数
                int columnNum = rowName.length;
                HSSFRow rowRowName = sheet.createRow(2); // 在索引2的位置创建行(最顶端的行开始的第二行)
                // 将列头设置到sheet的单元格中
                for (int n = 0; n < columnNum; n++) {
                    HSSFCell cellRowName = rowRowName.createCell(n); //创建列头对应个数的单元格
                    cellRowName.setCellType(HSSFCell.CELL_TYPE_STRING); //设置列头单元格的数据类型
                    HSSFRichTextString text = new HSSFRichTextString(rowName[n]);
                    cellRowName.setCellValue(text); //设置列头单元格的值
                    //cellRowName.setCellStyle(columnTopStyle); //设置列头单元格样式
                }
                //将查询出的数据设置到sheet对应的单元格中
                for (int k = 0; k < dataList.size(); k++) {
                    Object[] obj = dataList.get(k);//遍历每个对象
                    HSSFRow row = sheet.createRow(k + 3);//创建所需的行数
                    for (int j = 0; j < obj.length; j++) {
                        HSSFCell cell = null; //设置单元格的数据类型
                        // if(j == 0){
                        // cell = row.createCell(j,HSSFCell.CELL_TYPE_NUMERIC);
                        // cell.setCellValue(i+1);
                        // }else{
                        cell = row.createCell(j, HSSFCell.CELL_TYPE_STRING);
                        if (!"".equals(obj[j]) && obj[j] != null) {
                            cell.setCellValue(obj[j].toString()); //设置单元格的值
                        }
                        // }
                        //cell.setCellStyle(style); //设置单元格样式
                    }
                }
                //让列宽随着导出的列长自动适应
            /*for (int colNum = 0; colNum < columnNum; colNum++)
            {
               int columnWidth = sheet.getColumnWidth(colNum) / 256;
               for (int rowNum = 0; rowNum < sheet.getLastRowNum(); rowNum++)
               {
                  HSSFRow currentRow;
                  //当前行未被使用过
                  if (sheet.getRow(rowNum) == null)
                  {
                     currentRow = sheet.createRow(rowNum);
                  } else
                  {
                     currentRow = sheet.getRow(rowNum);
                  }
                  if (currentRow.getCell(colNum) != null)
                  {
                     HSSFCell currentCell = currentRow.getCell(colNum);
                     // logger.info("celltype:" + currentCell.getCellType());
                     if (currentCell.getCellType() == HSSFCell.CELL_TYPE_STRING)
                     {
                        int length = currentCell.getStringCellValue().getBytes().length;
                        // logger.info("length:" + length);
                        if (columnWidth < length)
                        {
                           columnWidth = length;
                        }
                     }
                  }
               }
               if (colNum == 0)
               {
                  sheet.setColumnWidth(colNum, (columnWidth - 2) * 256);
               } else
               {
                  sheet.setColumnWidth(colNum, (columnWidth + 4) * 256);
               }
            }*/
            }
            if (workbook != null) {
                try {
                    //String fileName = "Excel-" + String.valueOf(System.currentTimeMillis()).substring(4, 13) + ".xls";
               //String headStr = "attachment; filename=\"" + fileName + "\"";
                    // response.reset();
                    // response.setContentType("APPLICATION/OCTET-STREAM");
                    // response.setHeader("Content-Disposition", headStr);
                    // OutputStream out = response.getOutputStream();
                    // workbook.write(out);
                    OutputStream out = new FileOutputStream(outSrc);
                    workbook.write(out);
                    System.out.println(outSrc + "生成成功!");
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /*
     * 列头单元格样式
     */
    public HSSFCellStyle getColumnTopStyle(HSSFWorkbook workbook) {
        // 设置字体
        HSSFFont font = workbook.createFont();
        //设置字体大小
        font.setFontHeightInPoints((short) 11);
        //字体加粗
        font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
        //设置字体名字
        font.setFontName("Courier New");
        //设置样式;
        HSSFCellStyle style = workbook.createCellStyle();
        //设置底边框;
        style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
        //设置底边框颜色;
        style.setBottomBorderColor(HSSFColor.BLACK.index);
        //设置左边框;
        style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
        //设置左边框颜色;
        style.setLeftBorderColor(HSSFColor.BLACK.index);
        //设置右边框;
        style.setBorderRight(HSSFCellStyle.BORDER_THIN);
        //设置右边框颜色;
        style.setRightBorderColor(HSSFColor.BLACK.index);
        //设置顶边框;
        style.setBorderTop(HSSFCellStyle.BORDER_THIN);
        //设置顶边框颜色;
        style.setTopBorderColor(HSSFColor.BLACK.index);
        //在样式用应用设置的字体;
        style.setFont(font);
        //设置自动换行;
        style.setWrapText(false);
        //设置水平对齐的样式为居中对齐;
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        //设置垂直对齐的样式为居中对齐;
        style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
        return style;
    }

    /*
     * 列数据信息单元格样式
     */
    public HSSFCellStyle getStyle(HSSFWorkbook workbook) {
        // 设置字体
        HSSFFont font = workbook.createFont();
        //设置字体大小
        //font.setFontHeightInPoints((short)10);
        //字体加粗
        //font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
        //设置字体名字
        font.setFontName("Courier New");
        //设置样式;
        HSSFCellStyle style = workbook.createCellStyle();
        //设置底边框;
        style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
        //设置底边框颜色;
        style.setBottomBorderColor(HSSFColor.BLACK.index);
        //设置左边框;
        style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
        //设置左边框颜色;
        style.setLeftBorderColor(HSSFColor.BLACK.index);
        //设置右边框;
        style.setBorderRight(HSSFCellStyle.BORDER_THIN);
        //设置右边框颜色;
        style.setRightBorderColor(HSSFColor.BLACK.index);
        //设置顶边框;
        style.setBorderTop(HSSFCellStyle.BORDER_THIN);
        //设置顶边框颜色;
        style.setTopBorderColor(HSSFColor.BLACK.index);
        //在样式用应用设置的字体;
        style.setFont(font);
        //设置自动换行;
        style.setWrapText(false);
        //设置水平对齐的样式为居中对齐;
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        //设置垂直对齐的样式为居中对齐;
        style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
        return style;
    }

    public static void main(String[] args) {
        List sheetTitle = new ArrayList<>();
        //导出表的列名
        String[] rowName = {"序号", "姓名", "年龄", "爱好"};
        List dataList = new ArrayList();

        sheetTitle.add("员工信息表1");
        sheetTitle.add("员工信息表2");
        List list = new ArrayList<>();
        Staff staff1 = new Staff("101", "小明", 20, "打球");
        Staff staff2 = new Staff("102", "小石", 22, "爬山");
        list.add(staff1);
        list.add(staff2);
        for (Staff sta:list
             ) {
            Object[] objects = new Object[4];
            objects[0] = sta.getNumber();
            objects[1] = sta.getName();
            objects[2] = sta.getAge();
            objects[3] = sta.getLike();
            dataList.add(objects);
        }
        ExcelUtil excelUtils = new ExcelUtil(sheetTitle, rowName, dataList);
        try {
            excelUtils.export("f:/getExcel/" + new SimpleDateFormat("yyyy-MM-dd_hhmmss").format(new Date()) + ".xls");
        } catch (Exception e) {
            e.printStackTrace();
        }

    }
}

你可能感兴趣的:(Java工具类)