mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-21 20:47:08 -05:00
Merge pull request #39 from sharelatex/bg-use-aws-sdk-to-copy-files
use the aws sdk to copy files in S3PersistorManager
This commit is contained in:
commit
04176d479b
2 changed files with 18 additions and 9 deletions
|
@ -11,6 +11,7 @@ path = require("path")
|
|||
LocalFileWriter = require("./LocalFileWriter")
|
||||
Errors = require("./Errors")
|
||||
_ = require("underscore")
|
||||
awsS3 = require "aws-sdk/clients/s3"
|
||||
|
||||
thirtySeconds = 30 * 1000
|
||||
|
||||
|
@ -25,6 +26,12 @@ buildDefaultOptions = (bucketName, method, key)->
|
|||
uri:"https://#{bucketName}.s3.amazonaws.com/#{key}"
|
||||
}
|
||||
|
||||
s3 = new awsS3({
|
||||
credentials:
|
||||
accessKeyId: settings.filestore.s3.key,
|
||||
secretAccessKey: settings.filestore.s3.secret
|
||||
})
|
||||
|
||||
module.exports =
|
||||
|
||||
sendFile: (bucketName, key, fsPath, callback)->
|
||||
|
@ -91,11 +98,9 @@ module.exports =
|
|||
|
||||
copyFile: (bucketName, sourceKey, destKey, callback)->
|
||||
logger.log bucketName:bucketName, sourceKey:sourceKey, destKey: destKey, "copying file in s3"
|
||||
s3Client = knox.createClient
|
||||
key: settings.filestore.s3.key
|
||||
secret: settings.filestore.s3.secret
|
||||
bucket: bucketName
|
||||
s3Client.copyFile sourceKey, destKey, (err)->
|
||||
source = bucketName + '/' + sourceKey
|
||||
# use the AWS SDK instead of knox due to problems with error handling (https://github.com/Automattic/knox/issues/114)
|
||||
s3.copyObject {Bucket: bucketName, Key: destKey, CopySource: source}, (err) ->
|
||||
if err?
|
||||
logger.err err:err, bucketName:bucketName, sourceKey:sourceKey, destKey:destKey, "something went wrong copying file in aws"
|
||||
callback(err)
|
||||
|
|
|
@ -25,11 +25,15 @@ describe "S3PersistorManagerTests", ->
|
|||
get: sinon.stub()
|
||||
@knox =
|
||||
createClient: sinon.stub().returns(@stubbedKnoxClient)
|
||||
@stubbedS3Client =
|
||||
copyObject:sinon.stub()
|
||||
@awsS3 = sinon.stub().returns @stubbedS3Client
|
||||
@LocalFileWriter =
|
||||
writeStream: sinon.stub()
|
||||
deleteFile: sinon.stub()
|
||||
@requires =
|
||||
"knox": @knox
|
||||
"aws-sdk/clients/s3": @awsS3
|
||||
"settings-sharelatex": @settings
|
||||
"./LocalFileWriter":@LocalFileWriter
|
||||
"logger-sharelatex":
|
||||
|
@ -207,11 +211,11 @@ describe "S3PersistorManagerTests", ->
|
|||
@destKey = "my/dest/key"
|
||||
@S3PersistorManager = SandboxedModule.require modulePath, requires: @requires
|
||||
|
||||
it "should use knox to copy file", (done)->
|
||||
@stubbedKnoxClient.copyFile.callsArgWith(2, @error)
|
||||
it "should use AWS SDK to copy file", (done)->
|
||||
@stubbedS3Client.copyObject.callsArgWith(1, @error)
|
||||
@S3PersistorManager.copyFile @bucketName, @sourceKey, @destKey, (err)=>
|
||||
err.should.equal @error
|
||||
@stubbedKnoxClient.copyFile.calledWith(@sourceKey, @destKey).should.equal true
|
||||
@stubbedS3Client.copyObject.calledWith({Bucket: @bucketName, Key: @destKey, CopySource: @bucketName + '/' + @key}).should.equal true
|
||||
done()
|
||||
|
||||
describe "deleteDirectory", ->
|
||||
|
|
Loading…
Reference in a new issue