fix: Fix NotFound error in aws_shield_* resources (#916)

cloudquery · May 17, 2022 · fc9cdcc · fc9cdcc
1 parent 0f2091e
commit fc9cdcc
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/resources/services/shield/protection_groups.go b/resources/services/shield/protection_groups.go
@@ -83,6 +83,9 @@ func fetchShieldProtectionGroups(ctx context.Context, meta schema.ClientMeta, pa
 			o.Region = c.Region
 		})
 		if err != nil {
+			if c.IsNotFoundError(err) {
+				return nil
+			}
 			return diag.WrapError(err)
 		}
 		res <- output.ProtectionGroups

diff --git a/resources/services/shield/protections.go b/resources/services/shield/protections.go
@@ -90,6 +90,9 @@ func fetchShieldProtections(ctx context.Context, meta schema.ClientMeta, parent
 			o.Region = c.Region
 		})
 		if err != nil {
+			if c.IsNotFoundError(err) {
+				return nil
+			}
 			return diag.WrapError(err)
 		}
 		res <- output.Protections

diff --git a/resources/services/shield/subscriptions.go b/resources/services/shield/subscriptions.go
@@ -100,6 +100,9 @@ func fetchShieldSubscriptions(ctx context.Context, meta schema.ClientMeta, paren
 		o.Region = c.Region
 	})
 	if err != nil {
+		if c.IsNotFoundError(err) {
+			return nil
+		}
 		return diag.WrapError(err)
 	}
 	res <- output.Subscription