feat(azure): Add warning message when a namespace is not registered

cloudquery · erezrokah · Dec 18, 2022 · Dec 19, 2022 · Dec 19, 2022 · Dec 26, 2022
commit 3db27ce43e252787c5243c79a396f0679c1eabc2
@@ -16,13 +16,15 @@ func SingleSubscriptionMultiplex(meta schema.ClientMeta) []schema.ClientMeta {
 	}
 }
 
-func SubscriptionMultiplexRegisteredNamespace(namespace string) func(schema.ClientMeta) []schema.ClientMeta {
+func SubscriptionMultiplexRegisteredNamespace(table string, namespace string) func(schema.ClientMeta) []schema.ClientMeta {
 	return func(meta schema.ClientMeta) []schema.ClientMeta {
 		client := meta.(*Client)
 		var c = make([]schema.ClientMeta, 0)
 		for _, subId := range client.subscriptions {
 			if _, ok := client.registeredNamespaces[subId][namespace]; ok {
 				c = append(c, client.withSubscription(subId))
+			} else {
+				client.logger.Warn().Msgf("subscription with ID %s does not have the namespace %s registered. Table %s table will be skipped", subId, namespace, table)
 			}
 		}
 		return c

@@ -137,7 +137,9 @@ func initTable(parent *recipes.Table, r *recipes.Table) error {
 	if err != nil {
 		return fmt.Errorf("failed to create table for %s: %w", r.StructName, err)
 	}
-	r.Table.Multiplex = r.Multiplex
+	// Add table name to `SubscriptionMultiplexRegisteredNamespace` if needed
+	r.Table.Multiplex = strings.Replace(r.Multiplex, "client.SubscriptionMultiplexRegisteredNamespace(", "client.SubscriptionMultiplexRegisteredNamespace("+`"`+r.Table.Name+`",`, 1)
+
 	r.Table.Resolver = "fetch" + strcase.ToCamel(r.Name)
 	if r.PreResourceResolver != "" {
 		r.Table.PreResourceResolver = r.PreResourceResolver

diff --git a/plugins/source/azure/go.mod b/plugins/source/azure/go.mod
@@ -146,3 +146,5 @@ require (
 	google.golang.org/protobuf v1.28.1 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 )
+
+replace github.com/cloudquery/plugin-sdk => ../../../../plugin-sdk