mirror of
https://github.com/tinode/chat.git
synced 2026-06-06 20:18:25 +00:00
54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
// 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)
|
|
}
|