From 1fa587a0b0d0e8ac7facee6ff044e5d8afbe5bae Mon Sep 17 00:00:00 2001
From: Tedi Mitiku <tedi.m52@gmail.com>
Date: Sat, 9 Nov 2024 16:45:15 -0800
Subject: [PATCH] register pprof endpoint w web app port 9711

---
 engine/server/engine/main.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/engine/server/engine/main.go b/engine/server/engine/main.go
index 558bbb21af..936bf24aa9 100644
--- a/engine/server/engine/main.go
+++ b/engine/server/engine/main.go
@@ -73,6 +73,7 @@ const (
 	functionPathSeparator     = "."
 	emptyFunctionName         = ""
 	webappPortAddr            = ":9711"
+	pprofPath                 = "/debug/pprof/"
 
 	remoteBackendConfigFilename = "remote_backend_config.json"
 	pathToStaticFolder          = "/run/webapp"
@@ -213,8 +214,9 @@ func runMain() error {
 		}
 		logrus.Debugf("Created environment js file with content: \n%s", envJsFileContent)
 
+		handler := http.NewServeMux()
 		fileServer := http.FileServer(http.Dir(pathToStaticFolder))
-		handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		handler.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 			path, err := filepath.Abs(r.URL.Path)
 			if err != nil {
 				http.Error(w, err.Error(), http.StatusBadRequest)
@@ -236,6 +238,7 @@ func runMain() error {
 			w.Header().Add("Cache-Control", "no-store")
 			fileServer.ServeHTTP(w, r)
 		})
+		handler.Handle(pprofPath, http.HandlerFunc(http.DefaultServeMux.ServeHTTP))
 
 		err := http.ListenAndServe(webappPortAddr, handler)
 		if err != nil {