40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
package context
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/aws/smithy-go/middleware"
|
|
)
|
|
|
|
type s3BackendKey struct{}
|
|
type checksumInputAlgorithmKey struct{}
|
|
|
|
const (
|
|
// S3BackendS3Express identifies the S3Express backend
|
|
S3BackendS3Express = "S3Express"
|
|
)
|
|
|
|
// SetS3Backend stores the resolved endpoint backend within the request
|
|
// context, which is required for a variety of custom S3 behaviors.
|
|
func SetS3Backend(ctx context.Context, typ string) context.Context {
|
|
return middleware.WithStackValue(ctx, s3BackendKey{}, typ)
|
|
}
|
|
|
|
// GetS3Backend retrieves the stored endpoint backend within the context.
|
|
func GetS3Backend(ctx context.Context) string {
|
|
v, _ := middleware.GetStackValue(ctx, s3BackendKey{}).(string)
|
|
return v
|
|
}
|
|
|
|
// SetChecksumInputAlgorithm sets the request checksum algorithm on the
|
|
// context.
|
|
func SetChecksumInputAlgorithm(ctx context.Context, value string) context.Context {
|
|
return middleware.WithStackValue(ctx, checksumInputAlgorithmKey{}, value)
|
|
}
|
|
|
|
// GetChecksumInputAlgorithm returns the checksum algorithm from the context.
|
|
func GetChecksumInputAlgorithm(ctx context.Context) string {
|
|
v, _ := middleware.GetStackValue(ctx, checksumInputAlgorithmKey{}).(string)
|
|
return v
|
|
}
|