Handle pagination for IoTCertificates

fix #831
rebuy-de · Aug 27, 2022 · 18b226f · 18b226f
1 parent fe7fb59
commit 18b226f
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/resources/iot-certificates.go b/resources/iot-certificates.go
@@ -21,16 +21,23 @@ func ListIoTCertificates(sess *session.Session) ([]Resource, error) {
 
 	params := &iot.ListCertificatesInput{}
 
-	output, err := svc.ListCertificates(params)
-	if err != nil {
-		return nil, err
-	}
+	for {
+		output, err := svc.ListCertificates(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, certificate := range output.Certificates {
+			resources = append(resources, &IoTCertificate{
+				svc: svc,
+				ID:  certificate.CertificateId,
+			})
+		}
+		if output.NextMarker == nil {
+			break
+		}
 
-	for _, certificate := range output.Certificates {
-		resources = append(resources, &IoTCertificate{
-			svc: svc,
-			ID:  certificate.CertificateId,
-		})
+		params.Marker = output.NextMarker
 	}
 
 	return resources, nil