Profiler

Author: Tyrion85

README.md

@@ -92,6 +92,14 @@ data:
 
 Please see [chart/k8sdb-controller](https://github.com/DoodleScheduling/k8sdb-controller) for the helm chart docs.
 
+## Profiling
+To profile controller, access web server on #profilerPort (default 6060). 
+
+In Kubernetes, port-forward to this port, and open the `/debug/pprof` URL in browser. For example, if you port-forward 6060 from container to 6060 on your machine, access:
+```
+http://localhost:6060/debug/pprof/
+```
+
 ## Limitations
 
 Currently there is no garbage collection implemented, meaning all the things created are not removed.

chart/k8sdb-controller/templates/deployment.yaml

@@ -60,6 +60,9 @@ spec:
         - name: probes
           containerPort: {{ .Values.probesPort }}
           protocol: TCP
+        - name: profiler
+          containerPort: {{ .Values.profilerPort }}
+          protocol: TCP
         livenessProbe:
           {{- toYaml .Values.livenessProbe | nindent 10 }}
         readinessProbe:

chart/k8sdb-controller/values.yaml

@@ -39,6 +39,7 @@ podAnnotations: {}
 
 metricsPort: "8080"
 probesPort: "9558"
+profilerPort: "6060"
 
 # Change the metrics path
 metricsPath: /metrics

main.go

@@ -18,6 +18,8 @@ package main
 
 import (
 	"flag"
+	"net/http"
+	_ "net/http/pprof"
 
 	"github.com/spf13/pflag"
 	"github.com/spf13/viper"
@@ -49,6 +51,7 @@ var (
 const (
 	MetricAddr              = "metrics-addr"
 	ProbeAddr               = "probe-addr"
+	ProfilerAddr            = "profiler-addr"
 	EnableLeaderElection    = "enable-leader-election"
 	LeaderElectionNamespace = "leader-election-namespace"
 	Namespaces              = "namespaces"
@@ -59,6 +62,7 @@ const (
 var (
 	metricsAddr             = ":8080"
 	probesAddr              = ":9558"
+	profilerAddr            = ":6060"
 	enableLeaderElection    = false
 	leaderElectionNamespace = ""
 	namespacesConfig        = ""
@@ -75,6 +79,7 @@ func init() {
 func main() {
 	flag.StringVar(&metricsAddr, MetricAddr, ":8080", "The address the metric endpoint binds to.")
 	flag.StringVar(&probesAddr, ProbeAddr, ":9558", "The address of the probe endpoints bind to.")
+	flag.StringVar(&profilerAddr, ProfilerAddr, ":6060", "The address of the profiler endpoints bind to.")
 	flag.BoolVar(&enableLeaderElection, EnableLeaderElection, false,
 		"Enable leader election for controller manager. "+
 			"Enabling this will ensure there is only one active controller manager.")
@@ -101,6 +106,7 @@ func main() {
 	probesAddr = viper.GetString(ProbeAddr)
 	namespacesConfig = viper.GetString(Namespaces)
 	maxConcurrentReconciles = viper.GetInt(MaxConcurrentReconciles)
+	profilerAddr = viper.GetString(ProfilerAddr)
 
 	namespaces := strings.Split(namespacesConfig, ",")
 	options := ctrl.Options{
@@ -121,6 +127,14 @@ func main() {
 		options.LeaderElectionNamespace = leaderElectionNamespace
 	}
 
+	// Profiler
+	go func() {
+		setupLog.Info("Starting profiler...")
+		if err := http.ListenAndServe(profilerAddr, nil); err != nil {
+			setupLog.Error(err, "Profiler failed to start")
+		}
+	}()
+
 	mgr, err := ctrl.NewManager(ctrl.GetConfigOrDie(), options)
 	if err != nil {
 		setupLog.Error(err, "unable to start manager")