使用node.js中的Promise从s3存储桶下载图片

Question

我想使用node.js中的Promises从我的本地系统上的s3存储桶下载一些图像文件。

var params = {
    Bucket: bucket_name',
    Key: 'key'
};

var fileStream = fs.createWriteStream('path/to/file.jpg');

我试过这个，它起作用了

s3.getObject(params).createReadStream.pipe(fileStream);

但是我希望我的代码看起来像这样

return s3.getObject(params).promise()
    .then(function(data) {
        //console.log(data.Body);
        // No idea about this section
    })
    .catch(function(err) {
        throw err;
    });

我必须使用Promise来确保所有图像都应该被下载。

Answer 1

一种可能的解决方案是使用bluebird并创建一个在流的末尾返回promise的函数：

const B = require('bluebird');

function downloadFromS3 (object) {
    var p = B.Promise.defer();

    var stream = s3.getObject(params).createReadStream()

    stream.pipe(fileStream);

    stream.on('error', (e) => p.reject(e))

    stream.on('end', () => p.resolve())

    return p.promise;
}

downloadFromS3(params)
    .then(() => console.log('finished'))
    .catch(() => console.log('failed'))

我不确定这段代码是否能正常工作，但它可能会给你提供一个方向。

Answer 2

streamToPromise = require('stream-to-promise');

var fileStream = fs.createWriteStream('path/to/file.jpg');
streamToPromise(fileStream).then(function () {
  console.log('Image saved to file.');
});
s3.getObject(params).createReadStream.pipe(fileStream);

Answer 3

这是一个本机promise解决方案，可以对读写流进行错误检测。

function streamPromise(stream) {
    return new Promise((resolve, reject) => {
        stream.on('end', () => {
            resolve('end');
        });
        stream.on('finish', () => {
            resolve('finish');
        });
        stream.on('error', (error) => {
            reject(error);
        });
    });
}

async function s3Download(srcBucket, srcKey, outputPath) {
    var objReq = s3.getObject({
        Bucket: srcBucket,
        Key: srcKey
    });
    let outStream = fs.createWriteStream(outputPath);
    let readStream = objReq.createReadStream();
    readStream.on('error', (err) => {
        console.warn('s3download error', err);
        outStream.emit("error", err);
    });
    readStream.pipe(outStream);
    return streamPromise(outStream);
}