Merge pull request #243 from Ex4amp1e/add-pprof-enable-conf

Add configuration for turning profiling on/off

README.md

@@ -31,6 +31,8 @@ docker build .
 * `NSM_FEDERATES_WITH`     - Name of the federated domain
 * `NSM_TRUST_DOMAIN`       - Name of the trust domain
 * `NSM_LOG_LEVEL`          - Log level
+* `NSM_PPROF_ENABLED`      - is pprof enabled (default: "false")
+* `NSM_PPROF_LISTEN_ON`    - pprof URL to ListenAndServe (default: "localhost:6060")
 
 # Testing
 

main.go

@@ -1,4 +1,4 @@
-// Copyright (c) 2022-2023 Cisco and/or its affiliates.
+// Copyright (c) 2022-2024 Cisco and/or its affiliates.
 //
 // Copyright (c) 2024 OpenInfra Foundation Europe. All rights reserved.
 //
@@ -62,6 +62,7 @@ import (
 	"github.com/networkservicemesh/sdk/pkg/tools/log"
 	"github.com/networkservicemesh/sdk/pkg/tools/log/logruslogger"
 	"github.com/networkservicemesh/sdk/pkg/tools/nsurl"
+	"github.com/networkservicemesh/sdk/pkg/tools/pprofutils"
 	"github.com/networkservicemesh/sdk/pkg/tools/spiffejwt"
 	"github.com/networkservicemesh/sdk/pkg/tools/spire"
 	"github.com/networkservicemesh/sdk/pkg/tools/token"
@@ -94,6 +95,9 @@ type Config struct {
 	FederatesWith string `default:"k8s.nsm" desc:"Name of the federated domain" split_words:"true"`
 	TrustDomain   string `default:"docker.nsm" desc:"Name of the trust domain" split_words:"true"`
 	LogLevel      string `default:"INFO" desc:"Log level" split_words:"true"`
+
+	PprofEnabled  bool   `default:"false" desc:"is pprof enabled" split_words:"true"`
+	PprofListenOn string `default:"localhost:6060" desc:"pprof URL to ListenAndServe" split_words:"true"`
 }
 
 // Process prints and processes env to config
@@ -179,6 +183,11 @@ func main() {
 	}
 	logrus.SetLevel(level)
 
+	// Configure pprof
+	if config.PprofEnabled {
+		go pprofutils.ListenAndServe(ctx, config.PprofListenOn)
+	}
+
 	// ********************************************************************************
 	log.FromContext(ctx).Infof("executing phase 2: run vpp and get a connection to it")
 	// ********************************************************************************