Add configuration for turning profiling on/off

Signed-off-by: Vladislav Byrgazov <vladislav.byrgazov@xored.com>

README.md

@@ -31,6 +31,8 @@ docker build .
 * `NSM_DIAL_TIMEOUT`             - Timeout for the dial the next endpoint
 * `NSM_OPEN_TELEMETRY_ENDPOINT`  - OpenTelemetry Collector Endpoint
 * `NSM_METRICS_EXPORT_INTERVAL`  - interval between mertics exports
+* `NSM_PPROF_ENABLED`            - is pprof enabled (default: "false")
+* `NSM_PPROF_PORT`               - pprof port (default: "6060")
 * `NSM_TUNNEL_IP`                - IP to use for tunnels
 * `NSM_VXLAN_PORT`               - VXLAN port to use
 * `NSM_VPP_API_SOCKET`           - filename of socket to connect to existing VPP instance.

internal/config/config.go

@@ -47,6 +47,8 @@ type Config struct {
 	DialTimeout            time.Duration     `default:"750ms" desc:"Timeout for the dial the next endpoint" split_words:"true"`
 	OpenTelemetryEndpoint  string            `default:"otel-collector.observability.svc.cluster.local:4317" desc:"OpenTelemetry Collector Endpoint" split_words:"true"`
 	MetricsExportInterval  time.Duration     `default:"10s" desc:"interval between mertics exports" split_words:"true"`
+	PprofEnabled           bool              `default:"false" desc:"is pprof enabled" split_words:"true"`
+	PprofPort              string            `default:"6060" desc:"pprof port" split_words:"true"`
 
 	TunnelIP      net.IP       `desc:"IP to use for tunnels" split_words:"true"`
 	VxlanPort     uint16       `default:"0" desc:"VXLAN port to use" split_words:"true"`

main.go

@@ -22,6 +22,8 @@ package main
 import (
 	"context"
 	"crypto/tls"
+	"net/http"
+	"net/http/pprof"
 	"os"
 	"os/signal"
 	"path"
@@ -149,6 +151,20 @@ func main() {
 		}()
 	}
 
+	// ********************************************************************************
+	// Configure pprof
+	// ********************************************************************************
+	if cfg.PprofEnabled {
+		go func() {
+			http.HandleFunc("/debug/pprof/", pprof.Index)
+			http.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
+			http.HandleFunc("/debug/pprof/profile", pprof.Profile)
+			http.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
+			http.HandleFunc("/debug/pprof/trace", pprof.Trace)
+			http.ListenAndServe("localhost:"+cfg.PprofPort, nil)
+		}()
+	}
+
 	// ********************************************************************************
 	log.FromContext(ctx).Infof("executing phase 2: run vpp and get a connection to it (time since start: %s)", time.Since(starttime))
 	// ********************************************************************************