Spring Boot集成OpenPDF和Freemarker实现PDF导出功能并附水印

import org.springframework.stereotype.Service;
import org.springframework.core.io.InputStreamResource;
import org.springframework.core.io.Resource;
 
import com.openhtmltopdf.pdfboxout.PdfRendererBuilder;
import com.openhtmltopdf.pdfboxout.support.Watermark;
 
import java.io.*;
 
@Service
public class PdfExportService {
 
    public Resource generatePdfWithWatermark(String htmlContent, String watermarkText) {
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        try {
            // 使用OpenPDF和Freemarker生成HTML内容
            String processedHtmlContent = processHtmlTemplate(htmlContent);
 
            // 使用OpenPDF将HTML转换为PDF
            PdfRendererBuilder builder = new PdfRendererBuilder();
            builder.useFastMode();
            builder.withWatermark(new Watermark(watermarkText, 45, Color.GRAY));
            builder.withHtmlContent(processedHtmlContent, "file:///")
                    .toStream(outputStream);
            builder.run();
 
            // 将PDF内容封装为Spring的Resource对象
            return new InputStreamResource(new ByteArrayInputStream(outputStream.toByteArray()));
        } catch (Exception e) {
            e.printStackTrace();
            // 异常处理逻辑
        } finally {
            try {
                outputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        // 异常或正常结束时的返回值逻辑
        return null;
    }
 
    private String processHtmlTemplate(String htmlContent) {
        // 使用Freemarker处理HTML模板，生成最终的HTML字符串
        // 这里是模拟的Freemarker处理逻辑
        return htmlContent;
    }
}

这段代码展示了如何在Spring Boot应用中使用OpenPDF和Freemarker来生成带有水印的PDF文件。首先，它创建了一个ByteArrayOutputStream来暂存PDF文件内容。然后，它使用OpenPDF的PdfRendererBuilder来处理HTML内容，并为PDF文档添加水印。最后，它将生成的PDF内容封装为Spring的Resource对象，以便可以在控制器中返回并下载给客户端。