0

我在创建 pdf 的 lamda 函数中使用 pdfkit,然后应该将 pdf 上传到 S3 存储桶。但是当我测试函数时,我得到错误:无法确定 [object PDFDocument] 的长度

这是我的功能:

var PDFDocument = require('pdfkit');
var AWS = require('aws-sdk');
process.env['PATH'] = process.env['PATH'] + ':' +         
process.env['LAMBDA_TASK_ROOT'];
exports.handler = function(event, context) {

// create a document and pipe to a blob
var doc = new PDFDocument();

// draw some text
doc.fontSize(25)
.text('Hello World', 100, 80);

var params = {
  Bucket : "test-bucket",
  Key : event.pdf_name + ".pdf",
  Body : doc
}

var s3 = new AWS.S3();
s3.putObject(params, function(err, data) {
  if (err) {
    console.log(err)
  } else {
    context.done(null, { status: 'pdf created' });
    doc.end();
  }
 });

};

我究竟做错了什么?如果需要,我如何提供文件大小?这是执行此操作的好方法还是有更好的方法将 pdf 文件流上传到 s3 存储桶?

4

1 回答 1

2

这是我的解决方案:

const PDFDocument = require('pdfkit');
const fs = require("fs");
const AWS = require('aws-sdk');
const s3 = new AWS.S3();

exports.handler = function (event, callback) {
    let doc = new PDFDocument;
    let fileName = "yourfile.pdf";

    //We use Lambdas temp folder to store file temporarily.
    //When Lambda execution ends temp is flushed
    let file = fs.createWriteStream("/tmp/" + fileName);

    doc.pipe(file);
    doc.text("hello");
    // # Finalize PDF file 
    doc.end();

    // Send pdf file to s3
    file.on("finish", function () {

        //get the file size
        const stats = fs.statSync("/tmp/" + fileName);
        console.log("filesize: " + stats.size);

        console.log("starting s3 putObject");
        s3.putObject({
            Bucket: "[your-bucket]",
            Key: fileName,
            Body: fs.createReadStream("/tmp/" + fileName),
            ContentType: "application/pdf",
            ContentLength: stats.size,
        }, function (err) {
            if (err) {
                console.log(err, err.stack);
                callback(err);
            } else {
                console.log("Done");
                callback(null, "done");
            }
        });
    });
}

该解决方案的关键要素是使用文件流和 lambda 临时文件夹。file.on("finish") 用于实际检查文件写入是否结束。

于 2018-11-12T12:41:47.277 回答