1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
|
package customizations
import (
"context"
awsmiddleware "github.com/aws/aws-sdk-go-v2/aws/middleware"
"github.com/aws/smithy-go/middleware"
smithyhttp "github.com/aws/smithy-go/transport/http"
)
// DisableHostPrefixMiddleware disables host prefix serialization when v2
// endpoint resolution was used.
type DisableHostPrefixMiddleware struct{}
// ID identifies the middleware.
func (*DisableHostPrefixMiddleware) ID() string {
return "S3ControlDisableHostPrefix"
}
// HandleFinalize controls whether to serialize modeled host prefixes based on
// the type of endpoint resolution used.
func (m *DisableHostPrefixMiddleware) HandleFinalize(ctx context.Context, in middleware.FinalizeInput, next middleware.FinalizeHandler) (
out middleware.FinalizeOutput, md middleware.Metadata, err error,
) {
if awsmiddleware.GetRequiresLegacyEndpoints(ctx) {
return next.HandleFinalize(ctx, in)
}
ctx = smithyhttp.DisableEndpointHostPrefix(ctx, true)
return next.HandleFinalize(ctx, in)
}
// AddDisableHostPrefixMiddleware adds the middleware to the stack.
func AddDisableHostPrefixMiddleware(s *middleware.Stack) error {
return s.Finalize.Insert(&DisableHostPrefixMiddleware{}, "ResolveEndpointV2", middleware.After)
}
|