JAVA生成行程单PDF

一、pom依赖

首先引入PDF需要的pom依赖

         <!-- pdf-->
        <dependency>
            <groupId>com.itextpdf</groupId>
            <artifactId>itextpdf</artifactId>
            <version>5.5.13</version>
            <optional>true</optional>
        </dependency>
        <dependency>
            <groupId>com.itextpdf</groupId>
            <artifactId>itext-asian</artifactId>
            <version>5.2.0</version>
            <optional>true</optional>
        </dependency>

二、PDF操作工具类及行程单实体类

创建PDF操作工具类及行程单实体类，调用 PDFUtil.createPDFFile()方法创建行程单。

 /**
 * PDF操作工具类
 */
public class PDFUtil {
    private static Logger logger = LoggerFactory.getLogger(PDFUtil.class);
    public static final String PDF_FILE_SUFFIX = ".pdf";
    public static Font headFont;
    public static Font keyFont;
    // 标题
    public static Font titleFont;
    // 设置字体大小
    public static Font textFont;
    public static final int MAX_WIDTH = 520;
    static {
        //中文格式
        BaseFont bfChinese;
        try {
            // 设置中文显示
            bfChinese = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
            headFont = new Font(bfChinese, 10, java.awt.Font.BOLD);
            keyFont = new Font(bfChinese, 8, java.awt.Font.BOLD);
            textFont = new Font(bfChinese, 8, Font.NORMAL);
            titleFont = new Font(bfChinese, 16, Font.NORMAL);
        } catch (Exception e) {
            logger.error("创建pdf格式失败，异常={}", e);
            throw new RuntimeException(e);
        }
    }
    /**
     * 为表格添加一个内容
     *
     * @param value 值
     * @param font  字体
     * @param align 对齐方式
     * @return 添加的文本框
     */
    public static PdfPCell createCell(String value, Font font, int align) {
        PdfPCell cell = new PdfPCell();
        cell.setVerticalAlignment(Element.ALIGN_MIDDLE);
        cell.setHorizontalAlignment(align);
        font.setColor(BaseColor.WHITE);
        cell.setPhrase(new Phrase(value, font));
        cell.setBackgroundColor(BaseColor.GRAY);
        return cell;
    }
    /**
     * 为表格添加一个内容
     *
     * @param value 值
     * @param font  字体
     * @return 添加的文本框
     */
    public static PdfPCell createCell(String value, Font font) {
        PdfPCell cell = new PdfPCell();
        cell.setVerticalAlignment(Element.ALIGN_MIDDLE);
        cell.setHorizontalAlignment(Element.ALIGN_CENTER);
        cell.setPhrase(new Phrase(value, font));
        return cell;
    }
    /**
     * 为表格添加一个内容
     *
     * @param value   值
     * @param font    字体
     * @param align   对齐方式
     * @param colspan 占多少列
     * @return 添加的文本框
     */
    public static PdfPCell createCell(String value, Font font, int align, int colspan) {
        PdfPCell cell = new PdfPCell();
        cell.setVerticalAlignment(Element.ALIGN_MIDDLE);
        cell.setHorizontalAlignment(align);
        cell.setColspan(colspan);
        cell.setPhrase(new Phrase(value, font));
        return cell;
    }
    /**
     * 为表格添加一个内容
     *
     * @param value     值
     * @param font      字体
     * @param align     对齐方式
     * @param colspan   占多少列
     * @param boderFlag 是否有有边框
     * @return 添加的文本框
     */
    public static PdfPCell createCell(String value, Font font, int align, int colspan,
                                      boolean boderFlag) {
        PdfPCell cell = new PdfPCell();
        cell.setVerticalAlignment(Element.ALIGN_MIDDLE);
        cell.setHorizontalAlignment(align);
        cell.setColspan(colspan);
        cell.setPhrase(new Phrase(value, font));
        cell.setPadding(3.0f);
        if (!boderFlag) {
            cell.setBorder(0);
            cell.setPaddingTop(4.0f);
            cell.setPaddingBottom(8.0f);
        }
        return cell;
    }
    /**
     * 创建一个表格对象
     *
     * @param colNumber 表格的列数
     * @return 生成的表格对象
     */
    public static PdfPTable createTable(int colNumber) {
        PdfPTable table = new PdfPTable(colNumber);
        table.setTotalWidth(MAX_WIDTH);
        table.setLockedWidth(true);
        table.setHorizontalAlignment(Element.ALIGN_CENTER);
        table.getDefaultCell().setBorder(1);
        return table;
    }
    public PdfPTable createTable(float[] widths) {
        PdfPTable table = new PdfPTable(widths);
        try {
            table.setTotalWidth(MAX_WIDTH);
            table.setLockedWidth(true);
            table.setHorizontalAlignment(Element.ALIGN_CENTER);
            table.getDefaultCell().setBorder(1);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return table;
    }
    public static PdfPTable createBlankTable() {
        PdfPTable table = new PdfPTable(1);
        table.getDefaultCell().setBorder(0);
        table.addCell(createCell("", keyFont));
        table.setSpacingAfter(20.0f);
        table.setSpacingBefore(20.0f);
        return table;
    }
    /**
     * 提供外界调用的接口，生成以head为表头，list为数据的pdf
     *
     * @param fileName  文件名
     * @param titleName 表单名
     * @param headName  数据表头
     * @param headField 对应类的属性名
     * @param list      数据
     */
    public static <T> File generatePDFs(String fileName, String titleName, String[] headName, String[] headField,
                                        List<T> list) {
        // 建立一个Document对象
        Document document = new Document();
        // 设置页面大小
        document.setPageSize(PageSize.A4);
        File tempFile = null;
        try {
            tempFile = File.createTempFile(fileName, PDF_FILE_SUFFIX);
            PdfWriter.getInstance(document, new FileOutputStream(tempFile));
            document.open();
        } catch (Exception e) {
            e.printStackTrace();
        }
        Class classType = list.get(0).getClass();
        int colNum = headName.length;
        PdfPTable table = createTable(colNum);
        // 添加标题
        table.addCell(createCell(titleName, titleFont, Element.ALIGN_CENTER, colNum, false));
        //设置表头
        for (int i = 0; i < colNum; i++) {
            table.addCell(createCell(headName[i], keyFont, Element.ALIGN_CENTER));
        }
        for (int i = 0; i < list.size(); i++) {
            T t = list.get(i);
            for (int j = 0; j < colNum; j++) {
                //获得get方法,getName,getAge等
                String getMethodName = "get" + headField[j].substring(0, 1).toUpperCase() + headField[j].substring(1);
                try {
                    //通过反射获得相应的get方法，用于获得相应的属性值
                    Method method = classType.getMethod(getMethodName, new Class[]{});
                    logger.debug(getMethodName + ":" + method.invoke(t, new Class[]{}) + ",");
                    //添加数据
                    table.addCell(createCell(method.invoke(t, new Class[]{}).toString(), textFont));
                } catch (Exception e) {
                    logger.error("属性设值错误：入参={},异常={}", JSON.toJSONString(t), e);
                    throw new RuntimeException(e);
                }
            }
        }
        try {
            //将表格添加到文档中
            document.add(table);
        } catch (Exception e) {
            logger.error("");
            throw new RuntimeException(e);
        }
        //关闭流
        document.close();
        return tempFile;
    }
    /**
     * 构建行程单PDF文件
     * @param date 构建日期
     * @param phone 手机号
     * @param baseTripInfos 行程信息
     * @param outputStream 输出流
     */
    public static void createPDFFile(Date date, String phone, List<BaseTripInfo> baseTripInfos, OutputStream outputStream) {
        String[] header = new String[]{"序号", "出站时间", "入站站点", "出站站点", "金额（元）"};
        int[] width = new int[]{20, 45, 35, 35, 35};
        Document document = null;
        try {
            // 建立一个Document对象
            document = new Document();
            // 设置页面大小
            document.setPageSize(PageSize.A4);
            PdfWriter.getInstance(document, outputStream);
            document.open();
            // 设置列以及列宽度
            PdfPTable table = PDFUtil.createTable(header.length);
            table.setWidths(width);
            // 添加标题以及空行
            table.addCell(PDFUtil.createCell("广州地铁—行程单", PDFUtil.titleFont, Element.ALIGN_CENTER, header.length, false));
            table.addCell(PDFUtil.createCell(" ", PDFUtil.titleFont, Element.ALIGN_CENTER, header.length, false));
            // 设置申请日期以及行程起止日期
            table.addCell(PDFUtil.createCell("申请时间：" + cn.hutool.core.date.DateUtil.format(date, "yyyy-MM-dd"), PDFUtil.textFont, Element.ALIGN_LEFT, 2, false));
            table.addCell(PDFUtil.createCell("行程人手机号：" + phone, PDFUtil.textFont, Element.ALIGN_LEFT, 2, false));
            // 计算该行程单总金额
            double totalAmount = 0;
            for (BaseTripInfo baseTripInfo : baseTripInfos) {
                totalAmount += Double.parseDouble(baseTripInfo.getPrice());
            }
            table.addCell(PDFUtil.createCell("共" + baseTripInfos.size() + "笔行程, 合计 " + NumberUtil.round(totalAmount, 2).toString() + "元",
                    PDFUtil.textFont, Element.ALIGN_LEFT, header.length, false));
            // 设置表头
            for (String headName : header) {
                table.addCell(PDFUtil.createCell(headName, PDFUtil.keyFont, Element.ALIGN_CENTER));
            }
            for (int i = 0; i < baseTripInfos.size(); i++) {
                // 序号
                table.addCell(createTextCell(String.valueOf(i + 1)));
                // 出站时间
                table.addCell(createTextCell(baseTripInfos.get(i).getExitDate() == null ? "" :
                        baseTripInfos.get(i).getExitDate()));
                // 进站站点
                table.addCell(createTextCell(baseTripInfos.get(i).getEnterStation() == null ? "" :
                        baseTripInfos.get(i).getEnterStation()));
                // 出站站点
                table.addCell(createTextCell(baseTripInfos.get(i).getExitStation() == null ? "" :
                        baseTripInfos.get(i).getExitStation()));
                // 扣费金额
                double price = Double.parseDouble(baseTripInfos.get(i).getPrice());
                table.addCell(createTextCell(NumberUtil.round(price, 2).toString()));
            }
            //将表格添加到文档中
            document.add(table);
        } catch (DocumentException e) {
            logger.error("Document添加PdfWriter实例失败", e);
        } finally {
            //关闭流
            if (document != null) {
                document.close();
            }
        }
    }
    private static PdfPCell createTextCell(String value) {
        return PDFUtil.createCell(value, PDFUtil.textFont);
    }
}

/**
 * 行程单信息
 */
public class BaseTripInfo {
    /**
     * 行程ID
     */
    private String tripId;
    /**
     * 入站站点
     */
    private String enterStation;
    /**
     * 入站时间
     */
    private String enterDate;
    /**
     * 出站站点
     */
    private String exitStation;
    /**
     * 出站时间
     */
    private String exitDate;
    /**
     * 金额
     */
    private String price;
    /**
     * 设备ID
     */
    private String deviceId;
    public String getTripId() {
        return tripId;
    }
    public void setTripId(String tripId) {
        this.tripId = tripId;
    }
    public String getEnterStation() {
        return enterStation;
    }
    public void setEnterStation(String enterStation) {
        this.enterStation = enterStation;
    }
    public String getEnterDate() {
        return enterDate;
    }
    public void setEnterDate(String enterDate) {
        this.enterDate = enterDate;
    }
    public String getExitStation() {
        return exitStation;
    }
    public void setExitStation(String exitStation) {
        this.exitStation = exitStation;
    }
    public String getExitDate() {
        return exitDate;
    }
    public void setExitDate(String exitDate) {
        this.exitDate = exitDate;
    }
    public String getPrice() {
        return price;
    }
    public void setPrice(String price) {
        this.price = price;
    }
    public String getDeviceId() {
        return deviceId;
    }
    public void setDeviceId(String deviceId) {
        this.deviceId = deviceId;
    }
    @Override
    public String toString() {
        return "BaseTripInfo{" +
                "tripId='" + tripId + '\'' +
                ", enterStation='" + enterStation + '\'' +
                ", enterDate='" + enterDate + '\'' +
                ", exitStation='" + exitStation + '\'' +
                ", exitDate='" + exitDate + '\'' +
                ", price='" + price + '\'' +
                ", deviceId='" + deviceId + '\'' +
                '}';
    }
}