1// Code generated by smithy-go-codegen DO NOT EDIT.
2
3package sagemaker
4
5import (
6	"context"
7	awsmiddleware "github.com/aws/aws-sdk-go-v2/aws/middleware"
8	"github.com/aws/aws-sdk-go-v2/aws/signer/v4"
9	"github.com/aws/aws-sdk-go-v2/service/sagemaker/types"
10	"github.com/aws/smithy-go/middleware"
11	smithyhttp "github.com/aws/smithy-go/transport/http"
12)
13
14// Updates variant weight of one or more variants associated with an existing
15// endpoint, or capacity of one variant associated with an existing endpoint. When
16// it receives the request, Amazon SageMaker sets the endpoint status to Updating.
17// After updating the endpoint, it sets the status to InService. To check the
18// status of an endpoint, use the DescribeEndpoint API.
19func (c *Client) UpdateEndpointWeightsAndCapacities(ctx context.Context, params *UpdateEndpointWeightsAndCapacitiesInput, optFns ...func(*Options)) (*UpdateEndpointWeightsAndCapacitiesOutput, error) {
20	if params == nil {
21		params = &UpdateEndpointWeightsAndCapacitiesInput{}
22	}
23
24	result, metadata, err := c.invokeOperation(ctx, "UpdateEndpointWeightsAndCapacities", params, optFns, addOperationUpdateEndpointWeightsAndCapacitiesMiddlewares)
25	if err != nil {
26		return nil, err
27	}
28
29	out := result.(*UpdateEndpointWeightsAndCapacitiesOutput)
30	out.ResultMetadata = metadata
31	return out, nil
32}
33
34type UpdateEndpointWeightsAndCapacitiesInput struct {
35
36	// An object that provides new capacity and weight values for a variant.
37	//
38	// This member is required.
39	DesiredWeightsAndCapacities []types.DesiredWeightAndCapacity
40
41	// The name of an existing Amazon SageMaker endpoint.
42	//
43	// This member is required.
44	EndpointName *string
45}
46
47type UpdateEndpointWeightsAndCapacitiesOutput struct {
48
49	// The Amazon Resource Name (ARN) of the updated endpoint.
50	//
51	// This member is required.
52	EndpointArn *string
53
54	// Metadata pertaining to the operation's result.
55	ResultMetadata middleware.Metadata
56}
57
58func addOperationUpdateEndpointWeightsAndCapacitiesMiddlewares(stack *middleware.Stack, options Options) (err error) {
59	err = stack.Serialize.Add(&awsAwsjson11_serializeOpUpdateEndpointWeightsAndCapacities{}, middleware.After)
60	if err != nil {
61		return err
62	}
63	err = stack.Deserialize.Add(&awsAwsjson11_deserializeOpUpdateEndpointWeightsAndCapacities{}, middleware.After)
64	if err != nil {
65		return err
66	}
67	if err = addSetLoggerMiddleware(stack, options); err != nil {
68		return err
69	}
70	if err = awsmiddleware.AddClientRequestIDMiddleware(stack); err != nil {
71		return err
72	}
73	if err = smithyhttp.AddComputeContentLengthMiddleware(stack); err != nil {
74		return err
75	}
76	if err = addResolveEndpointMiddleware(stack, options); err != nil {
77		return err
78	}
79	if err = v4.AddComputePayloadSHA256Middleware(stack); err != nil {
80		return err
81	}
82	if err = addRetryMiddlewares(stack, options); err != nil {
83		return err
84	}
85	if err = addHTTPSignerV4Middleware(stack, options); err != nil {
86		return err
87	}
88	if err = awsmiddleware.AddRawResponseToMetadata(stack); err != nil {
89		return err
90	}
91	if err = awsmiddleware.AddRecordResponseTiming(stack); err != nil {
92		return err
93	}
94	if err = addClientUserAgent(stack); err != nil {
95		return err
96	}
97	if err = smithyhttp.AddErrorCloseResponseBodyMiddleware(stack); err != nil {
98		return err
99	}
100	if err = smithyhttp.AddCloseResponseBodyMiddleware(stack); err != nil {
101		return err
102	}
103	if err = addOpUpdateEndpointWeightsAndCapacitiesValidationMiddleware(stack); err != nil {
104		return err
105	}
106	if err = stack.Initialize.Add(newServiceMetadataMiddleware_opUpdateEndpointWeightsAndCapacities(options.Region), middleware.Before); err != nil {
107		return err
108	}
109	if err = addRequestIDRetrieverMiddleware(stack); err != nil {
110		return err
111	}
112	if err = addResponseErrorMiddleware(stack); err != nil {
113		return err
114	}
115	if err = addRequestResponseLogging(stack, options); err != nil {
116		return err
117	}
118	return nil
119}
120
121func newServiceMetadataMiddleware_opUpdateEndpointWeightsAndCapacities(region string) *awsmiddleware.RegisterServiceMetadata {
122	return &awsmiddleware.RegisterServiceMetadata{
123		Region:        region,
124		ServiceID:     ServiceID,
125		SigningName:   "sagemaker",
126		OperationName: "UpdateEndpointWeightsAndCapacities",
127	}
128}
129