Added pprof to discovery-engine

deployments/helm/configmapfiles/discovery-engine/conf.yaml

@@ -83,4 +83,7 @@ recommend:
 # license
 license:
   enabled: false
-  validate: "user-id"
+  validate: "user-id"
+
+# pprof
+pprof: false

src/conf/local-file.yaml

@@ -124,4 +124,6 @@ recommend:
 license:
   enabled: false
   # validate can be user-id or platform-uuid
-  validate: "user-id"
+  validate: "user-id"
+
+pprof: false

src/conf/local.yaml

@@ -76,4 +76,6 @@ recommend:
 license:
   enabled: false
   # validate can be user-id or platform-uuid
-  validate: "user-id"
+  validate: "user-id"
+
+pprof: false

src/libs/common.go

@@ -153,6 +153,9 @@ func SetDefaultConfig() {
 	viper.SetDefault("recommend.host-policy", true)
 	viper.SetDefault("recommend.admission-controller-policy", true)
 	viper.SetDefault("license.enabled", false)
+
+	// pprof
+	viper.SetDefault("pprof", false)
 }
 
 type cfgArray []string

src/main.go

@@ -1,6 +1,15 @@
 package main
 
 import (
+	"fmt"
+	"math/rand"
+	"net"
+	"os"
+	"time"
+
+	"net/http"
+	_ "net/http/pprof"
+
 	"github.com/accuknox/auto-policy-discovery/src/cluster"
 	"github.com/accuknox/auto-policy-discovery/src/config"
 	"github.com/accuknox/auto-policy-discovery/src/libs"
@@ -10,10 +19,6 @@ import (
 	"github.com/rs/zerolog"
 	"github.com/spf13/viper"
 	"google.golang.org/grpc"
-	"math/rand"
-	"net"
-	"os"
-	"time"
 )
 
 var cfg cluster.Config
@@ -33,6 +38,16 @@ func init() {
 	logger.SetLogLevel(viper.GetString("logging.level"))
 	log = logger.GetInstance()
 
+	//Get pprof flag
+	pprof := viper.GetBool("pprof")
+	if pprof {
+		// Server for pprof
+		go func() {
+			log.Info().Msgf("pprof enabled on :6060\n")
+			fmt.Println(http.ListenAndServe(":6060", nil))
+		}()
+	}
+
 	log.Info().Msgf("NETWORK-POLICY: %+v", config.GetCfgNet())
 	log.Info().Msgf("CILIUM: %+v", config.GetCfgCiliumHubble())
 	log.Info().Msgf("SYSTEM-POLICY: %+v", config.GetCfgSys())