/
trace_provider.go
37 lines (29 loc) · 1.06 KB
/
trace_provider.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
package runtime
import (
"time"
"github.com/khulnasoft-lab/gobaseline/database/record"
"github.com/khulnasoft-lab/gobaseline/log"
)
// traceValueProvider can be used to wrap an
// existing value provider to trace an calls to
// their Set and Get methods.
type traceValueProvider struct {
ValueProvider
}
// TraceProvider returns a new ValueProvider that wraps
// vp but traces all Set and Get methods calls.
func TraceProvider(vp ValueProvider) ValueProvider {
return &traceValueProvider{vp}
}
func (tvp *traceValueProvider) Set(r record.Record) (res record.Record, err error) {
defer func(start time.Time) {
log.Tracef("runtime: setting record %q: duration=%s err=%v", r.Key(), time.Since(start), err)
}(time.Now())
return tvp.ValueProvider.Set(r)
}
func (tvp *traceValueProvider) Get(keyOrPrefix string) (records []record.Record, err error) {
defer func(start time.Time) {
log.Tracef("runtime: loading records %q: duration=%s err=%v #records=%d", keyOrPrefix, time.Since(start), err, len(records))
}(time.Now())
return tvp.ValueProvider.Get(keyOrPrefix)
}