programing tip

스트림을 s3.upload ()로 파이프

itbloger 2020. 12. 10. 19:07
반응형

스트림을 s3.upload ()로 파이프


저는 현재 s3-upload-stream 이라는 node.js 플러그인을 사용 하여 매우 큰 파일을 Amazon S3 로 스트리밍 하고 있습니다. 멀티 파트 API를 사용하며 대부분 잘 작동합니다.

그러나이 모듈은 그 나이를 보여주고 있으며 이미 수정을해야했습니다 (저자도이 모듈을 더 이상 사용하지 않습니다). 오늘 저는 Amazon에서 또 다른 문제를 만났고 작성자의 추천을 받아 공식 aws-sdk를 사용하여 업로드를 완료하고 싶습니다.

그러나.

공식 SDK는 s3.upload(). s3.upload의 특성은 읽을 수있는 스트림을 S3 생성자에 인수로 전달해야한다는 것입니다.

다양한 파일 처리를 수행하는 약 120 개 이상의 사용자 코드 모듈이 있으며 출력의 최종 대상에 대해 알 수 없습니다. 엔진은 파이프 가능한 쓰기 가능한 출력 스트림을 전달하고 여기에 파이프합니다. 나는 그들에게 AWS.S3객체를 건네 upload()주고 모든 모듈에 코드를 추가하지 않고는 그것을 호출하도록 요청할 수 없다 . 내가 사용한 이유 s3-upload-stream는 배관을 지원했기 때문입니다.

s3.upload()스트림을 파이프 할 수 있는 대상으로 aws-sdk를 만드는 방법 이 있습니까?


S3 upload()함수를 node.js stream.PassThrough()스트림으로 래핑합니다 .

예를 들면 다음과 같습니다.

inputStream
  .pipe(uploadFromStream(s3));

function uploadFromStream(s3) {
  var pass = new stream.PassThrough();

  var params = {Bucket: BUCKET, Key: KEY, Body: pass};
  s3.upload(params, function(err, data) {
    console.log(err, data);
  });

  return pass;
}

조금 늦게 대답하면 다른 사람에게 도움이 될 수 있습니다. 쓰기 가능한 스트림과 promise를 모두 반환 할 수 있으므로 업로드가 완료되면 응답 데이터를 얻을 수 있습니다.

const AWS = require('aws-sdk');
const stream = require('stream');

const uploadStream = ({ Bucket, Key }) => {
  const s3 = new AWS.S3();
  const pass = new stream.PassThrough();
  return {
    writeStream: pass,
    promise: s3.upload({ Bucket, Key, Body: pass }).promise(),
  };
}

그리고 다음과 같이 기능을 사용할 수 있습니다.

const { writeStream, promise } = uploadStream({Bucket: 'yourbucket', Key: 'yourfile.mp4'});
const readStream = fs.createReadStream('/path/to/yourfile.mp4');

readStream.pipe(writeStream);
promise.then(console.log);

수락 된 답변에서는 업로드가 완료되기 전에 함수가 종료되므로 잘못된 것입니다. 아래 코드는 읽을 수있는 스트림에서 올바르게 파이프됩니다.

참조 업로드

async function uploadReadableStream(stream) {
  const params = {Bucket: bucket, Key: key, Body: stream};
  return s3.upload(params).promise();
}

async function upload() {
  const readable = getSomeReadableStream();
  const results = await uploadReadableStream(readable);
  console.log('upload complete', results);
}

다음과 같이 한 단계 더 나아가 진행 정보를 출력 할 수도 있습니다 ManagedUpload.

const manager = s3.upload(params);
manager.on('httpUploadProgress', (progress) => {
  console.log('progress', progress) // { loaded: 4915, total: 192915, part: 1, key: 'foo.jpg' }
});

ManagedUpload 참조

사용 가능한 이벤트 목록


유형 스크립트 솔루션 :
이 예제에서는 다음을 사용합니다.

import * as AWS from "aws-sdk";
import * as fsExtra from "fs-extra";
import * as zlib from "zlib";
import * as stream from "stream";

그리고 비동기 기능 :

public async saveFile(filePath: string, s3Bucket: AWS.S3, key: string, bucketName: string): Promise<boolean> { 

         const uploadStream = (S3: AWS.S3, Bucket: string, Key: string) => {
            const passT = new stream.PassThrough();
            return {
              writeStream: passT,
              promise: S3.upload({ Bucket, Key, Body: passT }).promise(),
            };
          };
        const { writeStream, promise } = uploadStream(s3Bucket, bucketName, key);
        fsExtra.createReadStream(filePath).pipe(writeStream);     //  NOTE: Addition You can compress to zip by  .pipe(zlib.createGzip()).pipe(writeStream)
        let output = true;
        await promise.catch((reason)=> { output = false; console.log(reason);});
        return output;
}

이 메서드를 다음과 같이 호출하십시오.

let result = await saveFileToS3(testFilePath, someS3Bucket, someKey, someBucketName);

s3 api 업로드 기능과 0 바이트 파일을 사용할 때 s3 (@ Radar155 및 @gabo)에서 종료된다는 불평을하는 사람들을 위해이 문제도 발생했습니다.

두 번째 PassThrough 스트림을 만들고 첫 번째에서 두 번째로 모든 데이터를 파이프하고 두 번째에 대한 참조를 s3에 전달합니다. 몇 가지 다른 방법으로이 작업을 수행 할 수 있습니다. 아마도 더러운 방법은 첫 번째 스트림에서 "data"이벤트를 수신 한 다음 두 번째 스트림에 동일한 데이터를 쓰는 것입니다. "end"이벤트와 유사하게 두 번째 스트림의 종료 기능. 이것이 aws api의 버그인지, 노드 버전인지 또는 다른 문제인지는 모르겠지만 문제를 해결했습니다.

다음과 같이 보일 수 있습니다.

var PassThroughStream = require('stream').PassThrough;
var srcStream = new PassThroughStream();

var rstream = fs.createReadStream('Learning/stocktest.json');
var sameStream = rstream.pipe(srcStream);
// interesting note: (srcStream == sameStream) at this point
var destStream = new PassThroughStream();
// call your s3.upload function here - passing in the destStream as the Body parameter
srcStream.on('data', function (chunk) {
    destStream.write(chunk);
});

srcStream.on('end', function () {
    dataStream.end();
});

그것이 내가 클라이언트에서 s3로 성공적으로 스트리밍 할 수 있었던 사람에게 도움이된다면 :

https://gist.github.com/mattlockyer/532291b6194f6d9ca40cb82564db9d2a

서버 측 코드는 req스트림 객체 라고 가정합니다 . 제 경우에는 헤더에 파일 정보가 설정된 클라이언트에서 전송되었습니다.

const fileUploadStream = (req, res) => {
  //get "body" args from header
  const { id, fn } = JSON.parse(req.get('body'));
  const Key = id + '/' + fn; //upload to s3 folder "id" with filename === fn
  const params = {
    Key,
    Bucket: bucketName, //set somewhere
    Body: req, //req is a stream
  };
  s3.upload(params, (err, data) => {
    if (err) {
      res.send('Error Uploading Data: ' + JSON.stringify(err) + '\n' + JSON.stringify(err.stack));
    } else {
      res.send(Key);
    }
  });
};

예, 관습을 어기지만 요점을 보면 multer, busboy 등을 사용하여 찾은 다른 것보다 훨씬 깨끗합니다.

실용주의에 +1하고 도움을 주신 @SalehenRahman에게 감사드립니다.


KnexJS를 사용하고 있으며 스트리밍 API를 사용하는 데 문제가 있습니다. 나는 마침내 그것을 고쳤고, 다음이 누군가를 도울 것입니다.

const knexStream = knex.select('*').from('my_table').stream();
const passThroughStream = new stream.PassThrough();

knexStream.on('data', (chunk) => passThroughStream.write(JSON.stringify(chunk) + '\n'));
knexStream.on('end', () => passThroughStream.end());

const uploadResult = await s3
  .upload({
    Bucket: 'my-bucket',
    Key: 'stream-test.txt',
    Body: passThroughStream
  })
  .promise();

내가 원했기 때문에 어떤 답변도 나를 위해 일하지 않았습니다.

  • 파이프 s3.upload()
  • 결과 s3.upload()를 다른 스트림으로 파이프

받아 들여진 대답은 후자를하지 않습니다. 다른 것들은 promise api에 의존하는데, 이는 하천 파이프로 작업 할 때 번거 롭습니다.

이것은 받아 들여진 대답의 수정입니다.

const s3 = new S3();

function writeToS3({Key, Bucket}) {
  const Body = new stream.PassThrough();

  s3.upload({
    Body,
    Key,
    Bucket: process.env.adpBucket
  })
   .on('httpUploadProgress', progress => {
       console.log('progress', progress);
   })
   .send((err, data) => {
     if (err) {
       Body.destroy(err);
     } else {
       console.log(`File uploaded and available at ${data.Location}`);
       Body.destroy();
     }
  });

  return Body;
}

const pipeline = myReadableStream.pipe(writeToS3({Key, Bucket});

pipeline.on('close', () => {
  // upload finished, do something else
})
pipeline.on('error', () => {
  // upload wasn't successful. Handle it
})


The thing here to note in the most accepted answer above is that: You need to return the pass in the function if you are using pipe like,

fs.createReadStream(<filePath>).pipe(anyUploadFunction())

function anyUploadFunction () { 
 let pass = new stream.PassThrough();
 return pass // <- Returning this pass is important for the stream to understand where it needs to write to.
}

Otherwise it will silently move onto next without throwing an error or will throw an error of TypeError: dest.on is not a function depending upon how you have written the function


If you know the size of the stream you can use minio-js to upload the stream like this:

  s3Client.putObject('my-bucketname', 'my-objectname.ogg', stream, size, 'audio/ogg', function(e) {
    if (e) {
      return console.log(e)
    }
    console.log("Successfully uploaded the stream")
  })

참고URL : https://stackoverflow.com/questions/37336050/pipe-a-stream-to-s3-upload

반응형