chore: sync project files

server/http_pprof.go

@@ -0,0 +1,53 @@
+// Debug tooling. Dumps named profile in response to HTTP request at
+// 		http(s)://<host-name>/<configured-path>/<profile-name>
+// See godoc for the list of possible profile names: https://golang.org/pkg/runtime/pprof/#Profile
+
+package main
+
+import (
+	"fmt"
+	"net/http"
+	"path"
+	"runtime/pprof"
+	"strings"
+
+	"github.com/tinode/chat/server/logs"
+)
+
+var pprofHttpRoot string
+
+// Expose debug profiling at the given URL path.
+func servePprof(mux *http.ServeMux, serveAt string) {
+	if serveAt == "" || serveAt == "-" {
+		return
+	}
+
+	pprofHttpRoot = path.Clean("/"+serveAt) + "/"
+	mux.HandleFunc(pprofHttpRoot, profileHandler)
+
+	logs.Info.Printf("pprof: profiling info exposed at '%s'", pprofHttpRoot)
+}
+
+func profileHandler(wrt http.ResponseWriter, req *http.Request) {
+	wrt.Header().Set("X-Content-Type-Options", "nosniff")
+	wrt.Header().Set("Content-Type", "text/plain; charset=utf-8")
+
+	profileName := strings.TrimPrefix(req.URL.Path, pprofHttpRoot)
+
+	profile := pprof.Lookup(profileName)
+	if profile == nil {
+		servePprofError(wrt, http.StatusNotFound, "Unknown profile '"+profileName+"'")
+		return
+	}
+
+	// Respond with the requested profile.
+	profile.WriteTo(wrt, 2)
+}
+
+func servePprofError(wrt http.ResponseWriter, status int, txt string) {
+	wrt.Header().Set("Content-Type", "text/plain; charset=utf-8")
+	wrt.Header().Set("X-Go-Pprof", "1")
+	wrt.Header().Del("Content-Disposition")
+	wrt.WriteHeader(status)
+	fmt.Fprintln(wrt, txt)
+}