iText PDF合并优化：避免内存溢出的高效流式处理

public static byte[] mergePdf(List < InputStream > inputStreams) {
    Document document = new Document();
    ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); // 内存占用源
    PdfCopy copy = new PdfSmartCopy(document, byteArrayOutputStream);
    document.open();
    for (InputStream inputStream: inputStreams) {
        PdfReader pdfReader = new PdfReader(inputStream);
        copy.addDocument(pdfReader);
        copy.freeReader(pdfReader); // 释放PdfReader的内部缓存
        pdfReader.close();
    }
    document.close();
    return byteArrayOutputStream.toByteArray(); // 在此处将整个PDF转换为字节数组
}

import com.itextpdf.text.Document;
import com.itextpdf.text.pdf.PdfCopy;
import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.PdfSmartCopy;

import java.io.InputStream;
import java.io.OutputStream;
import java.io.IOException;
import java.util.List;

public class PdfMergeUtil {

    /**
     * 将多个PDF输入流合并到一个输出流中，避免内存溢出。
     *
     * @param inputStreams 待合并的PDF输入流列表
     * @param outputStream 目标输出流，合并后的PDF将直接写入此流
     * @throws IOException 如果在PDF处理过程中发生I/O错误
     */
    public static void mergePdfsToStream(List<InputStream> inputStreams, OutputStream outputStream) throws IOException {
        Document document = null;
        PdfCopy copy = null;
        try {
            document = new Document();
            // 直接将PdfCopy连接到目标OutputStream
            copy = new PdfSmartCopy(document, outputStream);
            document.open();

            for (InputStream inputStream : inputStreams) {
                PdfReader pdfReader = null;
                try {
                    pdfReader = new PdfReader(inputStream);
                    copy.addDocument(pdfReader);
                    copy.freeReader(pdfReader); // 释放PdfReader的内部缓存
                } finally {
                    // 确保PdfReader和其底层的InputStream被关闭
                    if (pdfReader != null) {
                        pdfReader.close();
                    }
                    if (inputStream != null) {
                        try {
                            inputStream.close(); // 关闭原始输入流
                        } catch (IOException e) {
                            // 记录或处理关闭异常
                            System.err.println("Error closing input stream: " + e.getMessage());
                        }
                    }
                }
            }
        } finally {
            // 确保Document和OutputStream被关闭
            if (document != null && document.isOpen()) {
                document.close();
            }
            // 注意：此处不关闭outputStream，因为调用者可能需要继续使用或由其负责关闭
            // 如果outputStream是方法内部创建的，则应在此处关闭
        }
    }

    // 示例用法：将合并后的PDF写入文件
    /*
    public static void main(String[] args) {
        List<InputStream> pdfInputStreams = new ArrayList<>();
        try {
            // 假设你有多个PDF文件，转换为InputStream
            pdfInputStreams.add(new FileInputStream("path/to/pdf1.pdf"));
            pdfInputStreams.add(new FileInputStream("path/to/pdf2.pdf"));
            // ...

            FileOutputStream fos = new FileOutputStream("merged_output.pdf");
            mergePdfsToStream(pdfInputStreams, fos);
            System.out.println("PDFs merged successfully to merged_output.pdf");
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            // 确保所有输入流和输出流都被关闭
            for (InputStream is : pdfInputStreams) {
                try {
                    if (is != null) is.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            // fos在try-with-resources或单独的finally块中关闭
        }
    }
    */
}