mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-16 17:21:04 +01:00
76 lines
2.6 KiB
Go
76 lines
2.6 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(computeSSEKeyMD5)
|
||
|
c.Handlers.Build.PushBack(computeCopySourceSSEKeyMD5)
|
||
|
|
||
|
// 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,
|
||
|
opPutObjectLegalHold, opPutObjectRetention, opPutObjectLockConfiguration,
|
||
|
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
|
||
|
}
|