图片转PDF下载

2023-10-29

把图片转换为PDF文件，推流到前端，再下载。

主要使用 itextpdf 包实现。

PdfUtil

实现图片转PDF，实现推流

import com.itextpdf.text.Document;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.Image;
import com.itextpdf.text.PageSize;
import com.itextpdf.text.pdf.PdfWriter;
import com.youyu.base.exception.YouyuException;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URL;
import java.net.URLEncoder;
import java.nio.file.Files;
import java.nio.file.Paths;

/**
 * @author 曳戈泰尔
 * @version 1.0
 * @description 图片转PDF工具类
 * @date 2023/7/20 6:42 PM
 */
@Slf4j
public class PdfUtil {

    /**
     * 图片转换PDF的公共接口
     *
     * @param file SpringMVC获取的图片文件
     * @param response HttpServletResponse
     * @throws IOException IO异常
     * @throws DocumentException PDF文档异常
     */
    public static void imageToPdf(MultipartFile file, HttpServletResponse response)
            throws IOException, DocumentException {

        // 生成PDF文件
        File pdfFile = generatePdfFile(file);

        // 下载PDF文件
        downloadPdfFile(pdfFile, response);
    }

    /**
     * 图片转换PDF的公共接口
     *
     * @param file 图片文件
     * @param response HttpServletResponse
     * @throws IOException IO异常
     * @throws DocumentException PDF文档异常
     */
    public static void imageToPdf(File file, HttpServletResponse response)
            throws IOException, DocumentException {

        // 生成PDF文件
        File pdfFile = generatePdfFile(file);

        // 下载PDF文件
        downloadPdfFile(pdfFile, response);

        // 直接下载图片文件
        // downloadPdfFile(file, response);
    }

    /**
     * 将图片转换为PDF文件
     *
     * @param file SpringMVC获取的图片文件
     * @return PDF文件
     * @throws IOException IO异常
     * @throws DocumentException PDF文档异常
     */
    private static File generatePdfFile(MultipartFile file) throws IOException, DocumentException {

        // 生成文件名
        String fileName = file.getOriginalFilename();
        if (StringUtils.isBlank(fileName)) {
            log.error("文件名为空");
            throw new YouyuException("下载证书文件名为空，请检查后重试！");
        }
        String pdfFileName = "/tmp/" + fileName.substring(0, fileName.lastIndexOf(".")) + ".pdf";

        // 生成PDF文件
        Document doc = new Document(PageSize.A4, 20, 20, 20, 20);
        PdfWriter.getInstance(doc, new FileOutputStream(pdfFileName));
        doc.open();

        // 添加图片到PDF
        doc.newPage();
        Image image = Image.getInstance(file.getBytes());
        float height = image.getHeight();
        float width = image.getWidth();
        int percent = getPercent(height, width);
        image.setAlignment(Image.MIDDLE);
        image.scalePercent(percent);
        doc.add(image);
        doc.close();

        // 返回PDF文件 FIle格式
        return new File(pdfFileName);
    }

    /**
     * 将图片转换为PDF文件
     *
     * @param file 图片文件
     * @return PDF文件
     * @throws IOException IO异常
     * @throws DocumentException PDF文档异常
     */
    private static File generatePdfFile(File file) throws IOException, DocumentException {

        // 生成文件名
        String fileName = file.getName();
        if (StringUtils.isBlank(fileName)) {
            log.error("文件名为空");
            throw new YouyuException("下载证书文件名为空，请检查后重试！");
        }
        String pdfFileName = "/tmp/" + fileName.substring(0, fileName.lastIndexOf(".")) + ".pdf";

        // 生成PDF文件
        Document doc = new Document(PageSize.A4, 20, 20, 20, 20);
        PdfWriter.getInstance(doc, new FileOutputStream(pdfFileName));
        doc.open();

        // 获取图片字节流
        byte[] bytes = Files.readAllBytes(Paths.get(file.getPath()));

        // 添加图片到PDF
        doc.newPage();
        Image image = Image.getInstance(bytes);
        float height = image.getHeight();
        float width = image.getWidth();
        int percent = getPercent(height, width);
        image.setAlignment(Image.MIDDLE);
        image.scalePercent(percent);
        doc.add(image);
        doc.close();

        // 删除临时图片文件
        FileUtils.deleteQuietly(file);

        // 返回PDF文件 FIle格式
        return new File(pdfFileName);
    }

    /**
     * 用于下载PDF文件
     *
     * @param pdfFile PDF文件
     * @param response HttpServletResponse
     * @throws IOException IO异常
     */
    private static void downloadPdfFile(File pdfFile, HttpServletResponse response)
            throws IOException {

        // 获取PDF文件字节流
        byte[] bytes = Files.readAllBytes(Paths.get(pdfFile.getPath()));

        // 设置响应头
        response.reset();
        response.setHeader("Content-Type", "application/pdf");
        response.setHeader(
                "Content-Disposition",
                "attachment; filename=" + URLEncoder.encode(pdfFile.getName(), "UTF-8"));
        response.setHeader("Content-Length", String.valueOf(bytes.length));
        // 设置暴露文件头
        response.setHeader("Access-Control-Expose-Headers", "Content-Disposition");

        // 写入响应流
        OutputStream out = response.getOutputStream();
        out.write(bytes);
        out.flush();
        out.close();

        // 删除临时pdf文件
        FileUtils.deleteQuietly(pdfFile);
    }

    /**
     * 等比压缩，获取压缩百分比
     *
     * @param height 图片的高度
     * @param weight 图片的宽度
     * @return 压缩百分比
     */
    private static int getPercent(float height, float weight) {

        // 压缩百分比
        float percent;
        if (height > weight) {
            percent = PageSize.A4.getHeight() / height * 100;
        } else {
            percent = PageSize.A4.getWidth() / weight * 100;
        }

        // 向下取整
        return (int) Math.floor(percent) - 2;
    }

    /**
     * 链接转换File
     *
     * @param fileUrl 链接
     * @return File
     */
    public static File fileUrlToFile(String fileUrl) {

        // 定义File对象
        File file = new File("/tmp/" + fileUrl.substring(fileUrl.lastIndexOf("/") + 1));
        try {
            // 将链接转换为File
            URL url = new URL(fileUrl);
            FileUtils.copyURLToFile(url, file);

            // 返回File
            return file;
        } catch (IOException e) {
            log.error("url转File失败，异常：{}", e.getMessage());

            throw new YouyuException("url转File失败，请检查后重试！");
        }
    }
}

BrowserServiceImpl

调用PdfUtil，实现证书下载

// 查询证书url路径
String fileUrl = fileBaseUrl + storageEntity.getCertificateUrl();

try {
    // 网络url转File
    File file = PdfUtil.fileUrlToFile(fileUrl);
    // 转换为PDF返回前端下载
    PdfUtil.imageToPdf(file, response);
} catch (IOException | DocumentException e) {
    log.error(
            "certificateDownload imageToPdf error, worksId: {}",
            certificateDownloadReq.getWorksId(),
            e);
    throw new YouyuException("证书生成失败，请检查后重试！");
}