javascript - 使用 nodejs aws sdk 将生成的 pdf 上传到 AWS S3

Question

我正在使用 pdfkit 生成包含一些自定义内容的 pdf，然后将其发送到 AWS S3 存储桶。

虽然如果我将文件作为一个整体生成并上传它可以完美地工作，但是，如果我想将生成的文件作为八位字节流进行流式传输，我将找不到任何相关的指针。

我正在寻找 nodejs 解决方案（或建议）。

Answer 1

我会尽量在这里准确。我不会详细介绍pdfKit的 nodejs sdk 的用法。

如果您希望生成的 pdf 作为文件。

var PDFDocument = require('pdfkit');

// Create a document
doc = new PDFDocument();

// Pipe it's output somewhere, like to a file or HTTP response
doc.pipe(fs.createWriteStream('output.pdf'));
doc.text('Whatever content goes here');
doc.end();
var params = {
  key : fileName,
  body : './output.pdf',
  bucket : 'bucketName',
  contentType : 'application/pdf'
}

s3.putObject(params, function(err, response) {

});

但是，如果您想流式传输它（在问题的上下文中说 S3 存储桶），那么值得记住每个 pdfkit 实例都是一个可读流。

S3 需要文件、缓冲区或可读流。所以，

var doc = new PDFDocument();

// Pipe it's output somewhere, like to a file or HTTP response
doc.text("Text for your PDF");
doc.end();

var params = {
  key : fileName,
  body : doc,
  bucket : 'bucketName',
  contentType : 'application/pdf'
}

//notice use of the upload function, not the putObject function
s3.upload(params, function(err, response) {

});

Answer 2

如果您使用的是 html-pdf 包和 aws-sdk，那么这很容易......

var pdf = require('html-pdf');
import aws from 'aws-sdk';
const s3 = new aws.S3();

pdf.create(html).toStream(function(err, stream){
  stream.pipe(fs.createWriteStream('foo.pdf'));
  const params = {
                Key: 'foo.pdf',
                Body: stream,
                Bucket: 'Bucket Name',
                ContentType: 'application/pdf',
            };
  s3.upload(params, (err, res) => {
                if (err) {
                    console.log(err, 'err');
                }
                console.log(res, 'res');
            });
});

Answer 3

试过这个并且工作。我创建了一个 readFileSync，然后将其上传到 S3。我还使用了“writeStream.on('finish'”，以便在上传 pdf 文件之前完全创建它，否则它会上传部分文件。

const PDFDocument = require('pdfkit');
const fs = require('fs');
const AWS = require('aws-sdk');
const path = require('path')

async function createPDF() {


const doc = new PDFDocument({size: 'A4'});
let writeStream = fs.createWriteStream('./output.pdf')
doc.pipe(writeStream);


// Finalize PDF file
doc.end();

writeStream.on('finish', function () {
    var appDir = path.dirname(require.main.filename);
    const fileContent = fs.readFileSync(appDir + '/output.pdf');
    var params = {
        Key : 'filName',
        Body : fileContent,
        Bucket : process.env.AWS_BUCKET,
        ContentType : 'application/pdf',
        ACL: "public-read"
      }
      
    const s3 = new AWS.S3({
        accessKeyId: process.env.AWS_ACCESS_KEY,
        secretAccessKey: process.env.AWS_SECRET_KEY
    });
      //notice use of the upload function, not the putObject function
    s3.upload(params, function(err, response) {
        
    });
});

}