mirror of
https://github.com/superseriousbusiness/gotosocial
synced 2024-12-21 10:13:11 +00:00
1131 lines
28 KiB
Go
1131 lines
28 KiB
Go
|
// Copyright 2015 The Go Authors. All rights reserved.
|
||
|
// Use of this source code is governed by a BSD-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
/*
|
||
|
Package trace implements tracing of requests and long-lived objects.
|
||
|
It exports HTTP interfaces on /debug/requests and /debug/events.
|
||
|
|
||
|
A trace.Trace provides tracing for short-lived objects, usually requests.
|
||
|
A request handler might be implemented like this:
|
||
|
|
||
|
func fooHandler(w http.ResponseWriter, req *http.Request) {
|
||
|
tr := trace.New("mypkg.Foo", req.URL.Path)
|
||
|
defer tr.Finish()
|
||
|
...
|
||
|
tr.LazyPrintf("some event %q happened", str)
|
||
|
...
|
||
|
if err := somethingImportant(); err != nil {
|
||
|
tr.LazyPrintf("somethingImportant failed: %v", err)
|
||
|
tr.SetError()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
The /debug/requests HTTP endpoint organizes the traces by family,
|
||
|
errors, and duration. It also provides histogram of request duration
|
||
|
for each family.
|
||
|
|
||
|
A trace.EventLog provides tracing for long-lived objects, such as RPC
|
||
|
connections.
|
||
|
|
||
|
// A Fetcher fetches URL paths for a single domain.
|
||
|
type Fetcher struct {
|
||
|
domain string
|
||
|
events trace.EventLog
|
||
|
}
|
||
|
|
||
|
func NewFetcher(domain string) *Fetcher {
|
||
|
return &Fetcher{
|
||
|
domain,
|
||
|
trace.NewEventLog("mypkg.Fetcher", domain),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (f *Fetcher) Fetch(path string) (string, error) {
|
||
|
resp, err := http.Get("http://" + f.domain + "/" + path)
|
||
|
if err != nil {
|
||
|
f.events.Errorf("Get(%q) = %v", path, err)
|
||
|
return "", err
|
||
|
}
|
||
|
f.events.Printf("Get(%q) = %s", path, resp.Status)
|
||
|
...
|
||
|
}
|
||
|
|
||
|
func (f *Fetcher) Close() error {
|
||
|
f.events.Finish()
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
The /debug/events HTTP endpoint organizes the event logs by family and
|
||
|
by time since the last error. The expanded view displays recent log
|
||
|
entries and the log's call stack.
|
||
|
*/
|
||
|
package trace // import "golang.org/x/net/trace"
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"context"
|
||
|
"fmt"
|
||
|
"html/template"
|
||
|
"io"
|
||
|
"log"
|
||
|
"net"
|
||
|
"net/http"
|
||
|
"net/url"
|
||
|
"runtime"
|
||
|
"sort"
|
||
|
"strconv"
|
||
|
"sync"
|
||
|
"sync/atomic"
|
||
|
"time"
|
||
|
|
||
|
"golang.org/x/net/internal/timeseries"
|
||
|
)
|
||
|
|
||
|
// DebugUseAfterFinish controls whether to debug uses of Trace values after finishing.
|
||
|
// FOR DEBUGGING ONLY. This will slow down the program.
|
||
|
var DebugUseAfterFinish = false
|
||
|
|
||
|
// HTTP ServeMux paths.
|
||
|
const (
|
||
|
debugRequestsPath = "/debug/requests"
|
||
|
debugEventsPath = "/debug/events"
|
||
|
)
|
||
|
|
||
|
// AuthRequest determines whether a specific request is permitted to load the
|
||
|
// /debug/requests or /debug/events pages.
|
||
|
//
|
||
|
// It returns two bools; the first indicates whether the page may be viewed at all,
|
||
|
// and the second indicates whether sensitive events will be shown.
|
||
|
//
|
||
|
// AuthRequest may be replaced by a program to customize its authorization requirements.
|
||
|
//
|
||
|
// The default AuthRequest function returns (true, true) if and only if the request
|
||
|
// comes from localhost/127.0.0.1/[::1].
|
||
|
var AuthRequest = func(req *http.Request) (any, sensitive bool) {
|
||
|
// RemoteAddr is commonly in the form "IP" or "IP:port".
|
||
|
// If it is in the form "IP:port", split off the port.
|
||
|
host, _, err := net.SplitHostPort(req.RemoteAddr)
|
||
|
if err != nil {
|
||
|
host = req.RemoteAddr
|
||
|
}
|
||
|
switch host {
|
||
|
case "localhost", "127.0.0.1", "::1":
|
||
|
return true, true
|
||
|
default:
|
||
|
return false, false
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func init() {
|
||
|
_, pat := http.DefaultServeMux.Handler(&http.Request{URL: &url.URL{Path: debugRequestsPath}})
|
||
|
if pat == debugRequestsPath {
|
||
|
panic("/debug/requests is already registered. You may have two independent copies of " +
|
||
|
"golang.org/x/net/trace in your binary, trying to maintain separate state. This may " +
|
||
|
"involve a vendored copy of golang.org/x/net/trace.")
|
||
|
}
|
||
|
|
||
|
// TODO(jbd): Serve Traces from /debug/traces in the future?
|
||
|
// There is no requirement for a request to be present to have traces.
|
||
|
http.HandleFunc(debugRequestsPath, Traces)
|
||
|
http.HandleFunc(debugEventsPath, Events)
|
||
|
}
|
||
|
|
||
|
// NewContext returns a copy of the parent context
|
||
|
// and associates it with a Trace.
|
||
|
func NewContext(ctx context.Context, tr Trace) context.Context {
|
||
|
return context.WithValue(ctx, contextKey, tr)
|
||
|
}
|
||
|
|
||
|
// FromContext returns the Trace bound to the context, if any.
|
||
|
func FromContext(ctx context.Context) (tr Trace, ok bool) {
|
||
|
tr, ok = ctx.Value(contextKey).(Trace)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// Traces responds with traces from the program.
|
||
|
// The package initialization registers it in http.DefaultServeMux
|
||
|
// at /debug/requests.
|
||
|
//
|
||
|
// It performs authorization by running AuthRequest.
|
||
|
func Traces(w http.ResponseWriter, req *http.Request) {
|
||
|
any, sensitive := AuthRequest(req)
|
||
|
if !any {
|
||
|
http.Error(w, "not allowed", http.StatusUnauthorized)
|
||
|
return
|
||
|
}
|
||
|
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||
|
Render(w, req, sensitive)
|
||
|
}
|
||
|
|
||
|
// Events responds with a page of events collected by EventLogs.
|
||
|
// The package initialization registers it in http.DefaultServeMux
|
||
|
// at /debug/events.
|
||
|
//
|
||
|
// It performs authorization by running AuthRequest.
|
||
|
func Events(w http.ResponseWriter, req *http.Request) {
|
||
|
any, sensitive := AuthRequest(req)
|
||
|
if !any {
|
||
|
http.Error(w, "not allowed", http.StatusUnauthorized)
|
||
|
return
|
||
|
}
|
||
|
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||
|
RenderEvents(w, req, sensitive)
|
||
|
}
|
||
|
|
||
|
// Render renders the HTML page typically served at /debug/requests.
|
||
|
// It does not do any auth checking. The request may be nil.
|
||
|
//
|
||
|
// Most users will use the Traces handler.
|
||
|
func Render(w io.Writer, req *http.Request, sensitive bool) {
|
||
|
data := &struct {
|
||
|
Families []string
|
||
|
ActiveTraceCount map[string]int
|
||
|
CompletedTraces map[string]*family
|
||
|
|
||
|
// Set when a bucket has been selected.
|
||
|
Traces traceList
|
||
|
Family string
|
||
|
Bucket int
|
||
|
Expanded bool
|
||
|
Traced bool
|
||
|
Active bool
|
||
|
ShowSensitive bool // whether to show sensitive events
|
||
|
|
||
|
Histogram template.HTML
|
||
|
HistogramWindow string // e.g. "last minute", "last hour", "all time"
|
||
|
|
||
|
// If non-zero, the set of traces is a partial set,
|
||
|
// and this is the total number.
|
||
|
Total int
|
||
|
}{
|
||
|
CompletedTraces: completedTraces,
|
||
|
}
|
||
|
|
||
|
data.ShowSensitive = sensitive
|
||
|
if req != nil {
|
||
|
// Allow show_sensitive=0 to force hiding of sensitive data for testing.
|
||
|
// This only goes one way; you can't use show_sensitive=1 to see things.
|
||
|
if req.FormValue("show_sensitive") == "0" {
|
||
|
data.ShowSensitive = false
|
||
|
}
|
||
|
|
||
|
if exp, err := strconv.ParseBool(req.FormValue("exp")); err == nil {
|
||
|
data.Expanded = exp
|
||
|
}
|
||
|
if exp, err := strconv.ParseBool(req.FormValue("rtraced")); err == nil {
|
||
|
data.Traced = exp
|
||
|
}
|
||
|
}
|
||
|
|
||
|
completedMu.RLock()
|
||
|
data.Families = make([]string, 0, len(completedTraces))
|
||
|
for fam := range completedTraces {
|
||
|
data.Families = append(data.Families, fam)
|
||
|
}
|
||
|
completedMu.RUnlock()
|
||
|
sort.Strings(data.Families)
|
||
|
|
||
|
// We are careful here to minimize the time spent locking activeMu,
|
||
|
// since that lock is required every time an RPC starts and finishes.
|
||
|
data.ActiveTraceCount = make(map[string]int, len(data.Families))
|
||
|
activeMu.RLock()
|
||
|
for fam, s := range activeTraces {
|
||
|
data.ActiveTraceCount[fam] = s.Len()
|
||
|
}
|
||
|
activeMu.RUnlock()
|
||
|
|
||
|
var ok bool
|
||
|
data.Family, data.Bucket, ok = parseArgs(req)
|
||
|
switch {
|
||
|
case !ok:
|
||
|
// No-op
|
||
|
case data.Bucket == -1:
|
||
|
data.Active = true
|
||
|
n := data.ActiveTraceCount[data.Family]
|
||
|
data.Traces = getActiveTraces(data.Family)
|
||
|
if len(data.Traces) < n {
|
||
|
data.Total = n
|
||
|
}
|
||
|
case data.Bucket < bucketsPerFamily:
|
||
|
if b := lookupBucket(data.Family, data.Bucket); b != nil {
|
||
|
data.Traces = b.Copy(data.Traced)
|
||
|
}
|
||
|
default:
|
||
|
if f := getFamily(data.Family, false); f != nil {
|
||
|
var obs timeseries.Observable
|
||
|
f.LatencyMu.RLock()
|
||
|
switch o := data.Bucket - bucketsPerFamily; o {
|
||
|
case 0:
|
||
|
obs = f.Latency.Minute()
|
||
|
data.HistogramWindow = "last minute"
|
||
|
case 1:
|
||
|
obs = f.Latency.Hour()
|
||
|
data.HistogramWindow = "last hour"
|
||
|
case 2:
|
||
|
obs = f.Latency.Total()
|
||
|
data.HistogramWindow = "all time"
|
||
|
}
|
||
|
f.LatencyMu.RUnlock()
|
||
|
if obs != nil {
|
||
|
data.Histogram = obs.(*histogram).html()
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if data.Traces != nil {
|
||
|
defer data.Traces.Free()
|
||
|
sort.Sort(data.Traces)
|
||
|
}
|
||
|
|
||
|
completedMu.RLock()
|
||
|
defer completedMu.RUnlock()
|
||
|
if err := pageTmpl().ExecuteTemplate(w, "Page", data); err != nil {
|
||
|
log.Printf("net/trace: Failed executing template: %v", err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func parseArgs(req *http.Request) (fam string, b int, ok bool) {
|
||
|
if req == nil {
|
||
|
return "", 0, false
|
||
|
}
|
||
|
fam, bStr := req.FormValue("fam"), req.FormValue("b")
|
||
|
if fam == "" || bStr == "" {
|
||
|
return "", 0, false
|
||
|
}
|
||
|
b, err := strconv.Atoi(bStr)
|
||
|
if err != nil || b < -1 {
|
||
|
return "", 0, false
|
||
|
}
|
||
|
|
||
|
return fam, b, true
|
||
|
}
|
||
|
|
||
|
func lookupBucket(fam string, b int) *traceBucket {
|
||
|
f := getFamily(fam, false)
|
||
|
if f == nil || b < 0 || b >= len(f.Buckets) {
|
||
|
return nil
|
||
|
}
|
||
|
return f.Buckets[b]
|
||
|
}
|
||
|
|
||
|
type contextKeyT string
|
||
|
|
||
|
var contextKey = contextKeyT("golang.org/x/net/trace.Trace")
|
||
|
|
||
|
// Trace represents an active request.
|
||
|
type Trace interface {
|
||
|
// LazyLog adds x to the event log. It will be evaluated each time the
|
||
|
// /debug/requests page is rendered. Any memory referenced by x will be
|
||
|
// pinned until the trace is finished and later discarded.
|
||
|
LazyLog(x fmt.Stringer, sensitive bool)
|
||
|
|
||
|
// LazyPrintf evaluates its arguments with fmt.Sprintf each time the
|
||
|
// /debug/requests page is rendered. Any memory referenced by a will be
|
||
|
// pinned until the trace is finished and later discarded.
|
||
|
LazyPrintf(format string, a ...interface{})
|
||
|
|
||
|
// SetError declares that this trace resulted in an error.
|
||
|
SetError()
|
||
|
|
||
|
// SetRecycler sets a recycler for the trace.
|
||
|
// f will be called for each event passed to LazyLog at a time when
|
||
|
// it is no longer required, whether while the trace is still active
|
||
|
// and the event is discarded, or when a completed trace is discarded.
|
||
|
SetRecycler(f func(interface{}))
|
||
|
|
||
|
// SetTraceInfo sets the trace info for the trace.
|
||
|
// This is currently unused.
|
||
|
SetTraceInfo(traceID, spanID uint64)
|
||
|
|
||
|
// SetMaxEvents sets the maximum number of events that will be stored
|
||
|
// in the trace. This has no effect if any events have already been
|
||
|
// added to the trace.
|
||
|
SetMaxEvents(m int)
|
||
|
|
||
|
// Finish declares that this trace is complete.
|
||
|
// The trace should not be used after calling this method.
|
||
|
Finish()
|
||
|
}
|
||
|
|
||
|
type lazySprintf struct {
|
||
|
format string
|
||
|
a []interface{}
|
||
|
}
|
||
|
|
||
|
func (l *lazySprintf) String() string {
|
||
|
return fmt.Sprintf(l.format, l.a...)
|
||
|
}
|
||
|
|
||
|
// New returns a new Trace with the specified family and title.
|
||
|
func New(family, title string) Trace {
|
||
|
tr := newTrace()
|
||
|
tr.ref()
|
||
|
tr.Family, tr.Title = family, title
|
||
|
tr.Start = time.Now()
|
||
|
tr.maxEvents = maxEventsPerTrace
|
||
|
tr.events = tr.eventsBuf[:0]
|
||
|
|
||
|
activeMu.RLock()
|
||
|
s := activeTraces[tr.Family]
|
||
|
activeMu.RUnlock()
|
||
|
if s == nil {
|
||
|
activeMu.Lock()
|
||
|
s = activeTraces[tr.Family] // check again
|
||
|
if s == nil {
|
||
|
s = new(traceSet)
|
||
|
activeTraces[tr.Family] = s
|
||
|
}
|
||
|
activeMu.Unlock()
|
||
|
}
|
||
|
s.Add(tr)
|
||
|
|
||
|
// Trigger allocation of the completed trace structure for this family.
|
||
|
// This will cause the family to be present in the request page during
|
||
|
// the first trace of this family. We don't care about the return value,
|
||
|
// nor is there any need for this to run inline, so we execute it in its
|
||
|
// own goroutine, but only if the family isn't allocated yet.
|
||
|
completedMu.RLock()
|
||
|
if _, ok := completedTraces[tr.Family]; !ok {
|
||
|
go allocFamily(tr.Family)
|
||
|
}
|
||
|
completedMu.RUnlock()
|
||
|
|
||
|
return tr
|
||
|
}
|
||
|
|
||
|
func (tr *trace) Finish() {
|
||
|
elapsed := time.Since(tr.Start)
|
||
|
tr.mu.Lock()
|
||
|
tr.Elapsed = elapsed
|
||
|
tr.mu.Unlock()
|
||
|
|
||
|
if DebugUseAfterFinish {
|
||
|
buf := make([]byte, 4<<10) // 4 KB should be enough
|
||
|
n := runtime.Stack(buf, false)
|
||
|
tr.finishStack = buf[:n]
|
||
|
}
|
||
|
|
||
|
activeMu.RLock()
|
||
|
m := activeTraces[tr.Family]
|
||
|
activeMu.RUnlock()
|
||
|
m.Remove(tr)
|
||
|
|
||
|
f := getFamily(tr.Family, true)
|
||
|
tr.mu.RLock() // protects tr fields in Cond.match calls
|
||
|
for _, b := range f.Buckets {
|
||
|
if b.Cond.match(tr) {
|
||
|
b.Add(tr)
|
||
|
}
|
||
|
}
|
||
|
tr.mu.RUnlock()
|
||
|
|
||
|
// Add a sample of elapsed time as microseconds to the family's timeseries
|
||
|
h := new(histogram)
|
||
|
h.addMeasurement(elapsed.Nanoseconds() / 1e3)
|
||
|
f.LatencyMu.Lock()
|
||
|
f.Latency.Add(h)
|
||
|
f.LatencyMu.Unlock()
|
||
|
|
||
|
tr.unref() // matches ref in New
|
||
|
}
|
||
|
|
||
|
const (
|
||
|
bucketsPerFamily = 9
|
||
|
tracesPerBucket = 10
|
||
|
maxActiveTraces = 20 // Maximum number of active traces to show.
|
||
|
maxEventsPerTrace = 10
|
||
|
numHistogramBuckets = 38
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
// The active traces.
|
||
|
activeMu sync.RWMutex
|
||
|
activeTraces = make(map[string]*traceSet) // family -> traces
|
||
|
|
||
|
// Families of completed traces.
|
||
|
completedMu sync.RWMutex
|
||
|
completedTraces = make(map[string]*family) // family -> traces
|
||
|
)
|
||
|
|
||
|
type traceSet struct {
|
||
|
mu sync.RWMutex
|
||
|
m map[*trace]bool
|
||
|
|
||
|
// We could avoid the entire map scan in FirstN by having a slice of all the traces
|
||
|
// ordered by start time, and an index into that from the trace struct, with a periodic
|
||
|
// repack of the slice after enough traces finish; we could also use a skip list or similar.
|
||
|
// However, that would shift some of the expense from /debug/requests time to RPC time,
|
||
|
// which is probably the wrong trade-off.
|
||
|
}
|
||
|
|
||
|
func (ts *traceSet) Len() int {
|
||
|
ts.mu.RLock()
|
||
|
defer ts.mu.RUnlock()
|
||
|
return len(ts.m)
|
||
|
}
|
||
|
|
||
|
func (ts *traceSet) Add(tr *trace) {
|
||
|
ts.mu.Lock()
|
||
|
if ts.m == nil {
|
||
|
ts.m = make(map[*trace]bool)
|
||
|
}
|
||
|
ts.m[tr] = true
|
||
|
ts.mu.Unlock()
|
||
|
}
|
||
|
|
||
|
func (ts *traceSet) Remove(tr *trace) {
|
||
|
ts.mu.Lock()
|
||
|
delete(ts.m, tr)
|
||
|
ts.mu.Unlock()
|
||
|
}
|
||
|
|
||
|
// FirstN returns the first n traces ordered by time.
|
||
|
func (ts *traceSet) FirstN(n int) traceList {
|
||
|
ts.mu.RLock()
|
||
|
defer ts.mu.RUnlock()
|
||
|
|
||
|
if n > len(ts.m) {
|
||
|
n = len(ts.m)
|
||
|
}
|
||
|
trl := make(traceList, 0, n)
|
||
|
|
||
|
// Fast path for when no selectivity is needed.
|
||
|
if n == len(ts.m) {
|
||
|
for tr := range ts.m {
|
||
|
tr.ref()
|
||
|
trl = append(trl, tr)
|
||
|
}
|
||
|
sort.Sort(trl)
|
||
|
return trl
|
||
|
}
|
||
|
|
||
|
// Pick the oldest n traces.
|
||
|
// This is inefficient. See the comment in the traceSet struct.
|
||
|
for tr := range ts.m {
|
||
|
// Put the first n traces into trl in the order they occur.
|
||
|
// When we have n, sort trl, and thereafter maintain its order.
|
||
|
if len(trl) < n {
|
||
|
tr.ref()
|
||
|
trl = append(trl, tr)
|
||
|
if len(trl) == n {
|
||
|
// This is guaranteed to happen exactly once during this loop.
|
||
|
sort.Sort(trl)
|
||
|
}
|
||
|
continue
|
||
|
}
|
||
|
if tr.Start.After(trl[n-1].Start) {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
// Find where to insert this one.
|
||
|
tr.ref()
|
||
|
i := sort.Search(n, func(i int) bool { return trl[i].Start.After(tr.Start) })
|
||
|
trl[n-1].unref()
|
||
|
copy(trl[i+1:], trl[i:])
|
||
|
trl[i] = tr
|
||
|
}
|
||
|
|
||
|
return trl
|
||
|
}
|
||
|
|
||
|
func getActiveTraces(fam string) traceList {
|
||
|
activeMu.RLock()
|
||
|
s := activeTraces[fam]
|
||
|
activeMu.RUnlock()
|
||
|
if s == nil {
|
||
|
return nil
|
||
|
}
|
||
|
return s.FirstN(maxActiveTraces)
|
||
|
}
|
||
|
|
||
|
func getFamily(fam string, allocNew bool) *family {
|
||
|
completedMu.RLock()
|
||
|
f := completedTraces[fam]
|
||
|
completedMu.RUnlock()
|
||
|
if f == nil && allocNew {
|
||
|
f = allocFamily(fam)
|
||
|
}
|
||
|
return f
|
||
|
}
|
||
|
|
||
|
func allocFamily(fam string) *family {
|
||
|
completedMu.Lock()
|
||
|
defer completedMu.Unlock()
|
||
|
f := completedTraces[fam]
|
||
|
if f == nil {
|
||
|
f = newFamily()
|
||
|
completedTraces[fam] = f
|
||
|
}
|
||
|
return f
|
||
|
}
|
||
|
|
||
|
// family represents a set of trace buckets and associated latency information.
|
||
|
type family struct {
|
||
|
// traces may occur in multiple buckets.
|
||
|
Buckets [bucketsPerFamily]*traceBucket
|
||
|
|
||
|
// latency time series
|
||
|
LatencyMu sync.RWMutex
|
||
|
Latency *timeseries.MinuteHourSeries
|
||
|
}
|
||
|
|
||
|
func newFamily() *family {
|
||
|
return &family{
|
||
|
Buckets: [bucketsPerFamily]*traceBucket{
|
||
|
{Cond: minCond(0)},
|
||
|
{Cond: minCond(50 * time.Millisecond)},
|
||
|
{Cond: minCond(100 * time.Millisecond)},
|
||
|
{Cond: minCond(200 * time.Millisecond)},
|
||
|
{Cond: minCond(500 * time.Millisecond)},
|
||
|
{Cond: minCond(1 * time.Second)},
|
||
|
{Cond: minCond(10 * time.Second)},
|
||
|
{Cond: minCond(100 * time.Second)},
|
||
|
{Cond: errorCond{}},
|
||
|
},
|
||
|
Latency: timeseries.NewMinuteHourSeries(func() timeseries.Observable { return new(histogram) }),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// traceBucket represents a size-capped bucket of historic traces,
|
||
|
// along with a condition for a trace to belong to the bucket.
|
||
|
type traceBucket struct {
|
||
|
Cond cond
|
||
|
|
||
|
// Ring buffer implementation of a fixed-size FIFO queue.
|
||
|
mu sync.RWMutex
|
||
|
buf [tracesPerBucket]*trace
|
||
|
start int // < tracesPerBucket
|
||
|
length int // <= tracesPerBucket
|
||
|
}
|
||
|
|
||
|
func (b *traceBucket) Add(tr *trace) {
|
||
|
b.mu.Lock()
|
||
|
defer b.mu.Unlock()
|
||
|
|
||
|
i := b.start + b.length
|
||
|
if i >= tracesPerBucket {
|
||
|
i -= tracesPerBucket
|
||
|
}
|
||
|
if b.length == tracesPerBucket {
|
||
|
// "Remove" an element from the bucket.
|
||
|
b.buf[i].unref()
|
||
|
b.start++
|
||
|
if b.start == tracesPerBucket {
|
||
|
b.start = 0
|
||
|
}
|
||
|
}
|
||
|
b.buf[i] = tr
|
||
|
if b.length < tracesPerBucket {
|
||
|
b.length++
|
||
|
}
|
||
|
tr.ref()
|
||
|
}
|
||
|
|
||
|
// Copy returns a copy of the traces in the bucket.
|
||
|
// If tracedOnly is true, only the traces with trace information will be returned.
|
||
|
// The logs will be ref'd before returning; the caller should call
|
||
|
// the Free method when it is done with them.
|
||
|
// TODO(dsymonds): keep track of traced requests in separate buckets.
|
||
|
func (b *traceBucket) Copy(tracedOnly bool) traceList {
|
||
|
b.mu.RLock()
|
||
|
defer b.mu.RUnlock()
|
||
|
|
||
|
trl := make(traceList, 0, b.length)
|
||
|
for i, x := 0, b.start; i < b.length; i++ {
|
||
|
tr := b.buf[x]
|
||
|
if !tracedOnly || tr.spanID != 0 {
|
||
|
tr.ref()
|
||
|
trl = append(trl, tr)
|
||
|
}
|
||
|
x++
|
||
|
if x == b.length {
|
||
|
x = 0
|
||
|
}
|
||
|
}
|
||
|
return trl
|
||
|
}
|
||
|
|
||
|
func (b *traceBucket) Empty() bool {
|
||
|
b.mu.RLock()
|
||
|
defer b.mu.RUnlock()
|
||
|
return b.length == 0
|
||
|
}
|
||
|
|
||
|
// cond represents a condition on a trace.
|
||
|
type cond interface {
|
||
|
match(t *trace) bool
|
||
|
String() string
|
||
|
}
|
||
|
|
||
|
type minCond time.Duration
|
||
|
|
||
|
func (m minCond) match(t *trace) bool { return t.Elapsed >= time.Duration(m) }
|
||
|
func (m minCond) String() string { return fmt.Sprintf("≥%gs", time.Duration(m).Seconds()) }
|
||
|
|
||
|
type errorCond struct{}
|
||
|
|
||
|
func (e errorCond) match(t *trace) bool { return t.IsError }
|
||
|
func (e errorCond) String() string { return "errors" }
|
||
|
|
||
|
type traceList []*trace
|
||
|
|
||
|
// Free calls unref on each element of the list.
|
||
|
func (trl traceList) Free() {
|
||
|
for _, t := range trl {
|
||
|
t.unref()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// traceList may be sorted in reverse chronological order.
|
||
|
func (trl traceList) Len() int { return len(trl) }
|
||
|
func (trl traceList) Less(i, j int) bool { return trl[i].Start.After(trl[j].Start) }
|
||
|
func (trl traceList) Swap(i, j int) { trl[i], trl[j] = trl[j], trl[i] }
|
||
|
|
||
|
// An event is a timestamped log entry in a trace.
|
||
|
type event struct {
|
||
|
When time.Time
|
||
|
Elapsed time.Duration // since previous event in trace
|
||
|
NewDay bool // whether this event is on a different day to the previous event
|
||
|
Recyclable bool // whether this event was passed via LazyLog
|
||
|
Sensitive bool // whether this event contains sensitive information
|
||
|
What interface{} // string or fmt.Stringer
|
||
|
}
|
||
|
|
||
|
// WhenString returns a string representation of the elapsed time of the event.
|
||
|
// It will include the date if midnight was crossed.
|
||
|
func (e event) WhenString() string {
|
||
|
if e.NewDay {
|
||
|
return e.When.Format("2006/01/02 15:04:05.000000")
|
||
|
}
|
||
|
return e.When.Format("15:04:05.000000")
|
||
|
}
|
||
|
|
||
|
// discarded represents a number of discarded events.
|
||
|
// It is stored as *discarded to make it easier to update in-place.
|
||
|
type discarded int
|
||
|
|
||
|
func (d *discarded) String() string {
|
||
|
return fmt.Sprintf("(%d events discarded)", int(*d))
|
||
|
}
|
||
|
|
||
|
// trace represents an active or complete request,
|
||
|
// either sent or received by this program.
|
||
|
type trace struct {
|
||
|
// Family is the top-level grouping of traces to which this belongs.
|
||
|
Family string
|
||
|
|
||
|
// Title is the title of this trace.
|
||
|
Title string
|
||
|
|
||
|
// Start time of the this trace.
|
||
|
Start time.Time
|
||
|
|
||
|
mu sync.RWMutex
|
||
|
events []event // Append-only sequence of events (modulo discards).
|
||
|
maxEvents int
|
||
|
recycler func(interface{})
|
||
|
IsError bool // Whether this trace resulted in an error.
|
||
|
Elapsed time.Duration // Elapsed time for this trace, zero while active.
|
||
|
traceID uint64 // Trace information if non-zero.
|
||
|
spanID uint64
|
||
|
|
||
|
refs int32 // how many buckets this is in
|
||
|
disc discarded // scratch space to avoid allocation
|
||
|
|
||
|
finishStack []byte // where finish was called, if DebugUseAfterFinish is set
|
||
|
|
||
|
eventsBuf [4]event // preallocated buffer in case we only log a few events
|
||
|
}
|
||
|
|
||
|
func (tr *trace) reset() {
|
||
|
// Clear all but the mutex. Mutexes may not be copied, even when unlocked.
|
||
|
tr.Family = ""
|
||
|
tr.Title = ""
|
||
|
tr.Start = time.Time{}
|
||
|
|
||
|
tr.mu.Lock()
|
||
|
tr.Elapsed = 0
|
||
|
tr.traceID = 0
|
||
|
tr.spanID = 0
|
||
|
tr.IsError = false
|
||
|
tr.maxEvents = 0
|
||
|
tr.events = nil
|
||
|
tr.recycler = nil
|
||
|
tr.mu.Unlock()
|
||
|
|
||
|
tr.refs = 0
|
||
|
tr.disc = 0
|
||
|
tr.finishStack = nil
|
||
|
for i := range tr.eventsBuf {
|
||
|
tr.eventsBuf[i] = event{}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// delta returns the elapsed time since the last event or the trace start,
|
||
|
// and whether it spans midnight.
|
||
|
// L >= tr.mu
|
||
|
func (tr *trace) delta(t time.Time) (time.Duration, bool) {
|
||
|
if len(tr.events) == 0 {
|
||
|
return t.Sub(tr.Start), false
|
||
|
}
|
||
|
prev := tr.events[len(tr.events)-1].When
|
||
|
return t.Sub(prev), prev.Day() != t.Day()
|
||
|
}
|
||
|
|
||
|
func (tr *trace) addEvent(x interface{}, recyclable, sensitive bool) {
|
||
|
if DebugUseAfterFinish && tr.finishStack != nil {
|
||
|
buf := make([]byte, 4<<10) // 4 KB should be enough
|
||
|
n := runtime.Stack(buf, false)
|
||
|
log.Printf("net/trace: trace used after finish:\nFinished at:\n%s\nUsed at:\n%s", tr.finishStack, buf[:n])
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
NOTE TO DEBUGGERS
|
||
|
|
||
|
If you are here because your program panicked in this code,
|
||
|
it is almost definitely the fault of code using this package,
|
||
|
and very unlikely to be the fault of this code.
|
||
|
|
||
|
The most likely scenario is that some code elsewhere is using
|
||
|
a trace.Trace after its Finish method is called.
|
||
|
You can temporarily set the DebugUseAfterFinish var
|
||
|
to help discover where that is; do not leave that var set,
|
||
|
since it makes this package much less efficient.
|
||
|
*/
|
||
|
|
||
|
e := event{When: time.Now(), What: x, Recyclable: recyclable, Sensitive: sensitive}
|
||
|
tr.mu.Lock()
|
||
|
e.Elapsed, e.NewDay = tr.delta(e.When)
|
||
|
if len(tr.events) < tr.maxEvents {
|
||
|
tr.events = append(tr.events, e)
|
||
|
} else {
|
||
|
// Discard the middle events.
|
||
|
di := int((tr.maxEvents - 1) / 2)
|
||
|
if d, ok := tr.events[di].What.(*discarded); ok {
|
||
|
(*d)++
|
||
|
} else {
|
||
|
// disc starts at two to count for the event it is replacing,
|
||
|
// plus the next one that we are about to drop.
|
||
|
tr.disc = 2
|
||
|
if tr.recycler != nil && tr.events[di].Recyclable {
|
||
|
go tr.recycler(tr.events[di].What)
|
||
|
}
|
||
|
tr.events[di].What = &tr.disc
|
||
|
}
|
||
|
// The timestamp of the discarded meta-event should be
|
||
|
// the time of the last event it is representing.
|
||
|
tr.events[di].When = tr.events[di+1].When
|
||
|
|
||
|
if tr.recycler != nil && tr.events[di+1].Recyclable {
|
||
|
go tr.recycler(tr.events[di+1].What)
|
||
|
}
|
||
|
copy(tr.events[di+1:], tr.events[di+2:])
|
||
|
tr.events[tr.maxEvents-1] = e
|
||
|
}
|
||
|
tr.mu.Unlock()
|
||
|
}
|
||
|
|
||
|
func (tr *trace) LazyLog(x fmt.Stringer, sensitive bool) {
|
||
|
tr.addEvent(x, true, sensitive)
|
||
|
}
|
||
|
|
||
|
func (tr *trace) LazyPrintf(format string, a ...interface{}) {
|
||
|
tr.addEvent(&lazySprintf{format, a}, false, false)
|
||
|
}
|
||
|
|
||
|
func (tr *trace) SetError() {
|
||
|
tr.mu.Lock()
|
||
|
tr.IsError = true
|
||
|
tr.mu.Unlock()
|
||
|
}
|
||
|
|
||
|
func (tr *trace) SetRecycler(f func(interface{})) {
|
||
|
tr.mu.Lock()
|
||
|
tr.recycler = f
|
||
|
tr.mu.Unlock()
|
||
|
}
|
||
|
|
||
|
func (tr *trace) SetTraceInfo(traceID, spanID uint64) {
|
||
|
tr.mu.Lock()
|
||
|
tr.traceID, tr.spanID = traceID, spanID
|
||
|
tr.mu.Unlock()
|
||
|
}
|
||
|
|
||
|
func (tr *trace) SetMaxEvents(m int) {
|
||
|
tr.mu.Lock()
|
||
|
// Always keep at least three events: first, discarded count, last.
|
||
|
if len(tr.events) == 0 && m > 3 {
|
||
|
tr.maxEvents = m
|
||
|
}
|
||
|
tr.mu.Unlock()
|
||
|
}
|
||
|
|
||
|
func (tr *trace) ref() {
|
||
|
atomic.AddInt32(&tr.refs, 1)
|
||
|
}
|
||
|
|
||
|
func (tr *trace) unref() {
|
||
|
if atomic.AddInt32(&tr.refs, -1) == 0 {
|
||
|
tr.mu.RLock()
|
||
|
if tr.recycler != nil {
|
||
|
// freeTrace clears tr, so we hold tr.recycler and tr.events here.
|
||
|
go func(f func(interface{}), es []event) {
|
||
|
for _, e := range es {
|
||
|
if e.Recyclable {
|
||
|
f(e.What)
|
||
|
}
|
||
|
}
|
||
|
}(tr.recycler, tr.events)
|
||
|
}
|
||
|
tr.mu.RUnlock()
|
||
|
|
||
|
freeTrace(tr)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (tr *trace) When() string {
|
||
|
return tr.Start.Format("2006/01/02 15:04:05.000000")
|
||
|
}
|
||
|
|
||
|
func (tr *trace) ElapsedTime() string {
|
||
|
tr.mu.RLock()
|
||
|
t := tr.Elapsed
|
||
|
tr.mu.RUnlock()
|
||
|
|
||
|
if t == 0 {
|
||
|
// Active trace.
|
||
|
t = time.Since(tr.Start)
|
||
|
}
|
||
|
return fmt.Sprintf("%.6f", t.Seconds())
|
||
|
}
|
||
|
|
||
|
func (tr *trace) Events() []event {
|
||
|
tr.mu.RLock()
|
||
|
defer tr.mu.RUnlock()
|
||
|
return tr.events
|
||
|
}
|
||
|
|
||
|
var traceFreeList = make(chan *trace, 1000) // TODO(dsymonds): Use sync.Pool?
|
||
|
|
||
|
// newTrace returns a trace ready to use.
|
||
|
func newTrace() *trace {
|
||
|
select {
|
||
|
case tr := <-traceFreeList:
|
||
|
return tr
|
||
|
default:
|
||
|
return new(trace)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// freeTrace adds tr to traceFreeList if there's room.
|
||
|
// This is non-blocking.
|
||
|
func freeTrace(tr *trace) {
|
||
|
if DebugUseAfterFinish {
|
||
|
return // never reuse
|
||
|
}
|
||
|
tr.reset()
|
||
|
select {
|
||
|
case traceFreeList <- tr:
|
||
|
default:
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func elapsed(d time.Duration) string {
|
||
|
b := []byte(fmt.Sprintf("%.6f", d.Seconds()))
|
||
|
|
||
|
// For subsecond durations, blank all zeros before decimal point,
|
||
|
// and all zeros between the decimal point and the first non-zero digit.
|
||
|
if d < time.Second {
|
||
|
dot := bytes.IndexByte(b, '.')
|
||
|
for i := 0; i < dot; i++ {
|
||
|
b[i] = ' '
|
||
|
}
|
||
|
for i := dot + 1; i < len(b); i++ {
|
||
|
if b[i] == '0' {
|
||
|
b[i] = ' '
|
||
|
} else {
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return string(b)
|
||
|
}
|
||
|
|
||
|
var pageTmplCache *template.Template
|
||
|
var pageTmplOnce sync.Once
|
||
|
|
||
|
func pageTmpl() *template.Template {
|
||
|
pageTmplOnce.Do(func() {
|
||
|
pageTmplCache = template.Must(template.New("Page").Funcs(template.FuncMap{
|
||
|
"elapsed": elapsed,
|
||
|
"add": func(a, b int) int { return a + b },
|
||
|
}).Parse(pageHTML))
|
||
|
})
|
||
|
return pageTmplCache
|
||
|
}
|
||
|
|
||
|
const pageHTML = `
|
||
|
{{template "Prolog" .}}
|
||
|
{{template "StatusTable" .}}
|
||
|
{{template "Epilog" .}}
|
||
|
|
||
|
{{define "Prolog"}}
|
||
|
<html>
|
||
|
<head>
|
||
|
<title>/debug/requests</title>
|
||
|
<style type="text/css">
|
||
|
body {
|
||
|
font-family: sans-serif;
|
||
|
}
|
||
|
table#tr-status td.family {
|
||
|
padding-right: 2em;
|
||
|
}
|
||
|
table#tr-status td.active {
|
||
|
padding-right: 1em;
|
||
|
}
|
||
|
table#tr-status td.latency-first {
|
||
|
padding-left: 1em;
|
||
|
}
|
||
|
table#tr-status td.empty {
|
||
|
color: #aaa;
|
||
|
}
|
||
|
table#reqs {
|
||
|
margin-top: 1em;
|
||
|
}
|
||
|
table#reqs tr.first {
|
||
|
{{if $.Expanded}}font-weight: bold;{{end}}
|
||
|
}
|
||
|
table#reqs td {
|
||
|
font-family: monospace;
|
||
|
}
|
||
|
table#reqs td.when {
|
||
|
text-align: right;
|
||
|
white-space: nowrap;
|
||
|
}
|
||
|
table#reqs td.elapsed {
|
||
|
padding: 0 0.5em;
|
||
|
text-align: right;
|
||
|
white-space: pre;
|
||
|
width: 10em;
|
||
|
}
|
||
|
address {
|
||
|
font-size: smaller;
|
||
|
margin-top: 5em;
|
||
|
}
|
||
|
</style>
|
||
|
</head>
|
||
|
<body>
|
||
|
|
||
|
<h1>/debug/requests</h1>
|
||
|
{{end}} {{/* end of Prolog */}}
|
||
|
|
||
|
{{define "StatusTable"}}
|
||
|
<table id="tr-status">
|
||
|
{{range $fam := .Families}}
|
||
|
<tr>
|
||
|
<td class="family">{{$fam}}</td>
|
||
|
|
||
|
{{$n := index $.ActiveTraceCount $fam}}
|
||
|
<td class="active {{if not $n}}empty{{end}}">
|
||
|
{{if $n}}<a href="?fam={{$fam}}&b=-1{{if $.Expanded}}&exp=1{{end}}">{{end}}
|
||
|
[{{$n}} active]
|
||
|
{{if $n}}</a>{{end}}
|
||
|
</td>
|
||
|
|
||
|
{{$f := index $.CompletedTraces $fam}}
|
||
|
{{range $i, $b := $f.Buckets}}
|
||
|
{{$empty := $b.Empty}}
|
||
|
<td {{if $empty}}class="empty"{{end}}>
|
||
|
{{if not $empty}}<a href="?fam={{$fam}}&b={{$i}}{{if $.Expanded}}&exp=1{{end}}">{{end}}
|
||
|
[{{.Cond}}]
|
||
|
{{if not $empty}}</a>{{end}}
|
||
|
</td>
|
||
|
{{end}}
|
||
|
|
||
|
{{$nb := len $f.Buckets}}
|
||
|
<td class="latency-first">
|
||
|
<a href="?fam={{$fam}}&b={{$nb}}">[minute]</a>
|
||
|
</td>
|
||
|
<td>
|
||
|
<a href="?fam={{$fam}}&b={{add $nb 1}}">[hour]</a>
|
||
|
</td>
|
||
|
<td>
|
||
|
<a href="?fam={{$fam}}&b={{add $nb 2}}">[total]</a>
|
||
|
</td>
|
||
|
|
||
|
</tr>
|
||
|
{{end}}
|
||
|
</table>
|
||
|
{{end}} {{/* end of StatusTable */}}
|
||
|
|
||
|
{{define "Epilog"}}
|
||
|
{{if $.Traces}}
|
||
|
<hr />
|
||
|
<h3>Family: {{$.Family}}</h3>
|
||
|
|
||
|
{{if or $.Expanded $.Traced}}
|
||
|
<a href="?fam={{$.Family}}&b={{$.Bucket}}">[Normal/Summary]</a>
|
||
|
{{else}}
|
||
|
[Normal/Summary]
|
||
|
{{end}}
|
||
|
|
||
|
{{if or (not $.Expanded) $.Traced}}
|
||
|
<a href="?fam={{$.Family}}&b={{$.Bucket}}&exp=1">[Normal/Expanded]</a>
|
||
|
{{else}}
|
||
|
[Normal/Expanded]
|
||
|
{{end}}
|
||
|
|
||
|
{{if not $.Active}}
|
||
|
{{if or $.Expanded (not $.Traced)}}
|
||
|
<a href="?fam={{$.Family}}&b={{$.Bucket}}&rtraced=1">[Traced/Summary]</a>
|
||
|
{{else}}
|
||
|
[Traced/Summary]
|
||
|
{{end}}
|
||
|
{{if or (not $.Expanded) (not $.Traced)}}
|
||
|
<a href="?fam={{$.Family}}&b={{$.Bucket}}&exp=1&rtraced=1">[Traced/Expanded]</a>
|
||
|
{{else}}
|
||
|
[Traced/Expanded]
|
||
|
{{end}}
|
||
|
{{end}}
|
||
|
|
||
|
{{if $.Total}}
|
||
|
<p><em>Showing <b>{{len $.Traces}}</b> of <b>{{$.Total}}</b> traces.</em></p>
|
||
|
{{end}}
|
||
|
|
||
|
<table id="reqs">
|
||
|
<caption>
|
||
|
{{if $.Active}}Active{{else}}Completed{{end}} Requests
|
||
|
</caption>
|
||
|
<tr><th>When</th><th>Elapsed (s)</th></tr>
|
||
|
{{range $tr := $.Traces}}
|
||
|
<tr class="first">
|
||
|
<td class="when">{{$tr.When}}</td>
|
||
|
<td class="elapsed">{{$tr.ElapsedTime}}</td>
|
||
|
<td>{{$tr.Title}}</td>
|
||
|
{{/* TODO: include traceID/spanID */}}
|
||
|
</tr>
|
||
|
{{if $.Expanded}}
|
||
|
{{range $tr.Events}}
|
||
|
<tr>
|
||
|
<td class="when">{{.WhenString}}</td>
|
||
|
<td class="elapsed">{{elapsed .Elapsed}}</td>
|
||
|
<td>{{if or $.ShowSensitive (not .Sensitive)}}... {{.What}}{{else}}<em>[redacted]</em>{{end}}</td>
|
||
|
</tr>
|
||
|
{{end}}
|
||
|
{{end}}
|
||
|
{{end}}
|
||
|
</table>
|
||
|
{{end}} {{/* if $.Traces */}}
|
||
|
|
||
|
{{if $.Histogram}}
|
||
|
<h4>Latency (µs) of {{$.Family}} over {{$.HistogramWindow}}</h4>
|
||
|
{{$.Histogram}}
|
||
|
{{end}} {{/* if $.Histogram */}}
|
||
|
|
||
|
</body>
|
||
|
</html>
|
||
|
{{end}} {{/* end of Epilog */}}
|
||
|
`
|