Java 導出excel、pdf

公共導出方法

package com.rexen.rest.common.util;


import com.itextpdf.text.*;
import com.itextpdf.text.html.simpleparser.HTMLWorker;
import com.itextpdf.text.pdf.*;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import org.apache.poi.xssf.usermodel.XSSFFont;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.io.StringReader;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.List;

/**
 * 導出表格
 * 27  * @since 2019-05-10
 */
public class Export {
    /**
     * 導出excel，xls格式
     *
     * @param sheetName sheet名稱
     * @param title     標題, 參數類型：數組
     * @param list      需要導出內容數據, 參數類型：List<List<String>> 需要將所有數據轉換成字符串類型
     * @param fileName  excel文件名，參數類型：字符串 38      * @since 2019-05-10
     */
    public static void getHSSFWorkbook(HttpServletResponse response, HttpServletRequest request, String sheetName, String[] title, List<List<String>> list, String fileName) {
        /* 獲取文件名稱后綴*/
        final String suffixName = fileName.substring(fileName.lastIndexOf(".") + 1);

        /* 創建HSSFWorkbook，對應一個Excel文件*/
        HSSFWorkbook workbook = new HSSFWorkbook();
        /* 在workbook中添加一個sheet,對應Excel文件中的sheet*/
        Sheet sheet = workbook.createSheet(sheetName);
        /* 在sheet中添加表頭第0行*/
        Row row = sheet.createRow(0);
        /* 創建單元格*/
        Cell cell = null;

        /* 創建單元格樣式*/
        CellStyle style = workbook.createCellStyle();
        /*設置單元格填充樣式，SOLID_FOREGROUND純色使用前景顏色填充*/
        style.setFillPattern(FillPatternType.SOLID_FOREGROUND);
        /* 設置填充顏色*/
        style.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex());
        /* 水平居中*/
        style.setAlignment(HorizontalAlignment.CENTER);
        /*垂直對齊居中*/
        style.setVerticalAlignment(VerticalAlignment.CENTER);
        /* 自動換行*/
        style.setWrapText(true);
        /*創建字體樣式*/
        HSSFFont font = workbook.createFont();
        /*設置字體大小*/
        font.setFontHeightInPoints((short) 14);
        style.setFont(font);

        /* 設置內容樣式*/
        HSSFCellStyle tableStyle = workbook.createCellStyle();
        /* 水平居中*/
        tableStyle.setAlignment(HorizontalAlignment.CENTER);
        /* 垂直對齊居中*/
        tableStyle.setVerticalAlignment(VerticalAlignment.CENTER);
        /* 自動換行*/
        tableStyle.setWrapText(true);

        /* 創建表頭*/
        for (int i = 0; i < title.length; i++) {
            // 設置列寬
            sheet.setColumnWidth(i, 20 * 256);
            cell = row.createCell(i);
            cell.setCellValue(title[i]);
            cell.setCellStyle(style);
        }

        /* 創建內容*/
        for (int i = 0; i < list.size(); i++) {
            row = sheet.createRow(i + 1);
            for (int j = 0; j < list.get(i).size(); j++) {
                /* 將內容按順序賦給對應的列對象*/
                Cell tableCell = row.createCell(j);
                tableCell.setCellValue(list.get(i).get(j));
                tableCell.setCellStyle(tableStyle);
            }
        }
        try {
            /* 響應到客戶端*/
            setResponseHeader(response, fileName + ".xls");
            OutputStream os = response.getOutputStream();
            workbook.write(os);
            os.flush();
            os.close();

        } catch (IOException e) {
            e.printStackTrace();
        }

    }


    /**
     * 導出pdf文檔
     *
     * @param lists    導出數據,注: 表頭需要添加到lists中第一條
     * @param request  請求對象
     * @param response 響應對象121      * @since 2019-05-10
     */
    // 下載pdf文檔
    public static void download(HttpServletRequest request, HttpServletResponse response, List<List<String>> lists, String fileName) throws Exception {
        // 告訴瀏覽器用什么軟件可以打開此文件
        response.setHeader("content-Type", "application/pdf");
        // 下載文件的默認名稱
        response.setHeader("Content-Disposition","attachment;fileName=" +URLEncoder.encode(fileName + ".pdf", "UTF-8"));
        BaseFont baseFont = BaseFont.createFont("STSongStd-Light", "UniGB-UCS2-H", false);
        // 自定義字體屬性
        com.itextpdf.text.Font font = new com.itextpdf.text.Font(baseFont, 11);
        com.itextpdf.text.Font titleFont = new com.itextpdf.text.Font(baseFont, 14);
        Document document = new Document(PageSize.A4);
        PdfWriter.getInstance(document, response.getOutputStream());
        document.open();
        if (lists.size() > 0 && lists != null) {
            for (int i = 0; i < lists.size(); i++) {
                PdfPTable table = new PdfPTable(lists.get(i).size());
                table.setWidthPercentage(100);
                PdfPCell cell = new PdfPCell();
                if (i == 0) {
                    if (lists.get(i).size() > 0 && lists.get(i) != null) {
                        for (int j = 0; j < lists.get(i).size(); j++) {
                            System.out.println(lists.get(i).get(j));
                            Paragraph p=new Paragraph(lists.get(i).get(j), titleFont);
                            p.setFont(titleFont);
                            // 水平居中
                            cell.setHorizontalAlignment(Element.ALIGN_CENTER);
                            cell.setPhrase(p);
                            cell.setBackgroundColor(new BaseColor(204, 204, 204));
                            // 文檔中加入該段落
                            table.addCell(cell);
                            document.add(table);
                        }
                    }
                }else {
                    if (lists.get(i).size() > 0 && lists.get(i) != null) {
                        for (int j = 0; j < lists.get(i).size(); j++) {
                            System.out.println(lists.get(i).get(j));
                            Paragraph p=new Paragraph(lists.get(i).get(j),font);
                            p.setFont(font);
                            // 設置段落居中，其中1為居中對齊，2為右對齊，3為左對齊
                            cell.setHorizontalAlignment(Element.ALIGN_CENTER);
                            cell.setPhrase(p);
                            table.addCell(cell);
                            document.add(table);
                        }
                    }
                }
            }
            document.close();
        }
    }

    /**
     * 導出excel，xlsx格式
     *
     * @param sheetName sheet名稱, 參數類型：數組
     * @param title     標題, 參數類型：數組
     * @param list      需要導出內容數據, 參數類型：List<List<String>> 需要將所有數據轉換成字符串類型
     * @param fileName  excel文件名，參數類型：字符串183      * @since 2019-05-10
     */
    public static void getSXSSFWorkbook(HttpServletResponse response, HttpServletRequest request, String sheetName, String[] title, List<List<String>> list, String fileName) {

        SXSSFWorkbook workbook = new SXSSFWorkbook();

        // 在workbook中添加一個sheet,對應Excel文件中的sheet
        Sheet sheet = workbook.createSheet(sheetName);

        // 在sheet中添加表頭第0行
        Row row = sheet.createRow(0);

        // 聲明列對象
        Cell cell = null;

        /* 創建單元格樣式*/
        CellStyle style = workbook.createCellStyle();
        /*設置單元格填充樣式，SOLID_FOREGROUND純色使用前景顏色填充*/
        style.setFillPattern(FillPatternType.SOLID_FOREGROUND);
        /* 設置填充顏色*/
        style.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex());
        /* 水平居中*/
        style.setAlignment(HorizontalAlignment.CENTER);
        /*垂直對齊居中*/
        style.setVerticalAlignment(VerticalAlignment.CENTER);
        /* 自動換行*/
        style.setWrapText(true);
        /*創建字體樣式*/
        XSSFFont font = (XSSFFont) workbook.createFont();
        /*設置字體大小*/
        font.setFontHeightInPoints((short) 14);
        style.setFont(font);

        /* 設置內容樣式*/
        CellStyle tableStyle = workbook.createCellStyle();
        /* 水平居中*/
        tableStyle.setAlignment(HorizontalAlignment.CENTER);
        /* 垂直對齊居中*/
        tableStyle.setVerticalAlignment(VerticalAlignment.CENTER);
        /* 自動換行*/
        tableStyle.setWrapText(true);

        // 創建標題
        for (int i = 0; i < title.length; i++) {
            // 設置列寬
            sheet.setColumnWidth(i, 20 * 256);
            cell = row.createCell(i);
            cell.setCellValue(title[i]);
            cell.setCellStyle(style);
        }

        // 創建內容
        for (int i = 0; i < list.size(); i++) {
            row = sheet.createRow(i + 1);
            for (int j = 0; j < list.get(i).size(); j++) {
                Cell celll = row.createCell(j);
                // 將內容按順序賦給對應的列對象
                celll.setCellValue(list.get(i).get(j));
                celll.setCellStyle(tableStyle);
            }
        }
        try {
            /* 響應到客戶端*/
            Export.setResponseHeader(response, fileName + ".xlsx");
            OutputStream os = response.getOutputStream();
            workbook.write(os);
            os.flush();
            os.close();

        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 發送響應流方法
     *
     * @param response 響應
     * @param fileName excel文件名263      * @since 2019-05-10
     */
    public static void setResponseHeader(HttpServletResponse response, String fileName) {
        try {
            response.setCharacterEncoding("utf-8");
            fileName = URLEncoder.encode(fileName, "UTF-8");
            response.setHeader("content-Type", "application/x-xls");
            response.setHeader("Content-Disposition", "inline;filename=" + fileName);
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }

}

實現導出業務

    @Override
    public void export(HttpServletRequest request, HttpServletResponse response, List<EventStatusVO> list, String reportType) {
        List<List<String>> listList = new ArrayList<>();
        String dataFormatStr = "yyyy-MM-dd HH:mm:ss";
        SimpleDateFormat dateFormat = new SimpleDateFormat(dataFormatStr);
        for (EventStatusVO item: list) {
            List<String> lists = new ArrayList<>();
            //事件名稱
            lists.add(item.getName());
            //事件類型
            lists.add(item.getEventType());
            //事件級別
            lists.add(String.valueOf(item.getEventRate()));
            //事件場所
            lists.add(item.getEventPlace());
            //上報日期
            lists.add(dateFormat.format(item.getReportDate()));
            //發生日期 起止
            lists.add(dateFormat.format(item.getHappenDateStart()) + "至" + dateFormat.format(item.getHappenDateEnd()));
            //事件進度
            lists.add(item.getEventStatus());
            listList.add(lists);
        }
        String fileName = "我的上報表" + dateFormat.format(new Date());
        String sheetName = "我的上報";
        String []title = {"事件名稱", "事件類型", "事件級別", "發生場所", "上報日期", "發生時間", "事件進度"};

        final String xls = "xls";
        final String xlsx = "xlsx";
        final String pdf = "pdf";
        if (xls.equals(reportType)) {
            Export.getHSSFWorkbook(response,request, sheetName, title, listList, fileName);
        }
        if (xlsx.equals(reportType)) {
            Export.getSXSSFWorkbook(response,request, sheetName, title, listList, fileName);
        }
        if (pdf.equals(reportType)) {
            try {
                List<String> allList = Arrays.asList(title);
                listList.add(0, allList);
                Export.download(request, response, listList, fileName);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }