diff options
author | Joey Hess <joey@kitenet.net> | 2014-11-03 15:53:22 -0400 |
---|---|---|
committer | Joey Hess <joey@kitenet.net> | 2014-11-03 16:04:55 -0400 |
commit | b69362e972190668a0742059f0798d2ce00f15c6 (patch) | |
tree | 9d3a8c88fdd885141f423be00978532e5faa9ddd /Remote/S3.hs | |
parent | 9d7f923e5b3466a8fe3b34781483cb6a115fe5fc (diff) |
finish multipart support using unreleased update to aws lib to yield etags
Untested and not even compiled yet.
Testing should include checks that file content streams through without
buffering in memory.
Note that CL.consume causes all the etags to be buffered in memory.
This is probably nearly unavoidable, since a request has to be constructed
that contains the list of etags in its body. (While it might be possible to
stream generation of the body, that would entail making a http request that
dribbles out parts of the body as the multipart uploads complete, which is
not likely to work well..
To limit this being a problem, it's best for partsize to be set to some
suitably large value, like 1gb. Then a full terabyte file will need only
1024 etags to be stored, which will probably use around 1 mb of memory.
Diffstat (limited to 'Remote/S3.hs')
-rw-r--r-- | Remote/S3.hs | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/Remote/S3.hs b/Remote/S3.hs index 9a618329a..9c90d4b2c 100644 --- a/Remote/S3.hs +++ b/Remote/S3.hs @@ -13,6 +13,10 @@ module Remote.S3 (remote, iaHost, configIA, iaItemUrl) where import qualified Aws as AWS import qualified Aws.Core as AWS import qualified Aws.S3 as S3 +#if MIN_VERSION_aws(0,10,4) +import qualified Aws.S3.Commands.Multipart as Multipart +import qualified Data.Conduit.List as CL +#endif import qualified Data.Text as T import qualified Data.Text.Encoding as T import qualified Data.ByteString.Lazy as L @@ -170,7 +174,7 @@ store r h = fileStorer $ \k f p -> do multipartupload sz k f p = do #if MIN_VERSION_aws(0,10,4) let info = hinfo h - let objects = bucketObject info h + let object = bucketObject info h uploadid <- S3.imurUploadId <$> sendS3Handle' h $ (S3.postInitiateMultipartUpload (bucket info) object) @@ -180,14 +184,13 @@ store r h = fileStorer $ \k f p -> do , S3.imuExpires = Nothing -- TODO set some reasonable expiry } - -- TODO open file, read each part of size sz (streaming - -- it); send part to S3, and get a list of etags of all - -- the parts - + etags <- sourceFile f + $= Multipart.chunkedConduit sz + $= Multipart.putConduit (hawscfg h) (hs3cfg h) (hmanager h) (bucket info) object uploadid + $$ CL.consume - void $ sendS3Handle' h $ - S3.postCompleteMultipartUpload (bucket info) object uploadid $ - zip [1..] (map T.pack etags) + void $ sendS3Handle' h $ S3.postCompleteMultipartUpload + (bucket info) object uploadid (zip [1..] etags) #else warning $ "Cannot do multipart upload (partsize " ++ show sz ++ "); built with too old a version of the aws library." singlepartupload k f p |