mirror of
https://github.com/gitpod-io/gitpod.git
synced 2025-12-08 17:36:30 +00:00
88 lines
2.8 KiB
Go
88 lines
2.8 KiB
Go
// Copyright (c) 2020 Gitpod GmbH. All rights reserved.
|
|
// Licensed under the GNU Affero General Public License (AGPL).
|
|
// See License-AGPL.txt in the project root for license information.
|
|
|
|
package pprof
|
|
|
|
import (
|
|
"math/rand"
|
|
"net/http"
|
|
"net/http/pprof"
|
|
"runtime"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gitpod-io/gitpod/common-go/log"
|
|
)
|
|
|
|
// Serve starts a new HTTP server serving pprof endpoints on the given addr
|
|
func Serve(addr string) {
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("/debug/pprof/", index)
|
|
mux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
|
|
mux.HandleFunc("/debug/pprof/profile", pprof.Profile)
|
|
mux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
|
|
mux.HandleFunc("/debug/pprof/trace", pprof.Trace)
|
|
|
|
log.WithField("addr", addr).Info("serving pprof service")
|
|
err := http.ListenAndServe(addr, mux)
|
|
if err != nil {
|
|
log.WithField("addr", addr).WithError(err).Warn("cannot serve pprof service")
|
|
}
|
|
}
|
|
|
|
func index(w http.ResponseWriter, r *http.Request) {
|
|
if strings.HasPrefix(r.URL.Path, "/debug/pprof/") {
|
|
// according to Ian Lance Taylor it's ok to turn on mutex and block profiling
|
|
// when asking for the actual profile [1]. This handler implements this idea, as
|
|
// discussed in [2]
|
|
//
|
|
// [1] https://groups.google.com/forum/#!topic/golang-nuts/qiHa97XzeCw
|
|
// [2] https://github.com/golang/go/issues/23401
|
|
|
|
var (
|
|
name = strings.TrimPrefix(r.URL.Path, "/debug/pprof/")
|
|
seconds, serr = strconv.ParseInt(r.URL.Query().Get("seconds"), 10, 64)
|
|
)
|
|
if name == "mutex" {
|
|
frac, ferr := strconv.ParseInt(r.URL.Query().Get("frac"), 10, 64)
|
|
if serr == nil && ferr == nil && seconds > 0 && frac > 0 {
|
|
id := rand.Uint32()
|
|
log.WithField("id", id).WithField("frac", frac).WithField("seconds", seconds).Debug("enabled mutex profiling")
|
|
|
|
runtime.SetMutexProfileFraction(int(frac))
|
|
defer func() {
|
|
runtime.SetMutexProfileFraction(0)
|
|
log.WithField("id", id).WithField("frac", frac).WithField("seconds", seconds).Debug("disabled mutex profiling")
|
|
}()
|
|
}
|
|
} else if name == "block" {
|
|
rate, rerr := strconv.ParseInt(r.URL.Query().Get("rate"), 10, 64)
|
|
if rerr == nil && rate > 0 && serr == nil && seconds > 0 {
|
|
id := rand.Uint32()
|
|
log.WithField("id", id).WithField("rate", rate).WithField("seconds", seconds).Debug("enabled mutex block sampling")
|
|
runtime.SetBlockProfileRate(int(rate))
|
|
|
|
defer func() {
|
|
runtime.SetBlockProfileRate(0)
|
|
log.WithField("id", id).WithField("rate", rate).WithField("seconds", seconds).Debug("disabled mutex block sampling")
|
|
}()
|
|
}
|
|
}
|
|
}
|
|
|
|
pprof.Index(w, r)
|
|
}
|
|
|
|
func sleep(w http.ResponseWriter, d time.Duration) {
|
|
var clientGone <-chan bool
|
|
if cn, ok := w.(http.CloseNotifier); ok {
|
|
clientGone = cn.CloseNotify()
|
|
}
|
|
select {
|
|
case <-time.After(d):
|
|
case <-clientGone:
|
|
}
|
|
}
|