当前位置：首页 > 文章列表 > 文章 > 前端 > 如何在Java中通过输入流高效获取PDF总页数

如何在Java中通过输入流高效获取PDF总页数

2026-05-24 18:43:03 0浏览收藏

哈喽！今天心血来潮给大家带来了《如何在Java中通过输入流高效获取PDF总页数》，想必大家应该对文章都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习文章，千万别错过这篇文章~希望能帮助到你！

本文介绍使用Apache PDFBox库，从InputStream（如网络流、文件流）中无需加载全文即可快速、可靠地获取PDF文档总页数，适用于大文件或内存受限场景。

本文介绍使用Apache PDFBox库，从InputStream（如网络流、文件流）中无需加载全文即可快速、可靠地获取PDF文档总页数，适用于大文件或内存受限场景。

在实际开发中，尤其是处理来自HTTP响应、云存储或用户上传的PDF时，我们往往无法或不应将整个PDF文件一次性读入内存（例如超大扫描件或批量处理场景）。此时，直接操作InputStream并精准提取页数，是兼顾性能与稳定性的关键能力。

Apache PDFBox（推荐使用 2.0.24+ 或最新 3.x 版本）原生支持流式解析：它仅需读取PDF文件的交叉引用表（xref）和 trailer 信息，即可确定页面树结构及总页数，无需解码内容、渲染页面或加载全部对象——这意味着即使是一个500MB的PDF，也只需几KB的IO即可返回准确页数。

✅ 正确做法：使用 PDDocument.load(InputStream) + getNumberOfPages()

以下为生产就绪的Java实现（含资源安全释放与异常防护）：

import org.apache.pdfbox.pdmodel.PDDocument;
import java.io.IOException;
import java.io.InputStream;

public class PdfPageCounter {

    /**
     * 从输入流安全获取PDF总页数
     * @param inputStream PDF源流（支持FileInputStream、ByteArrayInputStream、HTTP响应流等）
     * @return 页数（失败时返回-1，不抛出异常）
     */
    public static int getPageCount(InputStream inputStream) {
        if (inputStream == null) {
            return -1;
        }
        PDDocument document = null;
        try {
            // PDFBox自动检测流类型并进行最小化解析
            document = PDDocument.load(inputStream);
            return document.getNumberOfPages();
        } catch (IOException e) {
            // 常见异常：损坏PDF、加密PDF（无密码）、流已关闭、格式不合规
            System.err.println("Failed to read PDF page count: " + e.getMessage());
            return -1;
        } finally {
            // 必须显式关闭，避免内存泄漏和文件句柄占用
            if (document != null) {
                try {
                    document.close();
                } catch (IOException ignored) {
                    // close()异常可忽略，核心逻辑已完成
                }
            }
        }
    }
}

? 注意事项与避坑指南

不要手动缓冲流：避免用BufferedInputStream包装原始流后再传入PDFBox——PDFBox内部已做最优缓冲，额外包装可能干扰底层偏移计算，导致IOException: Error: End-of-File, expected line等错误。
加密PDF需额外处理：若PDF受密码保护，PDDocument.load(inputStream)会抛出InvalidPasswordException。如需支持，应改用PDDocument.load(inputStream, password)并捕获该异常。
流必须可重置（mark/reset）？否！ PDFBox 2.0+ 已完全支持不可标记流（如HttpURLConnection.getInputStream()），无需调用mark()/reset()，可直接传入。
对比其他方案：
- ❌ 正则匹配 /Type\s*/Page：极易误判（注释、字符串、嵌入对象中存在类似文本），且无法处理压缩/二进制PDF；
- ❌ pdf-parse（Node.js）：仅适用于JavaScript生态，且其流式支持依赖自定义Readable封装（如答案所示），本质仍是将流转为buffer再解析，不适用于Java；
- ✅ PDFBox是JVM生态事实标准，经多年生产验证，兼容ISO 32000-1/2全规格PDF。

✅ 完整调用示例（含HTTP流）

import java.io.InputStream;
import java.net.URL;
import javax.net.ssl.HttpsURLConnection;

public class StreamPageCountDemo {
    public static void main(String[] args) throws Exception {
        // 示例1：本地文件流
        try (InputStream fis = new FileInputStream("report.pdf")) {
            int pages = PdfPageCounter.getPageCount(fis);
            System.out.println("Local PDF pages: " + pages);
        }

        // 示例2：HTTP响应流（无需下载完整文件）
        URL pdfUrl = new URL("https://example.com/doc.pdf");
        HttpsURLConnection conn = (HttpsURLConnection) pdfUrl.openConnection();
        conn.setRequestMethod("GET");
        try (InputStream httpStream = conn.getInputStream()) {
            int pages = PdfPageCounter.getPageCount(httpStream);
            System.out.println("Remote PDF pages: " + pages);
        }
    }
}