operator: added --pprof flag/endpoint

Signed-off-by: Maxime VISONNEAU <[email protected]>
asbubam · Feb 14, 2021 · 9957f5c · 9957f5c
1 parent 1141308
commit 9957f5c
Show file tree

Hide file tree

Showing 7 changed files with 19 additions and 0 deletions.
diff --git a/Documentation/cmdref/cilium-operator-aws.md b/Documentation/cmdref/cilium-operator-aws.md
diff --git a/Documentation/cmdref/cilium-operator-azure.md b/Documentation/cmdref/cilium-operator-azure.md
diff --git a/Documentation/cmdref/cilium-operator-generic.md b/Documentation/cmdref/cilium-operator-generic.md
diff --git a/Documentation/cmdref/cilium-operator.md b/Documentation/cmdref/cilium-operator.md
diff --git a/operator/flags.go b/operator/flags.go
@@ -248,6 +248,9 @@ func init() {
 	flags.String(operatorOption.OperatorAPIServeAddr, "localhost:9234", "Address to serve API requests")
 	option.BindEnv(operatorOption.OperatorAPIServeAddr)
 
+	flags.Bool(operatorOption.PProf, false, "Enable pprof debugging endpoint")
+	option.BindEnv(operatorOption.PProf)
+
 	flags.Bool(operatorOption.SyncK8sServices, true, "Synchronize Kubernetes services to kvstore")
 	option.BindEnv(operatorOption.SyncK8sServices)
 

diff --git a/operator/main.go b/operator/main.go
@@ -40,6 +40,7 @@ import (
 	"github.com/cilium/cilium/pkg/logging/logfields"
 	"github.com/cilium/cilium/pkg/metrics"
 	"github.com/cilium/cilium/pkg/option"
+	"github.com/cilium/cilium/pkg/pprof"
 	"github.com/cilium/cilium/pkg/rand"
 	"github.com/cilium/cilium/pkg/rate"
 	"github.com/cilium/cilium/pkg/version"
@@ -256,6 +257,10 @@ func runOperator() {
 		operatorMetrics.Register()
 	}
 
+	if operatorOption.Config.PProf {
+		pprof.Enable()
+	}
+
 	initK8s(k8sInitDone)
 
 	capabilities := k8sversion.Capabilities()

diff --git a/operator/option/config.go b/operator/option/config.go
@@ -73,6 +73,9 @@ const (
 	// metrics (pass ":Port" to bind on all interfaces, "" is off).
 	OperatorPrometheusServeAddr = "operator-prometheus-serve-addr"
 
+	// PProf enabled pprof debugging endpoint
+	PProf = "pprof"
+
 	// SyncK8sServices synchronizes k8s services into the kvstore
 	SyncK8sServices = "synchronize-k8s-services"
 
@@ -216,6 +219,9 @@ type OperatorConfig struct {
 	OperatorAPIServeAddr        string
 	OperatorPrometheusServeAddr string
 
+	// PProf enables pprof debugging endpoint
+	PProf bool
+
 	// SyncK8sServices synchronizes k8s services into the kvstore
 	SyncK8sServices bool
 
@@ -328,6 +334,7 @@ func (c *OperatorConfig) Populate() {
 	c.NodesGCInterval = viper.GetDuration(NodesGCInterval)
 	c.OperatorAPIServeAddr = viper.GetString(OperatorAPIServeAddr)
 	c.OperatorPrometheusServeAddr = viper.GetString(OperatorPrometheusServeAddr)
+	c.PProf = viper.GetBool(PProf)
 	c.SyncK8sServices = viper.GetBool(SyncK8sServices)
 	c.SyncK8sNodes = viper.GetBool(SyncK8sNodes)
 	c.UnmanagedPodWatcherInterval = viper.GetInt(UnmanagedPodWatcherInterval)