feat: read vm size from instance metadata service for windows cse tel…

…emetry (#2663)
Azure · Feb 7, 2020 · 59c2ce0 · 59c2ce0
1 parent 16922c7
commit 59c2ce0
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 0 deletions.
diff --git a/parts/k8s/kuberneteswindowsfunctions.ps1 b/parts/k8s/kuberneteswindowsfunctions.ps1
@@ -86,6 +86,24 @@ function Get-CniVersion {
     }
 }
 
+function Get-InstanceMetadataServiceTelemetry {
+    $keys = @{}
+
+    try {
+        # Write-Log "Querying instance metadata service..."
+        # Note: 2019-04-30 is latest api available in all clouds
+        $metadata = Invoke-RestMethod -Headers @{"Metadata"="true"} -URI "http://169.254.169.254/metadata/instance?api-version=2019-04-30" -Method get
+        # Write-Log ($metadata | ConvertTo-Json)
+
+        $keys.Add("vm_size", $metadata.compute.vmSize)
+    }
+    catch {
+        Write-Log "Error querying instance metadata service."
+    }
+
+    return $keys
+}
+
 # https://stackoverflow.com/a/34559554/697126
 function New-TemporaryDirectory {
     $parent = [System.IO.Path]::GetTempPath()

diff --git a/parts/k8s/kuberneteswindowssetup.ps1 b/parts/k8s/kuberneteswindowssetup.ps1
@@ -193,6 +193,11 @@ try
         }
         $global:AppInsightsClient.Context.Properties["vhd_id"] = $vhdId
 
+        $imdsProperties = Get-InstanceMetadataServiceTelemetry
+        foreach ($key in $imdsProperties.keys) {
+            $global:AppInsightsClient.Context.Properties[$key] = $imdsProperties[$key]
+        }
+
         $global:globalTimer = [System.Diagnostics.Stopwatch]::StartNew()
 
         # Install OpenSSH if SSH enabled

diff --git a/pkg/engine/templates_generated.go b/pkg/engine/templates_generated.go