Add AKS resource provider in Statsbeat

microsoft · Nov 7, 2023 · 54e63e2 · 54e63e2
1 parent 892bbb1
commit 54e63e2
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 4 deletions.
diff --git a/AutoCollection/Statsbeat.ts b/AutoCollection/Statsbeat.ts
@@ -302,9 +302,9 @@ class Statsbeat {
             );
             if (currentCounter.totalSuccesfulRequestCount > 0) {
                 this._statbeatMetrics.push({
-                        name: Constants.StatsbeatCounter.REQUEST_SUCCESS,
-                        value: currentCounter.totalSuccesfulRequestCount,
-                        properties: properties
+                    name: Constants.StatsbeatCounter.REQUEST_SUCCESS,
+                    value: currentCounter.totalSuccesfulRequestCount,
+                    properties: properties
                 });
                 currentCounter.totalSuccesfulRequestCount = 0; //Reset
             }
@@ -385,7 +385,11 @@ class Statsbeat {
             let waiting: boolean = false;
             this._resourceProvider = Constants.StatsbeatResourceProvider.unknown;
             this._resourceIdentifier = Constants.StatsbeatResourceProvider.unknown;
-            if (process.env.WEBSITE_SITE_NAME) { // Web apps
+            if (process.env.AKS_ARM_NAMESPACE_ID) {//AKS
+                this._resourceProvider = Constants.StatsbeatResourceProvider.aks;
+                this._resourceIdentifier = process.env.AKS_ARM_NAMESPACE_ID;
+            }
+            else if (process.env.WEBSITE_SITE_NAME) { // Web apps
                 this._resourceProvider = Constants.StatsbeatResourceProvider.appsvc;
                 this._resourceIdentifier = process.env.WEBSITE_SITE_NAME;
                 if (process.env.WEBSITE_HOME_STAMPNAME) {

diff --git a/Declarations/Constants.ts b/Declarations/Constants.ts
@@ -135,6 +135,7 @@ export const StatsbeatTelemetryName = "Statsbeat";
 
 export const StatsbeatResourceProvider = {
     appsvc: "appsvc",
+    aks: "aks",
     functions: "functions",
     vm: "vm",
     unknown: "unknown"

diff --git a/Tests/AutoCollection/Statsbeat.tests.ts b/Tests/AutoCollection/Statsbeat.tests.ts
@@ -120,6 +120,19 @@ describe("AutoCollection/Statsbeat", () => {
                 done();
             }).catch((error) => { done(error); });
         });
+
+        it("aks", (done) => {
+            var newEnv = <{ [id: string]: string }>{};
+            newEnv["AKS_ARM_NAMESPACE_ID"] = "Test AKS cluster and namespace";
+            var originalEnv = process.env;
+            process.env = newEnv;
+            statsBeat["_getResourceProvider"]().then(() => {
+                process.env = originalEnv;
+                assert.equal(statsBeat["_resourceProvider"], "aks");
+                assert.equal(statsBeat["_resourceIdentifier"], "Test AKS cluster and namespace");
+                done();
+            }).catch((error) => { done(error); });
+        });
     });
 
     describe("#trackStatbeats", () => {