Merge pull request #5 from nodece/main

fix: add pprof
casbin · Mar 2, 2021 · f14a718 · f14a718
2 parents ac368bd + a8062a0
commit f14a718
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/http/service.go b/http/service.go
@@ -8,6 +8,7 @@ import (
 	"io/ioutil"
 	"net"
 	"net/http"
+	"net/http/pprof"
 	"strings"
 	"time"
 
@@ -102,6 +103,13 @@ func NewService(ln net.Listener, tlsConfig *tls.Config, store Store) (*Service,
 		r.Put("/remove", s.handleRemoveNode)
 	})
 
+	// add pprof
+	r.HandleFunc("/debug/pprof/", pprof.Index)
+	r.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
+	r.HandleFunc("/debug/pprof/profile", pprof.Profile)
+	r.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
+	r.HandleFunc("/debug/pprof/trace", pprof.Trace)
+
 	s.srv = &http.Server{
 		Handler:           r,
 		ReadHeaderTimeout: 10 * time.Second,