refactor: To ensure telemetry is only enabled when both conditions are met.

init/retina/main_linux.go

@@ -4,7 +4,10 @@
 package main
 
 import (
+	"flag"
+
 	"github.com/microsoft/retina/pkg/bpf"
+	"github.com/microsoft/retina/pkg/config"
 	"github.com/microsoft/retina/pkg/log"
 	"github.com/microsoft/retina/pkg/telemetry"
 	"go.uber.org/zap"
@@ -21,23 +24,31 @@ var (
 func main() {
 	// Initialize logger
 	opts := log.GetDefaultLogOpts()
+	zl, err := log.SetupZapLogger(opts)
+	if err != nil {
+		panic(err)
+	}
+
+	// Define a command-line flag "config" with a default value and parse the flags, then create a logger with a name and version.
+	configPath := flag.String("config", "/retina/config/config.yaml", "path to the config file")
+	flag.Parse()
+	l := zl.Named("init-retina").With(zap.String("version", version))
+
+	// Load configuration
+	cfg, err := config.GetConfig(*configPath)
+	if err != nil {
+		l.Fatal("Failed to get config", zap.Error(err))
+	}
 
 	// Enable telemetry if applicationInsightsID is provided
-	if applicationInsightsID != "" {
+	if applicationInsightsID != "" && cfg.EnableTelemetry {
 		opts.EnableTelemetry = true
 		opts.ApplicationInsightsID = applicationInsightsID
 		// Initialize application insights
 		telemetry.InitAppInsights(applicationInsightsID, version)
 		defer telemetry.ShutdownAppInsights()
 		defer telemetry.TrackPanic()
 	}
-
-	zl, err := log.SetupZapLogger(opts)
-	if err != nil {
-		panic(err)
-	}
-	l := zl.Named("init-retina").With(zap.String("version", version))
-
 	// Setup BPF
 	bpf.Setup(l)
 }