這篇文章主要介紹了node.js實(shí)現(xiàn)為PDF添加水印的示例代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

背景

項(xiàng)目中有用到node做中間層，期間有個(gè)為PDF文件添加水印的需求，網(wǎng)上搜了不少帖子，發(fā)現(xiàn)有以下幾點(diǎn)存在的問(wèn)題。

1、單頁(yè)P(yáng)DF的話，可以用PNG圖片加到PDF上，但是實(shí)際使用情況就沒(méi)那么理想，必須滿足多頁(yè)P(yáng)DF每頁(yè)都加上水印的效果方可。

2、中間想過(guò)，把PDF轉(zhuǎn)成圖片，圖片加上水印后再轉(zhuǎn)回PDF，但是感覺中間步驟太麻煩，而且會(huì)面臨清晰度下降的可能。

3、Python、Java有相關(guān)方案，可以實(shí)現(xiàn)，有想過(guò)用node去執(zhí)行Python腳本去為PDF加水印，考慮到開發(fā)人員都是前端、如果以后需要維護(hù)，擴(kuò)展可能就不是那么友好了。

解決方案

逛了Github和NPM，發(fā)現(xiàn)node-pdftk可以滿足需求，不過(guò)得依賴PDFTK這個(gè)軟件，進(jìn)一步了解，發(fā)現(xiàn)PDFTK支持linux，好，直接用這個(gè)開搞。

環(huán)境

項(xiàng)目部署是用了docker，那就用docker來(lái)吧，比較貼合情況也省事點(diǎn)；如果平時(shí)在自己電腦上開發(fā)，但需要linux環(huán)境的話練手或者模擬服務(wù)器環(huán)境的話，大家也可以用一個(gè)linux鏡像運(yùn)行一個(gè)dokcer容器，這里就不贅述了。

實(shí)現(xiàn)

1、app.js

API相對(duì)較為簡(jiǎn)單，input方法中我們輸入源文件，stamp方法中我們輸入水印文件，out方法中指定保存的路徑，這里的話我們起了一個(gè)服務(wù)，當(dāng)訪問(wèn)3100端口的時(shí)候就去生成我們帶水印的PDF。

'use strict';

const express = require('express');
const pdftk = require('node-pdftk');
const path = require('path');
const app = express();

app.get('/', (req, res) => {
  pdftk
    .input(path.resolve(__dirname + '/../files/javascript.pdf'))
    .stamp(path.resolve(__dirname + '/../files/watermark.pdf'))
    .output(path.resolve(__dirname + '/../files/javascript_watermark.pdf'))
    .then(buffer => {
      return res.send('successful');
    })
    .catch(err => {
      console.error(err);
      return res.send('failed');
    });
});

app.listen(3100);

2、dockerfile

FROM node:7.6.0

# 創(chuàng)建工作目錄
RUN mkdir -p /usr/src/app
WORKDIR /usr/src/app

COPY ./node_modules /usr/src/app/node_modules

RUN npm set registry https://registry.npm.taobao.org/

# 下載安裝PDFTK
RUN apt-get update
RUN apt-get install -y pdftk

# 使用PM2啟動(dòng)應(yīng)用
CMD ["./node_modules/pm2/bin/pm2-docker", "process.yml"]

3、效果

源文件