🧹 Use built-in AWS paginators. (#6941)

Signed-off-by: Vasil Sirakov <sirakov97@gmail.com>

Author: VasilSirakov

providers/aws/resources/aws_config.go

@@ -247,13 +247,13 @@ func (a *mqlAwsConfig) getRules(conn *connection.AwsConnection) []*jobpool.Job {
 			ctx := context.Background()
 			res := []any{}
 
-			var nextToken *string
-			for {
-				rules, err := svc.DescribeConfigRules(ctx, &configservice.DescribeConfigRulesInput{NextToken: nextToken})
+			paginator := configservice.NewDescribeConfigRulesPaginator(svc, &configservice.DescribeConfigRulesInput{})
+			for paginator.HasMorePages() {
+				page, err := paginator.NextPage(ctx)
 				if err != nil {
 					return nil, err
 				}
-				for _, r := range rules.ConfigRules {
+				for _, r := range page.ConfigRules {
 					jsonSource, err := convert.JsonToDict(r.Source)
 					if err != nil {
 						return nil, err
@@ -273,10 +273,6 @@ func (a *mqlAwsConfig) getRules(conn *connection.AwsConnection) []*jobpool.Job {
 					}
 					res = append(res, mqlRule)
 				}
-				if rules.NextToken == nil {
-					break
-				}
-				nextToken = rules.NextToken
 			}
 			return jobpool.JobResult(res), nil
 		}

providers/aws/resources/aws_redshift.go

@@ -239,20 +239,15 @@ func (a *mqlAwsRedshiftCluster) parameters() ([]any, error) {
 	res := []redshifttypes.Parameter{}
 	for _, name := range clusterGroupNames {
 		stringName := name.(string)
-		var marker *string
-		for {
-			params, err := svc.DescribeClusterParameters(ctx, &redshift.DescribeClusterParametersInput{
-				ParameterGroupName: &stringName,
-				Marker:             marker,
-			})
+		paginator := redshift.NewDescribeClusterParametersPaginator(svc, &redshift.DescribeClusterParametersInput{
+			ParameterGroupName: &stringName,
+		})
+		for paginator.HasMorePages() {
+			params, err := paginator.NextPage(ctx)
 			if err != nil {
 				return nil, err
 			}
 			res = append(res, params.Parameters...)
-			if params.Marker == nil {
-				break
-			}
-			marker = params.Marker
 		}
 	}
 	return convert.JsonToDictSlice(res)

providers/aws/resources/aws_sagemaker.go

@@ -1391,12 +1391,9 @@ func (a *mqlAwsSagemakerDomain) defaultUserSettings() (map[string]any, error) {
 
 func getSagemakerTags(ctx context.Context, svc *sagemaker.Client, arn *string) (map[string]any, error) {
 	tags := make(map[string]any)
-	var nextToken *string
-	for {
-		resp, err := svc.ListTags(ctx, &sagemaker.ListTagsInput{
-			ResourceArn: arn,
-			NextToken:   nextToken,
-		})
+	paginator := sagemaker.NewListTagsPaginator(svc, &sagemaker.ListTagsInput{ResourceArn: arn})
+	for paginator.HasMorePages() {
+		resp, err := paginator.NextPage(ctx)
 		var respErr *http.ResponseError
 		if err != nil {
 			if errors.As(err, &respErr) {
@@ -1411,10 +1408,6 @@ func getSagemakerTags(ctx context.Context, svc *sagemaker.Client, arn *string) (
 				tags[*t.Key] = *t.Value
 			}
 		}
-		if resp.NextToken == nil {
-			break
-		}
-		nextToken = resp.NextToken
 	}
 	return tags, nil
 }