add azr metrics in heartbeats

cns/internal/heartbeat.go

@@ -0,0 +1,41 @@
+// Copyright 2018 Microsoft. All rights reserved.
+// MIT License
+
+package internal
+
+import (
+	"context"
+	"strconv"
+	"time"
+
+	"github.com/Azure/azure-container-networking/aitelemetry"
+	"github.com/Azure/azure-container-networking/cns/logger"
+	"github.com/Azure/azure-container-networking/cns/restserver"
+	"github.com/Azure/azure-container-networking/cns/types"
+)
+
+// SendHeartBeat emits node metrics periodically
+func SendHeartBeat(ctx context.Context, heartbeatInterval time.Duration, homeAzMonitor *restserver.HomeAzMonitor) {
+	ticker := time.NewTicker(heartbeatInterval)
+	defer ticker.Stop()
+	for {
+		select {
+		case <-ctx.Done():
+			return
+		case <-ticker.C:
+			metric := aitelemetry.Metric{
+				Name: logger.HeartBeatMetricStr,
+				// This signifies 1 heartbeat is sent. Sum of this metric will give us number of heartbeats received
+				Value:            1.0,
+				CustomDimensions: make(map[string]string),
+			}
+			getHomeAzResp := homeAzMonitor.GetHomeAz(ctx)
+			if getHomeAzResp.Response.ReturnCode == types.Success {
+				metric.CustomDimensions[logger.IsAZRSupportedStr] = strconv.FormatBool(getHomeAzResp.HomeAzResponse.IsSupported)
+				metric.CustomDimensions[logger.HomeAZStr] = strconv.FormatUint(uint64(getHomeAzResp.HomeAzResponse.HomeAz), 10)
+			}
+
+			logger.SendMetric(metric)
+		}
+	}
+}

cns/logger/constants.go

@@ -8,6 +8,8 @@ const (
 	// Dimensions
 	OrchestratorTypeStr = "OrchestratorType"
 	NodeIDStr           = "NodeID"
+	HomeAZStr           = "HomeAZ"
+	IsAZRSupportedStr   = "IsAZRSupported"
 	// CNS Snspshot properties
 	CnsNCSnapshotEventStr         = "CNSNCSnapshot"
 	IpConfigurationStr            = "IPConfiguration"

cns/service/main.go

@@ -32,6 +32,7 @@ import (
 	"github.com/Azure/azure-container-networking/cns/healthserver"
 	"github.com/Azure/azure-container-networking/cns/hnsclient"
 	"github.com/Azure/azure-container-networking/cns/imds"
+	"github.com/Azure/azure-container-networking/cns/internal"
 	"github.com/Azure/azure-container-networking/cns/ipampool"
 	ipampoolv2 "github.com/Azure/azure-container-networking/cns/ipampool/v2"
 	cssctrl "github.com/Azure/azure-container-networking/cns/kubecontroller/clustersubnetstate"
@@ -932,7 +933,7 @@ func main() {
 	}
 
 	if !disableTelemetry {
-		go logger.SendHeartBeat(rootCtx, cnsconfig.TelemetrySettings.HeartBeatIntervalInMins)
+		go internal.SendHeartBeat(rootCtx, time.Minute*time.Duration(cnsconfig.TelemetrySettings.HeartBeatIntervalInMins), homeAzMonitor)
 		go httpRemoteRestService.SendNCSnapShotPeriodically(rootCtx, cnsconfig.TelemetrySettings.SnapshotIntervalInMins)
 	}