mirror of
https://github.com/Luzifer/share.git
synced 2024-12-30 07:21:19 +00:00
73 lines
2.4 KiB
Go
73 lines
2.4 KiB
Go
package s3
|
|
|
|
import (
|
|
"github.com/aws/aws-sdk-go/aws/client"
|
|
"github.com/aws/aws-sdk-go/aws/request"
|
|
"github.com/aws/aws-sdk-go/internal/s3err"
|
|
)
|
|
|
|
func init() {
|
|
initClient = defaultInitClientFn
|
|
initRequest = defaultInitRequestFn
|
|
}
|
|
|
|
func defaultInitClientFn(c *client.Client) {
|
|
// Support building custom endpoints based on config
|
|
c.Handlers.Build.PushFront(updateEndpointForS3Config)
|
|
|
|
// Require SSL when using SSE keys
|
|
c.Handlers.Validate.PushBack(validateSSERequiresSSL)
|
|
c.Handlers.Build.PushBack(computeSSEKeys)
|
|
|
|
// S3 uses custom error unmarshaling logic
|
|
c.Handlers.UnmarshalError.Clear()
|
|
c.Handlers.UnmarshalError.PushBack(unmarshalError)
|
|
c.Handlers.UnmarshalError.PushBackNamed(s3err.RequestFailureWrapperHandler())
|
|
}
|
|
|
|
func defaultInitRequestFn(r *request.Request) {
|
|
// Add reuest handlers for specific platforms.
|
|
// e.g. 100-continue support for PUT requests using Go 1.6
|
|
platformRequestHandlers(r)
|
|
|
|
switch r.Operation.Name {
|
|
case opPutBucketCors, opPutBucketLifecycle, opPutBucketPolicy,
|
|
opPutBucketTagging, opDeleteObjects, opPutBucketLifecycleConfiguration,
|
|
opPutBucketReplication:
|
|
// These S3 operations require Content-MD5 to be set
|
|
r.Handlers.Build.PushBack(contentMD5)
|
|
case opGetBucketLocation:
|
|
// GetBucketLocation has custom parsing logic
|
|
r.Handlers.Unmarshal.PushFront(buildGetBucketLocation)
|
|
case opCreateBucket:
|
|
// Auto-populate LocationConstraint with current region
|
|
r.Handlers.Validate.PushFront(populateLocationConstraint)
|
|
case opCopyObject, opUploadPartCopy, opCompleteMultipartUpload:
|
|
r.Handlers.Unmarshal.PushFront(copyMultipartStatusOKUnmarhsalError)
|
|
r.Handlers.Unmarshal.PushBackNamed(s3err.RequestFailureWrapperHandler())
|
|
case opPutObject, opUploadPart:
|
|
r.Handlers.Build.PushBack(computeBodyHashes)
|
|
// Disabled until #1837 root issue is resolved.
|
|
// case opGetObject:
|
|
// r.Handlers.Build.PushBack(askForTxEncodingAppendMD5)
|
|
// r.Handlers.Unmarshal.PushBack(useMD5ValidationReader)
|
|
}
|
|
}
|
|
|
|
// bucketGetter is an accessor interface to grab the "Bucket" field from
|
|
// an S3 type.
|
|
type bucketGetter interface {
|
|
getBucket() string
|
|
}
|
|
|
|
// sseCustomerKeyGetter is an accessor interface to grab the "SSECustomerKey"
|
|
// field from an S3 type.
|
|
type sseCustomerKeyGetter interface {
|
|
getSSECustomerKey() string
|
|
}
|
|
|
|
// copySourceSSECustomerKeyGetter is an accessor interface to grab the
|
|
// "CopySourceSSECustomerKey" field from an S3 type.
|
|
type copySourceSSECustomerKeyGetter interface {
|
|
getCopySourceSSECustomerKey() string
|
|
}
|