Merge pull request #4614 from Particular/john/aud

Adding audience to the metrics api client

Author: johnsimons

src/ServiceControl.Transports.ASBS/AzureQuery.cs

@@ -99,11 +99,11 @@ protected override void InitializeCore(ReadOnlyDictionary<string, string> settin
             Diagnostics.AppendLine("Client secret set");
         }
 
-        ArmEnvironment environment = GetEnvironment();
+        (ArmEnvironment armEnvironment, MetricsQueryAudience metricsQueryAudience) environment = GetEnvironment();
 
         if (managementUrl == null)
         {
-            Diagnostics.AppendLine($"Management Url not set, defaulted to \"{environment.Endpoint}\"");
+            Diagnostics.AppendLine($"Management Url not set, defaulted to \"{environment.armEnvironment.Endpoint}\"");
         }
         else
         {
@@ -126,9 +126,10 @@ protected override void InitializeCore(ReadOnlyDictionary<string, string> settin
             clientCredentials = new ClientSecretCredential(tenantId, clientId, clientSecret);
         }
 
-        client = new MetricsQueryClient(environment.Endpoint, clientCredentials,
+        client = new MetricsQueryClient(environment.armEnvironment.Endpoint, clientCredentials,
             new MetricsQueryClientOptions
             {
+                Audience = environment.metricsQueryAudience,
                 Transport = new HttpClientTransport(
                     new HttpClient(new SocketsHttpHandler
                     {
@@ -138,7 +139,7 @@ protected override void InitializeCore(ReadOnlyDictionary<string, string> settin
         armClient = new ArmClient(clientCredentials, subscriptionId,
             new ArmClientOptions
             {
-                Environment = environment,
+                Environment = environment.armEnvironment,
                 Transport = new HttpClientTransport(
                     new HttpClient(new SocketsHttpHandler
                     {
@@ -148,31 +149,31 @@ protected override void InitializeCore(ReadOnlyDictionary<string, string> settin
 
         return;
 
-        ArmEnvironment GetEnvironment()
+        (ArmEnvironment armEnvironment, MetricsQueryAudience metricsQueryAudience) GetEnvironment()
         {
             if (managementUrlParsed == null)
             {
-                return ArmEnvironment.AzurePublicCloud;
+                return (ArmEnvironment.AzurePublicCloud, MetricsQueryAudience.AzurePublicCloud);
             }
 
             if (managementUrlParsed == ArmEnvironment.AzurePublicCloud.Endpoint)
             {
-                return ArmEnvironment.AzurePublicCloud;
+                return (ArmEnvironment.AzurePublicCloud, MetricsQueryAudience.AzurePublicCloud);
             }
 
             if (managementUrlParsed == ArmEnvironment.AzureChina.Endpoint)
             {
-                return ArmEnvironment.AzureChina;
+                return (ArmEnvironment.AzureChina, MetricsQueryAudience.AzureChina);
             }
 
             if (managementUrlParsed == ArmEnvironment.AzureGermany.Endpoint)
             {
-                return ArmEnvironment.AzureGermany;
+                return (ArmEnvironment.AzureGermany, MetricsQueryAudience.AzurePublicCloud);
             }
 
             if (managementUrlParsed == ArmEnvironment.AzureGovernment.Endpoint)
             {
-                return ArmEnvironment.AzureGovernment;
+                return (ArmEnvironment.AzureGovernment, MetricsQueryAudience.AzureGovernment);
             }
 
             string options = string.Join(", ",
@@ -183,7 +184,7 @@ ArmEnvironment GetEnvironment()
                 }.Select(armEnvironment => $"\"{armEnvironment.Endpoint}\""));
             InitialiseErrors.Add($"Management url configuration is invalid, available options are {options}");
 
-            return ArmEnvironment.AzurePublicCloud;
+            return (ArmEnvironment.AzurePublicCloud, MetricsQueryAudience.AzurePublicCloud);
         }
     }