Потоковый файл из S3 с помощью Express, содержащий информацию о длине и типе файла

Используя модуль aws-sdk и Express 4.13, можно проксировать файл с S3 несколькими способами.

Эта версия обратного вызова вернет тело файла в виде буфера плюс другие соответствующие заголовки, такие как Content-Length:

function(req,res){

  var s3 = new AWS.S3();

  s3.getObject({Bucket: myBucket, Key: myFile},function(err,data){

    if (err) {
      return res.status(500).send("Error!");
    }

    // Headers
    res.set("Content-Length",data.ContentLength)
       .set("Content-Type",data.ContentType);

    res.send(data.Body); // data.Body is a buffer

  });

}

Проблема с этой версией заключается в том, что вы должны получить весь файл перед его отправкой, что не очень удобно, особенно если оно что-то большое, как видео.

Эта версия будет напрямую передавать файл:

function(req,res){

  var s3 = new AWS.S3();

  s3.getObject({Bucket: myBucket, Key: myFile})
    .createReadStream()
    .pipe(res);

}

Но в отличие от первого, он ничего не сделает с заголовками, которые браузеру может понадобиться для правильной обработки файла.

Есть ли способ получить лучшее из обоих миров, проходя через правильные заголовки с S3, но отправляя файл как поток? Это можно сделать, сначала сделав запрос HEAD на S3, чтобы получить метаданные, но может ли это быть сделано с помощью одного вызова API?

Ответы

Ответ 1

Один подход - это прослушивание события httpHeaders и создание потока внутри него.

s3.getObject(params)
    .on('httpHeaders', function (statusCode, headers) {
        res.set('Content-Length', headers['content-length']);
        res.set('Content-Type', headers['content-type']);
        this.response.httpResponse.createUnbufferedStream()
            .pipe(res);
    })
    .send();

Ответ 2

Для моего проекта я просто делаю headObject, чтобы получить только метаданные объекта (это очень быстро и не загружать объект). Затем добавляю в ответ все заголовки, которые мне нужны для распространения для трубопровода:

    var s3 = new AWS.S3();

    var params = {
        Bucket: bucket,
        Key: key
    };
    s3.headObject(params, function (err, data) {
        if (err) {
            // an error occurred
            console.error(err);
            return next();
        }
        var stream = s3.getObject(params).createReadStream();

        // forward errors
        stream.on('error', function error(err) {
            //continue to the next middlewares
            return next();
        });

        //Add the content type to the response (it not propagated from the S3 SDK)
        res.set('Content-Type', mime.lookup(key));
        res.set('Content-Length', data.ContentLength);
        res.set('Last-Modified', data.LastModified);
        res.set('ETag', data.ETag);

        stream.on('end', () => {
            console.log('Served by Amazon S3: ' + key);
        });
        //Pipe the s3 object to the response
        stream.pipe(res);
    });

Ответ 3

Основываясь на ответе Андре Верланга, мы сделали следующее для увеличения объектов AWS Request с помощью метода forwardToExpress:

const _ = require('lodash');
const AWS = require('aws-sdk');

AWS.Request.prototype.forwardToExpress = function forwardToExpress(res, next) {
    this
    .on('httpHeaders', function (code, headers) {
        if (code < 300) {
            res.set(_.pick(headers, 'content-type', 'content-length', 'last-modified'));
        }                            
    })
    .createReadStream()
    .on('error', next)
    .pipe(res);
};

Затем в наших обработчиках маршрута мы можем сделать что-то вроде этого:

s3.getObject({Bucket: myBucket, Key: myFile}).forwardToExpress(res, next);