feat: added paginator

hashicorp · May 14, 2024 · 2473e2c · 2473e2c
1 parent 92ced03
commit 2473e2c
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/internal/service/ec2/findv2.go b/internal/service/ec2/findv2.go
@@ -501,18 +501,26 @@ func FindCarrierGatewayByID(ctx context.Context, conn *ec2.Client, id string) (*
 }
 
 func FindCarrierGateways(ctx context.Context, conn *ec2.Client, input *ec2.DescribeCarrierGatewaysInput) ([]awstypes.CarrierGateway, error) {
-	output, err := conn.DescribeCarrierGateways(ctx, input)
+	var output []awstypes.CarrierGateway
 
-	if tfawserr.ErrCodeEquals(err, errCodeInvalidCarrierGatewayIDNotFound) {
-		return nil, &retry.NotFoundError{
-			LastError:   err,
-			LastRequest: input,
+	pages := ec2.NewDescribeCarrierGatewaysPaginator(conn, input)
+	for pages.HasMorePages() {
+		page, err := pages.NextPage(ctx)
+
+		if tfawserr.ErrCodeEquals(err, errCodeInvalidCarrierGatewayIDNotFound) {
+			return nil, &retry.NotFoundError{
+				LastError:   err,
+				LastRequest: input,
+			}
 		}
-	}
 
-	if err != nil {
-		return nil, err
+		if err != nil {
+			return nil, err
+		}
+
+		output = append(output, page.CarrierGateways...)
+
 	}
 
-	return output.CarrierGateways, nil
+	return output, nil
 }