File: disable_host_prefix.go

package info (click to toggle)
golang-github-aws-aws-sdk-go-v2 1.24.1-2~bpo12%2B1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm-backports
  • size: 554,032 kB
  • sloc: java: 15,941; makefile: 419; sh: 175
file content (36 lines) | stat: -rw-r--r-- 1,225 bytes parent folder | download | duplicates (4)
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)
}