揭秘：Node.js 如何实现 PDF 批量生成并导出 ZIP 压缩包

2023-10-01 09:50:40

导言

在日常工作或学习中，我们经常需要将大量数据或文档导出为 PDF 格式，以便于查阅、分享或归档。而当 PDF 文件数量较多时，手动逐个生成并导出 PDF 文件显然效率低下。因此，开发一个能够批量生成 PDF 并导出 ZIP 压缩包的工具就显得十分必要。

本文将详细介绍如何使用 Node.js 实现 PDF 批量生成并导出 ZIP 压缩包。我们将从技术选型开始，然后逐步讲解实现步骤并提供代码示例。希望本文能够帮助大家快速掌握这一实用的技能。

技术选型

在开始实现之前，我们需要先选择合适的技术栈。对于 PDF 生成，我们选择了 PDFKit 库。PDFKit 是一个基于 Node.js 的 PDF 生成库，它可以轻松地将数据或 HTML 转换为 PDF 格式。

对于 ZIP 压缩，我们选择了 JSZip 库。JSZip 是一个用于在浏览器中创建和解压缩 ZIP 文件的库。它可以轻松地将多个文件压缩成一个 ZIP 压缩包。

实现步骤

1. 安装依赖

首先，我们需要安装 PDFKit 和 JSZip 库。可以使用以下命令：

npm install pdfkit jszip

2. 准备数据

在生成 PDF 之前，我们需要先准备要生成 PDF 的数据。这些数据可以是文本、HTML、JSON 等任何格式。为了便于理解，我们这里使用一个简单的 JSON 数据作为示例：

[
  {
    "title": "标题一",
    "content": "内容一"
  },
  {
    "title": "标题二",
    "content": "内容二"
  },
  {
    "title": "标题三",
    "content": "内容三"
  }
]

3. 生成 PDF

接下来，我们可以使用 PDFKit 库来生成 PDF 文件。这里我们以第一个数据项为例，生成一个名为 "title1.pdf" 的 PDF 文件：

const PDFDocument = require('pdfkit');
const doc = new PDFDocument();

doc.pipe(fs.createWriteStream('title1.pdf'));

doc.text(data[0].title);
doc.text(data[0].content);

doc.end();

4. 导出 ZIP 压缩包

最后，我们可以使用 JSZip 库将生成的 PDF 文件压缩成一个 ZIP 压缩包。这里我们以 "output.zip" 为例：

const JSZip = require('jszip');
const zip = new JSZip();

zip.file('title1.pdf', fs.readFileSync('title1.pdf'));

zip.generateAsync({type: 'nodebuffer'}).then(function(content) {
  fs.writeFileSync('output.zip', content);
});

5. 完整代码

将以上步骤组合起来，完整的代码如下：

const PDFDocument = require('pdfkit');
const JSZip = require('jszip');
const fs = require('fs');

const data = [
  {
    "title": "标题一",
    "content": "内容一"
  },
  {
    "title": "标题二",
    "content": "内容二"
  },
  {
    "title": "标题三",
    "content": "内容三"
  }
];

data.forEach(item => {
  const doc = new PDFDocument();

  doc.pipe(fs.createWriteStream(`${item.title}.pdf`));

  doc.text(item.title);
  doc.text(item.content);

  doc.end();
});

const zip = new JSZip();

data.forEach(item => {
  zip.file(`${item.title}.pdf`, fs.readFileSync(`${item.title}.pdf`));
});

zip.generateAsync({type: 'nodebuffer'}).then(function(content) {
  fs.writeFileSync('output.zip', content);
});