Getting below mentioned error on azure_key_vault table. Closes #103

turbot · Apr 26, 2021 · f3c0f9c · f3c0f9c
1 parent 6da75a1
commit f3c0f9c
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/azure/table_azure_key_vault.go b/azure/table_azure_key_vault.go
@@ -249,7 +249,7 @@ func getKeyVault(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData
 
 func listKmsKeyVaultDiagnosticSettings(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData) (interface{}, error) {
 	plugin.Logger(ctx).Trace("listKmsKeyVaultDiagnosticSettings")
-	data := h.Item.(keyvault.Vault)
+	id := getKeyVaultID(h.Item)
 
 	// Create session
 	session, err := GetNewSession(ctx, d, "MANAGEMENT")
@@ -261,7 +261,7 @@ func listKmsKeyVaultDiagnosticSettings(ctx context.Context, d *plugin.QueryData,
 	client := insights.NewDiagnosticSettingsClient(subscriptionID)
 	client.Authorizer = session.Authorizer
 
-	op, err := client.List(ctx, *data.ID)
+	op, err := client.List(ctx, id)
 	if err != nil {
 		return nil, err
 	}
@@ -287,3 +287,13 @@ func listKmsKeyVaultDiagnosticSettings(ctx context.Context, d *plugin.QueryData,
 	}
 	return diagnosticSettings, nil
 }
+
+func getKeyVaultID(item interface{}) string {
+	switch item.(type) {
+	case keyvault.Vault:
+		return *item.(keyvault.Vault).ID
+	case keyvault.Resource:
+		return *item.(keyvault.Resource).ID
+	}
+	return ""
+}