使用pdfcpu高效处理PDF文档

pdfcpu是一个用Go编写的强大PDF处理库，它提供了丰富的功能来处理PDF文档，包括合并、拆分、加密、解密、添加水印等操作。下面我将介绍pdfcpu的主要功能并提供示例代码。

安装pdfcpu

首先安装pdfcpu库：

go get github.com/pdfcpu/pdfcpu/pkg/api
go get github.com/pdfcpu/pdfcpu/pkg/pdfcpu

基本功能示例

1. 合并PDF文件

package main

import (
	"fmt"
	"github.com/pdfcpu/pdfcpu/pkg/api"
)

func main() {
	// 输入文件列表
	inputFiles := []string{"file1.pdf", "file2.pdf"}
	// 输出文件
	outputFile := "merged.pdf"

	// 合并PDF
	if err := api.MergeCreateFile(inputFiles, outputFile, nil); err != nil {
		fmt.Printf("合并PDF失败: %v\n", err)
		return
	}
	fmt.Println("PDF合并成功!")
}

2. 拆分PDF文件

package main

import (
	"fmt"
	"github.com/pdfcpu/pdfcpu/pkg/api"
)

func main() {
	inputFile := "input.pdf"
	outputDir := "./split_pages"

	// 按页拆分PDF
	if err := api.SplitFile(inputFile, outputDir, 1, nil); err != nil {
		fmt.Printf("拆分PDF失败: %v\n", err)
		return
	}
	fmt.Println("PDF拆分成功!")
}

3. 提取PDF页面

package main

import (
	"fmt"
	"github.com/pdfcpu/pdfcpu/pkg/api"
)

func main() {
	inputFile := "input.pdf"
	outputFile := "extracted.pdf"
	
	// 选择要提取的页面(例如提取1-3页)
	selectedPages := []string{"1-3"}

	if err := api.ExtractPagesFile(inputFile, outputFile, selectedPages, nil); err != nil {
		fmt.Printf("提取页面失败: %v\n", err)
		return
	}
	fmt.Println("页面提取成功!")
}

高级功能示例

4. 添加水印

package main

import (
	"fmt"
	"github.com/pdfcpu/pdfcpu/pkg/api"
	"github.com/pdfcpu/pdfcpu/pkg/pdfcpu"
)

func main() {
	inputFile := "input.pdf"
	outputFile := "watermarked.pdf"
	watermarkText := "CONFIDENTIAL"

	// 配置水印
	wm, err := pdfcpu.ParseTextWatermarkDetails(watermarkText, "scale:0.5, rot:45, op:0.5", true)
	if err != nil {
		fmt.Printf("创建水印失败: %v\n", err)
		return
	}

	// 添加水印
	if err := api.AddWatermarksFile(inputFile, outputFile, nil, wm, nil); err != nil {
		fmt.Printf("添加水印失败: %v\n", err)
		return
	}
	fmt.Println("水印添加成功!")
}

5. 加密PDF文件

package main

import (
	"fmt"
	"github.com/pdfcpu/pdfcpu/pkg/api"
)

func main() {
	inputFile := "input.pdf"
	outputFile := "encrypted.pdf"
	password := "secret123"

	// 配置加密选项
	conf := api.NewAESConfiguration(password, password, 256)

	// 加密PDF
	if err := api.EncryptFile(inputFile, outputFile, conf); err != nil {
		fmt.Printf("加密PDF失败: %v\n", err)
		return
	}
	fmt.Println("PDF加密成功!")
}

6. 解密PDF文件

package main

import (
	"fmt"
	"github.com/pdfcpu/pdfcpu/pkg/api"
)

func main() {
	inputFile := "encrypted.pdf"
	outputFile := "decrypted.pdf"
	password := "secret123"

	// 配置解密选项
	conf := api.NewAESConfiguration(password, password, 256)

	// 解密PDF
	if err := api.DecryptFile(inputFile, outputFile, conf); err != nil {
		fmt.Printf("解密PDF失败: %v\n", err)
		return
	}
	fmt.Println("PDF解密成功!")
}

性能优化建议

1. 批量处理：对于大量PDF操作，考虑使用批处理模式
2. 内存管理：处理大文件时注意内存使用，可以分块处理
3. 并发处理：对于独立操作可以使用goroutine并行处理

pdfcpu是一个功能强大且高效的PDF处理库，它提供了丰富的API来满足各种PDF处理需求。通过合理使用这些功能，可以轻松实现PDF文档的自动化处理。

更多高级功能和详细配置可以参考pdfcpu的官方文档：https://github.com/pdfcpu/pdfcpu