Fixes kubeflow#37, add pprof support

dhirajsb · Oct 5, 2023 · e3df569 · e3df569
1 parent ab5d90a
commit e3df569
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -3,10 +3,18 @@ package main
 import (
 	"github.com/golang/glog"
 	"github.com/opendatahub-io/model-registry/cmd"
+	"log"
+	"net/http"
+	_ "net/http/pprof"
 )
 
 func main() {
 	defer glog.Flush()
 
+	// start pprof server on 6060
+	go func() {
+		log.Println(http.ListenAndServe("localhost:6060", nil))
+	}()
+
 	cmd.Execute()
 }