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

Add configuration for turning profiling on/off

README.md

@@ -19,6 +19,8 @@ This driver mounts a directory containing a Network Service API socket as an eph
 * `NSM_SOCKET_DIR`      - Path to the NSM API socket directory
 * `NSM_CSI_SOCKET_PATH` - Path to the CSI socket (default: "/nsm-csi/csi.sock")
 * `NSM_VERSION`         - Version (default: "undefined")
+* `NSM_PPROF_ENABLED`   - is pprof enabled (default: "false")
+* `NSM_PPROF_LISTEN_ON` - pprof URL to ListenAndServe (default: "localhost:6060")
 
 ## How it Works
 

internal/config/config.go

@@ -1,4 +1,4 @@
-// Copyright (c) 2023 Cisco and/or its affiliates.
+// Copyright (c) 2023-2024 Cisco and/or its affiliates.
 //
 // SPDX-License-Identifier: Apache-2.0
 //
@@ -28,6 +28,8 @@ type Config struct {
 	SocketDir     string `default:"" desc:"Path to the NSM API socket directory" split_words:"true"`
 	CSISocketPath string `default:"/nsm-csi/csi.sock" desc:"Path to the CSI socket" split_words:"true"`
 	Version       string `default:"undefined" desc:"Version"`
+	PprofEnabled  bool   `default:"false" desc:"is pprof enabled" split_words:"true"`
+	PprofListenOn string `default:"localhost:6060" desc:"pprof URL to ListenAndServe" split_words:"true"`
 }
 
 // IsValid - check if configuration is valid

main.go

@@ -1,4 +1,4 @@
-// Copyright (c) 2023 Cisco and/or its affiliates.
+// Copyright (c) 2023-2024 Cisco and/or its affiliates.
 //
 // SPDX-License-Identifier: Apache-2.0
 //
@@ -26,6 +26,7 @@ import (
 	"github.com/networkservicemesh/cmd-csi-driver/pkg/logkeys"
 	"github.com/networkservicemesh/cmd-csi-driver/pkg/server"
 	"github.com/networkservicemesh/sdk/pkg/tools/log"
+	"github.com/networkservicemesh/sdk/pkg/tools/pprofutils"
 )
 
 func main() {
@@ -42,6 +43,11 @@ func main() {
 		logger.Fatalf("error processing rootConf from env: %+v", err)
 	}
 
+	// Configure pprof
+	if c.PprofEnabled {
+		go pprofutils.ListenAndServe(ctx, c.PprofListenOn)
+	}
+
 	logger.WithField(logkeys.Version, c.Version).
 		WithField(logkeys.NodeID, c.NodeName).
 		WithField(logkeys.NSMSocketDir, c.SocketDir).