Add loglevel config option when printing application insight logs

Author: rayaisaiah

npm/cmd/start.go

@@ -135,7 +135,7 @@ func start(config npmconfig.Config, flags npmconfig.Flags) error {
 		)
 	}
 
-	err = metrics.CreateTelemetryHandle(config.NPMVersion(), version, npm.GetAIMetadata())
+	err = metrics.CreateTelemetryHandle(config.NPMVersion(), version, npm.GetAIMetadata(), npmconfig.DefaultConfig.LogLevel)
 	if err != nil {
 		klog.Infof("CreateTelemetryHandle failed with error %v. AITelemetry is not initialized.", err)
 	}

npm/cmd/start_daemon.go

@@ -94,7 +94,7 @@ func startDaemon(config npmconfig.Config) error {
 		return fmt.Errorf("failed to create dataplane: %w", err)
 	}
 
-	err = metrics.CreateTelemetryHandle(config.NPMVersion(), version, npm.GetAIMetadata())
+	err = metrics.CreateTelemetryHandle(config.NPMVersion(), version, npm.GetAIMetadata(), npmconfig.DefaultConfig.LogLevel)
 	if err != nil {
 		klog.Infof("CreateTelemetryHandle failed with error %v. AITelemetry is not initialized.", err)
 	}

npm/cmd/start_server.go

@@ -113,7 +113,7 @@ func startControlplane(config npmconfig.Config, flags npmconfig.Flags) error {
 		return fmt.Errorf("failed to create NPM controlplane manager: %w", err)
 	}
 
-	err = metrics.CreateTelemetryHandle(config.NPMVersion(), version, npm.GetAIMetadata())
+	err = metrics.CreateTelemetryHandle(config.NPMVersion(), version, npm.GetAIMetadata(), npmconfig.DefaultConfig.LogLevel)
 	if err != nil {
 		klog.Infof("CreateTelemetryHandle failed with error %v. AITelemetry is not initialized.", err)
 	}

npm/config/config.go

@@ -53,6 +53,9 @@ var DefaultConfig = Config{
 		NetPolInBackground: true,
 		EnableNPMLite:      false,
 	},
+
+	// Setting LogLevel to "info" by default. Set to "debug" to get application insight logs (creates a listener that outputs diagnosticMessageWriter logs).
+	LogLevel: "debug",
 }
 
 type GrpcServerConfig struct {
@@ -82,6 +85,7 @@ type Config struct {
 	MaxPendingNetPols            int     `json:"MaxPendingNetPols,omitempty"`
 	NetPolInvervalInMilliseconds int     `json:"NetPolInvervalInMilliseconds,omitempty"`
 	Toggles                      Toggles `json:"Toggles,omitempty"`
+	LogLevel                     string  `json:"LogLevel,omitempty"`
 }
 
 type Toggles struct {

npm/metrics/ai-utils.go

@@ -3,6 +3,7 @@ package metrics
 import (
 	"fmt"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/Azure/azure-container-networking/aitelemetry"
@@ -21,15 +22,18 @@ var (
 )
 
 // CreateTelemetryHandle creates a handler to initialize AI telemetry
-func CreateTelemetryHandle(npmVersionNum int, imageVersion, aiMetadata string) error {
+func CreateTelemetryHandle(npmVersionNum int, imageVersion, aiMetadata, logLevel string) error {
 	npmVersion = npmVersionNum
+	debugMode := strings.EqualFold(logLevel, "debug")
+	klog.Infof("LogLevel is %s. Debugmode is set to: %v.", logLevel, debugMode)
+
 	aiConfig := aitelemetry.AIConfig{
 		AppName:                   util.AzureNpmFlag,
 		AppVersion:                imageVersion,
 		BatchSize:                 util.BatchSizeInBytes,
 		BatchInterval:             util.BatchIntervalInSecs,
 		RefreshTimeout:            util.RefreshTimeoutInSecs,
-		DebugMode:                 util.DebugMode,
+		DebugMode:                 debugMode,
 		GetEnvRetryCount:          util.GetEnvRetryCount,
 		GetEnvRetryWaitTimeInSecs: util.GetEnvRetryWaitTimeInSecs,
 	}

npm/util/const.go

@@ -238,8 +238,6 @@ const (
 	AiInitializeRetryCount    int = 3
 	AiInitializeRetryInMin    int = 1
 
-	DebugMode bool = true
-
 	ErrorValue float64 = 1
 )