Question

我正在尝试使用canvas模块在node.js脚本中创建PNG文件。

我有一个文件image.js用于处理图像创建：

const { createCanvas, loadImage } = require('canvas');
const fs = require('fs');

function generateImage(param1, param2) {
  const canvas = createCanvas(4096, 4096);
  const ctx = canvas.getContext('2d');
  // ...
  return new Promise(function (resolve, reject) {
    loadImage('base.png').then((image) => {
      ctx.drawImage(image, 40, 40, 200, 200);

      let out = fs.createWriteStream(__dirname + '/temp.png');
      let stream = canvas.pngStream();

      stream.on('data', function (chunk) {
        out.write(chunk);
      });

      stream.on('end', function () {
        console.log('saved png');
        out.close(); // i probably don't need both, but neither seems to work properly...
        out.end();
        resolve();
      });
    });
  });
}

module.exports = generateImage;

当我在文件的末尾附加generateImage('a', 'b');并单独运行（node image.js）时，它可以按预期工作并生成我的图像。但是，当我尝试从另一个文件（在同一文件夹中）调用它时，似乎文件句柄没有关闭，并且在脚本运行时磁盘上的图像是空文件。

main.js（简体）：

const generateImage = require('./image');
const fs = require('fs');

async function main() {
    var stats = fs.statSync("temp.png");
    console.log(stats["size"]); // output: 0, expected: > 0
}

generateImage("text", "text").then(() => main());

我是node.js的新手，所以很可能我遗漏了一些明显的东西。

Answer 1

根据Bergi的建议，我需要为out流的close事件添加另一个事件侦听器，并且仅在触发后调用resolve()。我也将诺言链接起来，而不是像在问题中那样嵌入诺言。

function generateImage(param1, param2) {
  const canvas = createCanvas(4096, 4096);
  const ctx = canvas.getContext('2d');
  // ...
  return loadImage('base.png').then((image) => new Promise(function (resolve, reject) {
    ctx.drawImage(image, 40, 40, 200, 200);

    let out = fs.createWriteStream(__dirname + '/temp.png');
    let stream = canvas.pngStream();

    stream.on('data', (chunk) => out.write(chunk));
    stream.on('end', () => out.end());
    out.on('close', () => resolve());  // this was missing
  }));
}

如何用承诺关闭文件句柄？

1 个答案: