ROX-27244: serve pprof server with fleetshard-sync (#2117)

serve pprof server with fleetshard-sync
stackrox · Dec 4, 2024 · d7bf50a · d7bf50a
1 parent c4b135c
commit d7bf50a
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/fleetshard/main.go b/fleetshard/main.go
@@ -4,20 +4,22 @@ package main
 import (
 	"context"
 	"flag"
+	"os"
+	"os/signal"
+	"time"
+
 	"github.com/stackrox/acs-fleet-manager/fleetshard/pkg/central/reconciler"
 	"github.com/stackrox/acs-fleet-manager/internal/certmonitor"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/client-go/informers"
-	"os"
-	"os/signal"
-	"time"
 
 	"github.com/golang/glog"
 	"github.com/stackrox/acs-fleet-manager/fleetshard/config"
 	"github.com/stackrox/acs-fleet-manager/fleetshard/pkg/fleetshardmetrics"
 	"github.com/stackrox/acs-fleet-manager/fleetshard/pkg/k8s"
 	"github.com/stackrox/acs-fleet-manager/fleetshard/pkg/runtime"
 	"github.com/stackrox/acs-fleet-manager/pkg/logger"
+	"github.com/stackrox/acs-fleet-manager/pkg/server/profiler"
 	"golang.org/x/sys/unix"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
@@ -150,6 +152,10 @@ func main() {
 		}
 	}()
 
+	pprofServer := profiler.SingletonPprofServer()
+	pprofServer.Start()
+	defer pprofServer.Stop()
+
 	sigs := make(chan os.Signal, 1)
 	notifySignals := []os.Signal{os.Interrupt, unix.SIGTERM}
 	signal.Notify(sigs, notifySignals...)