Add configuration for turning profiling on/off.

Signed-off-by: Vitaliy Guschin <vitaliy.guschin@spirent.com>

.golangci.yml

@@ -36,7 +36,7 @@ linters-settings:
   dupl:
     threshold: 150
   funlen:
-    lines: 120
+    lines: 130
     statements: 60
   goconst:
     min-len: 2
@@ -57,6 +57,10 @@ linters-settings:
     simple: true
     range-loops: true
     for-loops: false
+  gosec:
+    excludes:
+      - G108
+      - G114
   gocritic:
     enabled-checks:
       - appendAssign

README.md

@@ -33,6 +33,8 @@ docker build .
 * `NSM_OPEN_TELEMETRY_ENDPOINT`  - OpenTelemetry Collector Endpoint (default: "otel-collector.observability.svc.cluster.local:4317")
 * `NSM_METRICS_EXPORT_INTERVAL`  - interval between mertics exports (default: "10s")
 * `NSM_KUBELET_QPS`              - kubelet config settings (default: "205")
+* `NSM_ENABLE_PROFILER`          - Enable pprof package HTTP server for profiling runtime data.
+* `NSM_PROFILER_HTTP_PORT`       - Profiling server HTTP port providing data in the format expected by the pprof visualization tool.
 
 # Testing
 

main.go

@@ -24,6 +24,7 @@ package main
 import (
 	"context"
 	"crypto/tls"
+	"fmt"
 	"net/url"
 	"os"
 	"os/signal"
@@ -52,6 +53,9 @@ import (
 	"github.com/networkservicemesh/sdk/pkg/tools/grpcutils"
 	"github.com/networkservicemesh/sdk/pkg/tools/log"
 	"github.com/networkservicemesh/sdk/pkg/tools/log/logruslogger"
+
+	"net/http"
+	_ "net/http/pprof"
 )
 
 // Config is configuration for cmd-registry-memory
@@ -70,7 +74,9 @@ type Config struct {
 	// FWD Refreshes: 1 refresh per sec. 				* 5 fwds
 	// NSC Refreshes: 4 finds (in 1 refresh) per sec. 	* 40 nscs
 	// Total:											= 205
-	KubeletQPS int `default:"205" desc:"kubelet config settings" split_words:"true"`
+	KubeletQPS       int    `default:"205" desc:"kubelet config settings" split_words:"true"`
+	EnableProfiler   bool   `default:"false" desc:"Enable pprof package HTTP server for profiling runtime data. The handled paths all begin with /debug/pprof/." split_words:"true"`
+	ProfilerHTTPPort uint16 `default:"6060" desc:"Profiling server HTTP port providing data in the format expected by the pprof visualization tool." split_words:"true"`
 }
 
 func main() {
@@ -113,6 +119,18 @@ func main() {
 	logrus.SetLevel(l)
 	log.FromContext(ctx).Infof("Config: %#v", config)
 
+	// Enable profiling server
+	if config.EnableProfiler {
+		go func() {
+			log.FromContext(ctx).Infof("Profiler is enabled. Starting HTTP server on %d", config.ProfilerHTTPPort)
+
+			address := fmt.Sprintf("localhost:%d", config.ProfilerHTTPPort)
+			if err = http.ListenAndServe(address, nil); err != nil {
+				log.FromContext(ctx).Errorf("Failed to start profiler: %s", err.Error())
+			}
+		}()
+	}
+
 	// Configure Open Telemetry
 	if opentelemetry.IsEnabled() {
 		collectorAddress := config.OpenTelemetryEndpoint