status-go/vendor/github.com/anacrolix/missinggo/perf/events.go

61 lines
1.2 KiB
Go

package perf
import (
"fmt"
"io"
"net/http"
"sort"
"sync"
"text/tabwriter"
)
var (
mu sync.RWMutex
events = map[string]*Event{}
)
func init() {
http.HandleFunc("/debug/perf", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/plain; charset=UTF-8")
switch r.FormValue("sort") {
case "desc":
writeEventsTableCustomSort(w, func(l, r NamedEvent) bool {
return l.Name < r.Name
})
default:
WriteEventsTable(w)
}
})
}
type NamedEvent struct {
Name string
Event
}
func WriteEventsTable(w io.Writer) {
writeEventsTableCustomSort(w, func(l, r NamedEvent) bool {
return l.Total > r.Total
})
}
func writeEventsTableCustomSort(w io.Writer, less func(l, r NamedEvent) bool) {
tw := tabwriter.NewWriter(w, 0, 0, 2, ' ', 0)
fmt.Fprint(tw, "description\ttotal\tcount\tmin\tmean\tmax\n")
mu.RLock()
es := make([]NamedEvent, 0, len(events))
for d, e := range events {
e.Mu.RLock()
es = append(es, NamedEvent{d, *e})
e.Mu.RUnlock()
}
mu.RUnlock()
sort.Slice(es, func(i, j int) bool {
return less(es[i], es[j])
})
for _, ne := range es {
fmt.Fprintf(tw, "%s\t%v\t%v\t%v\t%v\t%v\n", ne.Name, ne.Total, ne.Count, ne.Min, ne.MeanTime(), ne.Max)
}
tw.Flush()
}