Enable remote profiling (#38)

Author: Mario Macias

cmd/netobserv-ebpf-agent.go

@@ -5,6 +5,7 @@ package main
 import (
 	"context"
 	"fmt"
+	"net/http"
 	"os"
 	"os/signal"
 	"syscall"
@@ -13,6 +14,8 @@ import (
 	"github.com/sirupsen/logrus"
 
 	"github.com/netobserv/netobserv-ebpf-agent/pkg/agent"
+
+	_ "net/http/pprof"
 )
 
 func main() {
@@ -23,6 +26,14 @@ func main() {
 	}
 	setLoggerVerbosity(&config)
 
+	if config.ProfilePort != 0 {
+		go func() {
+			logrus.WithField("port", config.ProfilePort).Info("starting PProf HTTP listener")
+			logrus.WithError(http.ListenAndServe(fmt.Sprintf(":%d", config.ProfilePort), nil)).
+				Error("PProf HTTP listener stopped working")
+		}()
+	}
+
 	logrus.WithField("configuration", fmt.Sprintf("%#v", config)).Debugf("configuration loaded")
 
 	flowsAgent, err := agent.FlowsAgent(&config)

docs/config.md

@@ -45,4 +45,6 @@ The following environment variables are available to configure the NetObserv eBF
 * `KAFKA_ASYNC` (default: `true`). If `true`, the message writing process will never block. It also
   means that errors are ignored since the caller will not receive the returned value.
 * `KAFKA_COMPRESSION` (default: `none`). Compression codec to be used to compress messages. Accepted
-  values: `none`, `gzip`, `snappy`, `lz4`, `zstd`.
+  values: `none`, `gzip`, `snappy`, `lz4`, `zstd`.
+* `PROFILE_PORT` (default: unset). Sets the listening port for [Go's Pprof tool](https://pkg.go.dev/net/http/pprof).
+  If it is not set, profile is disabled.

pkg/agent/config.go

@@ -67,4 +67,6 @@ type Config struct {
 	// KafkaCompression sets the compression codec to be used to compress messages. The accepted
 	// values are: none (default), gzip, snappy, lz4, zstd.
 	KafkaCompression string `env:"KAFKA_COMPRESSION" envDefault:"none"`
+	// ProfilePort sets the listening port for Go's Pprof tool. If it is not set, profile is disabled
+	ProfilePort int `env:"PROFILE_PORT"`
 }