Add pprof capabilities to the binary.

clems4ever · clems4ever · commit cc6ea8e7ad43 · 2021-02-23T11:12:02.000+01:00
diff --git a/cmd/go-graphkb/main.go b/cmd/go-graphkb/main.go
@@ -6,6 +6,8 @@ import (
 	"strings"
 	"time"
 
+	_ "net/http/pprof"
+
 	"github.com/clems4ever/go-graphkb/internal/database"
 	"github.com/clems4ever/go-graphkb/internal/knowledge"
 	"github.com/clems4ever/go-graphkb/internal/server"
diff --git a/internal/server/server.go b/internal/server/server.go
@@ -204,6 +204,7 @@ func StartServer(listenInterface string,
 		flushDatabaseHandler = AuthMiddleware(flushDatabaseHandler)
 	}
 
+	r.PathPrefix("/debug/pprof/").Handler(http.DefaultServeMux)
 	r.HandleFunc("/api/sources", listSourcesHandler).Methods("GET")
 	r.HandleFunc("/api/schema", getSourceGraphHandler).Methods("GET")
 	r.HandleFunc("/api/database", getDatabaseDetailsHandler).Methods("GET")

Original file line number	Diff line number	Diff line change
`@@ -204,6 +204,7 @@ func StartServer(listenInterface string,`
`204`	`204`	`flushDatabaseHandler = AuthMiddleware(flushDatabaseHandler)`
`205`	`205`	`}`
`206`	`206`
	`207`	`+ r.PathPrefix("/debug/pprof/").Handler(http.DefaultServeMux)`
`207`	`208`	`r.HandleFunc("/api/sources", listSourcesHandler).Methods("GET")`
`208`	`209`	`r.HandleFunc("/api/schema", getSourceGraphHandler).Methods("GET")`
`209`	`210`	`r.HandleFunc("/api/database", getDatabaseDetailsHandler).Methods("GET")`