add cpu profiling

ebpf/main.go

@@ -185,6 +185,9 @@ func run() {
 	doProfiling := false
 	trafficUtils.InitVar("AKTO_DEBUG_MEM_PROFILING", &doProfiling)
 
+	doProfilingCpu := false
+	trafficUtils.InitVar("AKTO_DEBUG_CPU_PROFILING", &doProfilingCpu)
+
 	if doProfiling {
 		ticker := time.NewTicker(time.Minute) // Create a ticker to trigger every minute
 		defer ticker.Stop()
@@ -194,6 +197,15 @@ func run() {
 		}
 	}
 
+	if doProfilingCpu {
+		ticker := time.NewTicker(time.Minute) // Create a ticker to trigger every minute
+		defer ticker.Stop()
+
+		for range ticker.C {
+			captureCpuProfile() // Capture memory profile every time the ticker ticks
+		}
+	}
+
 	sig := make(chan os.Signal, 1)
 	signal.Notify(sig, syscall.SIGHUP, syscall.SIGINT, syscall.SIGQUIT, syscall.SIGTERM)
 	log.Println("Sniffer is ready")
@@ -207,3 +219,16 @@ func captureMemoryProfile() {
 
 	pprof.WriteHeapProfile(f) // Write memory profile
 }
+
+func captureCpuProfile() {
+	f, err := os.Create("cpu.prof")
+	if err != nil {
+		panic("could not create CPU profile: " + err.Error())
+	}
+	defer f.Close()
+
+	if err := pprof.StartCPUProfile(f); err != nil {
+		panic("could not start CPU profile: " + err.Error())
+	}
+	defer pprof.StopCPUProfile()
+}