Added additional logic for Azure Functions to use DefaultAzureCredential

RJSonnenberg · Feb 17, 2023 · 64f67bc · 64f67bc
1 parent b876ed9
commit 64f67bc
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/...t.WebJobs.ServiceBusIntegration/Configuration/AzureFunctionsBusConfigurationExtensions.cs b/...t.WebJobs.ServiceBusIntegration/Configuration/AzureFunctionsBusConfigurationExtensions.cs
@@ -50,12 +50,22 @@ public static IServiceCollection AddMassTransitForAzureFunctions(this IServiceCo
                         var connectionString = config[connectionStringConfigurationKey];
 
                         if (string.IsNullOrWhiteSpace(connectionString))
-                            throw new ArgumentNullException(connectionStringConfigurationKey, "A connection string must be used for Azure Functions.");
+                        {
+                            var ns = config["ServiceBusConnection__fullyQualifiedNamespace"]
+                                ?? config[$"{connectionStringConfigurationKey}__fullyQualifiedNamespace"];
+                            if (string.IsNullOrWhiteSpace(ns))
+                            {
+                                throw new ArgumentNullException(connectionStringConfigurationKey,
+                                    "The service bus connection string or namespace was not configured");
+                            }
+
+                            connectionString = $"Endpoint=sb://{ns}";
+                        }
 
                         cfg.Host(connectionString, h =>
                         {
                             if (IsMissingCredentials(connectionString))
-                                h.TokenCredential = new ManagedIdentityCredential();
+                                h.TokenCredential = new DefaultAzureCredential();
                         });
 
                         cfg.UseServiceBusMessageScheduler();