Collect AKS attach rate numerator (Azure#39663)

azure-sdk · Nov 1, 2023 · d73a82a · d73a82a
1 parent 2ea411a
commit d73a82a
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/...onitor/Azure.Monitor.OpenTelemetry.Exporter/src/Internals/EnvironmentVariableConstants.cs b/...onitor/Azure.Monitor.OpenTelemetry.Exporter/src/Internals/EnvironmentVariableConstants.cs
@@ -56,6 +56,16 @@ internal static class EnvironmentVariableConstants
         /// </summary>
         public const string WEBSITE_SITE_NAME = "WEBSITE_SITE_NAME";
 
+        /// <summary>
+        /// INTERNAL ONLY. Used by Statsbeat to get the Kubernetes Service Host.
+        /// </summary>
+        public const string KUBERNETES_SERVICE_HOST = "KUBERNETES_SERVICE_HOST";
+
+        /// <summary>
+        /// INTERNAL ONLY. Used by Statsbeat to get the AKS ARM Namespace ID.
+        /// </summary>
+        public const string AKS_ARM_NAMESPACE_ID = "AKS_ARM_NAMESPACE_ID";
+
         /// <summary>
         /// When set to true, exporter will emit resources as metric telemetry.
         /// </summary>

diff --git a/...tor/Azure.Monitor.OpenTelemetry.Exporter/src/Internals/Statsbeat/AzureMonitorStatsbeat.cs b/...tor/Azure.Monitor.OpenTelemetry.Exporter/src/Internals/Statsbeat/AzureMonitorStatsbeat.cs
@@ -189,6 +189,15 @@ private void SetResourceProviderDetails(IPlatform platform)
                 return;
             }
 
+            var kubernetesSrvHost = platform.GetEnvironmentVariable(EnvironmentVariableConstants.KUBERNETES_SERVICE_HOST);
+            if (kubernetesSrvHost != null)
+            {
+                _resourceProvider = "aks";
+                _resourceProviderId = platform.GetEnvironmentVariable(EnvironmentVariableConstants.AKS_ARM_NAMESPACE_ID) ?? "unknown";
+
+                return;
+            }
+
             _resourceProvider = "unknown";
             _resourceProviderId = "unknown";
         }