NodeJS Plotly - 如何在继续之前等待 getImage 完成

Question

所以，我想要完成的基本上是：通过使用 NodeJS 和 plotly 生成 plot 并将其放入我的文件系统（使用getImage ），然后继续处理图像。 我的问题是，function 在完成并将图像放入我的文件系统之前返回。 所以我的结果基本上是在我调用generate_plot()之后，该文件还不存在并且我得到了一个错误。 现在我的问题是：如何等待generate_plot()完成并将图像放入我的文件系统，然后再继续使用此图像/使用此图像？

const fs = require("fs");
var plotly = require('plotly')(username, api_key);

function generate_plot(){
    var trace = {
      type: 'bar',
      x: ['Mario', 'Luigi', 'Bowser'],
      y: [50000, 10000, 2000],
      marker: {color: ["red", "green", "darkblue"]},
    };

    var layout = {
      plot_bgcolor: 'rgb(52, 54, 60)',
      paper_bgcolor: 'rgb(52, 54, 60)',
    };

    var chart = {data: [trace], layout: layout};

    var pngOptions = {format: 'png', width: 1000, height: 500};

    plotly.getImage(chart, pngOptions, function (err, imageData) {
        if (err) throw err
        var pngStream = fs.createWriteStream('test.png');
        imageData.pipe(pngStream);
    })
}

function run(){
    generate_plot()
    // proceed with the generated plot which should be in the filesystem
}

Answer 1

欢迎来到堆栈溢出！

我已经删除了我以前的答案，因为（正如您正确指出的那样）他们没有解决全部问题。

在我们在generate_plot() function 之外返回控制之前需要完成两个异步任务：从 Plotly 检索数据，并将该内容写入磁盘。

下面的示例调用generate_plot() ，然后（作为验证文件是否真的在磁盘上的测试）它立即将test.png复制到test2.png 。 因为这会导致两个文件大小相同，这表明fs.copyFileSync直到test.png文件完全在磁盘上才运行。

我确实稍微更改了一些变量名，但这现在应该可以工作了。

为了等待文件 stream 完成，我参考了这个问题： event associated with fs.createWriteStream in node.js

function generate_plot() {
  return new Promise((resolve, reject) => {
    var trace = {
      type: 'bar',
      x: ['Mario', 'Luigi', 'Bowser'],
      y: [50000, 10000, 2000],
      marker: { color: ["red", "green", "darkblue"] },
    };

    var layout = {
      plot_bgcolor: 'rgb(52, 54, 60)',
      paper_bgcolor: 'rgb(52, 54, 60)',
    };

    var chart = { data: [trace], layout: layout };

    var pngOptions = { format: 'png', width: 1000, height: 500 };

    plotly.getImage(chart, pngOptions, (err, imageStream) => {
      if ( err ) return reject(err);
      var fileStream = fs.createWriteStream('test.png');
      imageStream.pipe(fileStream);
      fileStream.on('error', reject);
      fileStream.on('finish', resolve);
    });
  });
}

function run() {
  generate_plot()
    .then(() => {
      fs.copyFileSync('test.png', 'test2.png');
      console.log('done');
    })
    .catch(err => {
      console.log(err);
    });
}