mirror of
https://github.com/crazy-max/diun.git
synced 2024-12-22 19:38:28 +00:00
36 lines
951 B
Go
36 lines
951 B
Go
package fgprof
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
// Handler returns an http handler that takes an optional "seconds" query
|
|
// argument that defaults to "30" and produces a profile over this duration.
|
|
// The optional "format" parameter controls if the output is written in
|
|
// Google's "pprof" format (default) or Brendan Gregg's "folded" stack format.
|
|
func Handler() http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
var seconds int
|
|
var err error
|
|
if s := r.URL.Query().Get("seconds"); s == "" {
|
|
seconds = 30
|
|
} else if seconds, err = strconv.Atoi(s); err != nil || seconds <= 0 {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
fmt.Fprintf(w, "bad seconds: %d: %s\n", seconds, err)
|
|
return
|
|
}
|
|
|
|
format := Format(r.URL.Query().Get("format"))
|
|
if format == "" {
|
|
format = FormatPprof
|
|
}
|
|
|
|
stop := Start(w, format)
|
|
defer stop()
|
|
time.Sleep(time.Duration(seconds) * time.Second)
|
|
})
|
|
}
|