1// Code generated by smithy-go-codegen DO NOT EDIT.
2
3package xray
4
5import (
6	"context"
7	"fmt"
8	awsmiddleware "github.com/aws/aws-sdk-go-v2/aws/middleware"
9	"github.com/aws/aws-sdk-go-v2/aws/signer/v4"
10	"github.com/aws/aws-sdk-go-v2/service/xray/types"
11	"github.com/aws/smithy-go/middleware"
12	smithyhttp "github.com/aws/smithy-go/transport/http"
13)
14
15// Retrieves all sampling rules.
16func (c *Client) GetSamplingRules(ctx context.Context, params *GetSamplingRulesInput, optFns ...func(*Options)) (*GetSamplingRulesOutput, error) {
17	if params == nil {
18		params = &GetSamplingRulesInput{}
19	}
20
21	result, metadata, err := c.invokeOperation(ctx, "GetSamplingRules", params, optFns, addOperationGetSamplingRulesMiddlewares)
22	if err != nil {
23		return nil, err
24	}
25
26	out := result.(*GetSamplingRulesOutput)
27	out.ResultMetadata = metadata
28	return out, nil
29}
30
31type GetSamplingRulesInput struct {
32
33	// Pagination token.
34	NextToken *string
35}
36
37type GetSamplingRulesOutput struct {
38
39	// Pagination token.
40	NextToken *string
41
42	// Rule definitions and metadata.
43	SamplingRuleRecords []types.SamplingRuleRecord
44
45	// Metadata pertaining to the operation's result.
46	ResultMetadata middleware.Metadata
47}
48
49func addOperationGetSamplingRulesMiddlewares(stack *middleware.Stack, options Options) (err error) {
50	err = stack.Serialize.Add(&awsRestjson1_serializeOpGetSamplingRules{}, middleware.After)
51	if err != nil {
52		return err
53	}
54	err = stack.Deserialize.Add(&awsRestjson1_deserializeOpGetSamplingRules{}, middleware.After)
55	if err != nil {
56		return err
57	}
58	if err = addSetLoggerMiddleware(stack, options); err != nil {
59		return err
60	}
61	if err = awsmiddleware.AddClientRequestIDMiddleware(stack); err != nil {
62		return err
63	}
64	if err = smithyhttp.AddComputeContentLengthMiddleware(stack); err != nil {
65		return err
66	}
67	if err = addResolveEndpointMiddleware(stack, options); err != nil {
68		return err
69	}
70	if err = v4.AddComputePayloadSHA256Middleware(stack); err != nil {
71		return err
72	}
73	if err = addRetryMiddlewares(stack, options); err != nil {
74		return err
75	}
76	if err = addHTTPSignerV4Middleware(stack, options); err != nil {
77		return err
78	}
79	if err = awsmiddleware.AddRawResponseToMetadata(stack); err != nil {
80		return err
81	}
82	if err = awsmiddleware.AddRecordResponseTiming(stack); err != nil {
83		return err
84	}
85	if err = addClientUserAgent(stack); err != nil {
86		return err
87	}
88	if err = smithyhttp.AddErrorCloseResponseBodyMiddleware(stack); err != nil {
89		return err
90	}
91	if err = smithyhttp.AddCloseResponseBodyMiddleware(stack); err != nil {
92		return err
93	}
94	if err = stack.Initialize.Add(newServiceMetadataMiddleware_opGetSamplingRules(options.Region), middleware.Before); err != nil {
95		return err
96	}
97	if err = addRequestIDRetrieverMiddleware(stack); err != nil {
98		return err
99	}
100	if err = addResponseErrorMiddleware(stack); err != nil {
101		return err
102	}
103	if err = addRequestResponseLogging(stack, options); err != nil {
104		return err
105	}
106	return nil
107}
108
109// GetSamplingRulesAPIClient is a client that implements the GetSamplingRules
110// operation.
111type GetSamplingRulesAPIClient interface {
112	GetSamplingRules(context.Context, *GetSamplingRulesInput, ...func(*Options)) (*GetSamplingRulesOutput, error)
113}
114
115var _ GetSamplingRulesAPIClient = (*Client)(nil)
116
117// GetSamplingRulesPaginatorOptions is the paginator options for GetSamplingRules
118type GetSamplingRulesPaginatorOptions struct {
119	// Set to true if pagination should stop if the service returns a pagination token
120	// that matches the most recent token provided to the service.
121	StopOnDuplicateToken bool
122}
123
124// GetSamplingRulesPaginator is a paginator for GetSamplingRules
125type GetSamplingRulesPaginator struct {
126	options   GetSamplingRulesPaginatorOptions
127	client    GetSamplingRulesAPIClient
128	params    *GetSamplingRulesInput
129	nextToken *string
130	firstPage bool
131}
132
133// NewGetSamplingRulesPaginator returns a new GetSamplingRulesPaginator
134func NewGetSamplingRulesPaginator(client GetSamplingRulesAPIClient, params *GetSamplingRulesInput, optFns ...func(*GetSamplingRulesPaginatorOptions)) *GetSamplingRulesPaginator {
135	if params == nil {
136		params = &GetSamplingRulesInput{}
137	}
138
139	options := GetSamplingRulesPaginatorOptions{}
140
141	for _, fn := range optFns {
142		fn(&options)
143	}
144
145	return &GetSamplingRulesPaginator{
146		options:   options,
147		client:    client,
148		params:    params,
149		firstPage: true,
150	}
151}
152
153// HasMorePages returns a boolean indicating whether more pages are available
154func (p *GetSamplingRulesPaginator) HasMorePages() bool {
155	return p.firstPage || p.nextToken != nil
156}
157
158// NextPage retrieves the next GetSamplingRules page.
159func (p *GetSamplingRulesPaginator) NextPage(ctx context.Context, optFns ...func(*Options)) (*GetSamplingRulesOutput, error) {
160	if !p.HasMorePages() {
161		return nil, fmt.Errorf("no more pages available")
162	}
163
164	params := *p.params
165	params.NextToken = p.nextToken
166
167	result, err := p.client.GetSamplingRules(ctx, &params, optFns...)
168	if err != nil {
169		return nil, err
170	}
171	p.firstPage = false
172
173	prevToken := p.nextToken
174	p.nextToken = result.NextToken
175
176	if p.options.StopOnDuplicateToken && prevToken != nil && p.nextToken != nil && *prevToken == *p.nextToken {
177		p.nextToken = nil
178	}
179
180	return result, nil
181}
182
183func newServiceMetadataMiddleware_opGetSamplingRules(region string) *awsmiddleware.RegisterServiceMetadata {
184	return &awsmiddleware.RegisterServiceMetadata{
185		Region:        region,
186		ServiceID:     ServiceID,
187		SigningName:   "xray",
188		OperationName: "GetSamplingRules",
189	}
190}
191