管道 got.stream 到文件

Question

我正在重构一些在 Node 中使用 http 模块的代码，改为使用 got。我尝试了以下方法：

function get(url, filePath) {
  return new Promise((resolve, reject) => {
    got.stream(url).on
        ("response", response => {
            const newFile = fs.createWriteStream(filePath);
            response.pipe(newFile);
            newFile.on("finish", () => {
              newFile.close(resolve());
            });
            newFile.on("error", err => {
              reject(err);
            });    
        }).on
        ("error", err => {
             reject(err);
        });
  });
}

finish 事件从未触发。文件 (filePath) 使用 0 字节创建。

使用 newFile 的代码块在我使用 Node http 模块时有效。

将got.stream 传送到文件的正确方法是什么？

Answer 1

根据got() documentation，您希望将流直接通过管道传输到您的文件，如果您使用pipeline() 执行此操作，它将收集错误并报告完成。

const pipeline = promisify(stream.pipeline);
const fsp = require('fs').promises;

function get(url, filePath) { 
    return pipeline(
        got.stream(url),
        fs.createWriteStream(filePath)
    );
}

// usage
get(...).then(() => {
    console.log("all done");
}).catch(err => {
    console.log(err);
});

仅供参考，got.stream() 的重点是返回一个您可以直接用作流的流，并且由于您希望它转到一个文件，您可以将该流通过管道传输到该文件。我使用pipeline() 而不是.pipe()，因为管道具有比.pipe() 更完整的错误处理，尽管在非错误条件下，.pipe() 也可以工作。

---

如果出现错误，这是一个清理输出文件的版本：

function get(url, filePath) { 
    return pipeline(
        got.stream(url),
        fs.createWriteStream(filePath)
    ).catch(err => {
         fsp.unlink(filePath).catch(err => {
             if (err.code !== 'ENOENT') {
             // trying to delete output file upon error
                 console.log('error trying to delete output file', err);
             }
         });
         throw err;
    });
}