github
/
go-libp2p
mirror of https://github.com/libp2p/go-libp2p.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

Add trace reporter to expose traces to users

pull/1677/head
Marco Munizaga 2 years ago
parent
3e94d039bb
commit
d76fbd7993
1 changed files with 75 additions and 44 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 119
      p2p/host/resource-manager/trace.go

119
p2p/host/resource-manager/trace.go
View File

@ -21,31 +21,54 @@ type trace struct {
cancel func()
closed chan struct{}
mx sync.Mutex
done bool
pend []interface{}
mx sync.Mutex
done bool
pendingWrites []interface{}
reporters []TraceReporter
}
type TraceReporter interface {
// ConsumeEvent consumes a trace event. This is called synchronously,
// implementations should process the event quickly.
ConsumeEvent(TraceEvt)
}
func WithTrace(path string) Option {
return func(r *resourceManager) error {
r.trace = &trace{path: path}
if r.trace == nil {
r.trace = &trace{path: path}
} else {
r.trace.path = path
}
return nil
}
}
func WithTraceReporter(reporter TraceReporter) Option {
return func(r *resourceManager) error {
if r.trace == nil {
r.trace = &trace{reporters: []TraceReporter{reporter}}
}
r.trace.reporters = append(r.trace.reporters, reporter)
return nil
}
}
type TraceEvtTyp string
const (
traceStartEvt = "start"
traceCreateScopeEvt = "create_scope"
traceDestroyScopeEvt = "destroy_scope"
traceReserveMemoryEvt = "reserve_memory"
traceBlockReserveMemoryEvt = "block_reserve_memory"
traceReleaseMemoryEvt = "release_memory"
traceAddStreamEvt = "add_stream"
traceBlockAddStreamEvt = "block_add_stream"
traceRemoveStreamEvt = "remove_stream"
traceAddConnEvt = "add_conn"
traceBlockAddConnEvt = "block_add_conn"
traceRemoveConnEvt = "remove_conn"
TraceStartEvt TraceEvtTyp = "start"
TraceCreateScopeEvt TraceEvtTyp = "create_scope"
TraceDestroyScopeEvt TraceEvtTyp = "destroy_scope"
TraceReserveMemoryEvt TraceEvtTyp = "reserve_memory"
TraceBlockReserveMemoryEvt TraceEvtTyp = "block_reserve_memory"
TraceReleaseMemoryEvt TraceEvtTyp = "release_memory"
TraceAddStreamEvt TraceEvtTyp = "add_stream"
TraceBlockAddStreamEvt TraceEvtTyp = "block_add_stream"
TraceRemoveStreamEvt TraceEvtTyp = "remove_stream"
TraceAddConnEvt TraceEvtTyp = "add_conn"
TraceBlockAddConnEvt TraceEvtTyp = "block_add_conn"
TraceRemoveConnEvt TraceEvtTyp = "remove_conn"
)
type scopeClass struct {
@ -163,7 +186,7 @@ func (s scopeClass) MarshalJSON() ([]byte, error) {
type TraceEvt struct {
Time string
Type string
Type TraceEvtTyp
Scope *scopeClass `json:",omitempty"`
Name string `json:",omitempty"`
@ -199,10 +222,16 @@ func (t *trace) push(evt TraceEvt) {
evt.Scope = &scopeClass{name: evt.Name}
}
t.pend = append(t.pend, evt)
for _, reporter := range t.reporters {
reporter.ConsumeEvent(evt)
}
if t.path != "" {
t.pendingWrites = append(t.pendingWrites, evt)
}
}
func (t *trace) background(out io.WriteCloser) {
func (t *trace) backgroundWriter(out io.WriteCloser) {
defer close(t.closed)
defer out.Close()
@ -218,8 +247,8 @@ func (t *trace) background(out io.WriteCloser) {
getEvents := func() {
t.mx.Lock()
tmp := t.pend
t.pend = pend[:0]
tmp := t.pendingWrites
t.pendingWrites = pend[:0]
pend = tmp
t.mx.Unlock()
}
@ -288,15 +317,17 @@ func (t *trace) Start(limits Limiter) error {
t.ctx, t.cancel = context.WithCancel(context.Background())
t.closed = make(chan struct{})
out, err := os.OpenFile(t.path, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
if err != nil {
return nil
}
if t.path != "" {
out, err := os.OpenFile(t.path, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
if err != nil {
return nil
}
go t.background(out)
go t.backgroundWriter(out)
}
t.push(TraceEvt{
Type: traceStartEvt,
Type: TraceStartEvt,
Limit: limits,
})
@ -329,7 +360,7 @@ func (t *trace) CreateScope(scope string, limit Limit) {
}
t.push(TraceEvt{
Type: traceCreateScopeEvt,
Type: TraceCreateScopeEvt,
Name: scope,
Limit: limit,
})
@ -341,7 +372,7 @@ func (t *trace) DestroyScope(scope string) {
}
t.push(TraceEvt{
Type: traceDestroyScopeEvt,
Type: TraceDestroyScopeEvt,
Name: scope,
})
}
@ -356,7 +387,7 @@ func (t *trace) ReserveMemory(scope string, prio uint8, size, mem int64) {
}
t.push(TraceEvt{
Type: traceReserveMemoryEvt,
Type: TraceReserveMemoryEvt,
Name: scope,
Priority: prio,
Delta: size,
@ -374,7 +405,7 @@ func (t *trace) BlockReserveMemory(scope string, prio uint8, size, mem int64) {
}
t.push(TraceEvt{
Type: traceBlockReserveMemoryEvt,
Type: TraceBlockReserveMemoryEvt,
Name: scope,
Priority: prio,
Delta: size,
@ -392,7 +423,7 @@ func (t *trace) ReleaseMemory(scope string, size, mem int64) {
}
t.push(TraceEvt{
Type: traceReleaseMemoryEvt,
Type: TraceReleaseMemoryEvt,
Name: scope,
Delta: size,
Memory: mem,
@ -412,7 +443,7 @@ func (t *trace) AddStream(scope string, dir network.Direction, nstreamsIn, nstre
}
t.push(TraceEvt{
Type: traceAddStreamEvt,
Type: TraceAddStreamEvt,
Name: scope,
DeltaIn: deltaIn,
DeltaOut: deltaOut,
@ -434,7 +465,7 @@ func (t *trace) BlockAddStream(scope string, dir network.Direction, nstreamsIn,
}
t.push(TraceEvt{
Type: traceBlockAddStreamEvt,
Type: TraceBlockAddStreamEvt,
Name: scope,
DeltaIn: deltaIn,
DeltaOut: deltaOut,
@ -456,7 +487,7 @@ func (t *trace) RemoveStream(scope string, dir network.Direction, nstreamsIn, ns
}
t.push(TraceEvt{
Type: traceRemoveStreamEvt,
Type: TraceRemoveStreamEvt,
Name: scope,
DeltaIn: deltaIn,
DeltaOut: deltaOut,
@ -475,7 +506,7 @@ func (t *trace) AddStreams(scope string, deltaIn, deltaOut, nstreamsIn, nstreams
}
t.push(TraceEvt{
Type: traceAddStreamEvt,
Type: TraceAddStreamEvt,
Name: scope,
DeltaIn: deltaIn,
DeltaOut: deltaOut,
@ -494,7 +525,7 @@ func (t *trace) BlockAddStreams(scope string, deltaIn, deltaOut, nstreamsIn, nst
}
t.push(TraceEvt{
Type: traceBlockAddStreamEvt,
Type: TraceBlockAddStreamEvt,
Name: scope,
DeltaIn: deltaIn,
DeltaOut: deltaOut,
@ -513,7 +544,7 @@ func (t *trace) RemoveStreams(scope string, deltaIn, deltaOut, nstreamsIn, nstre
}
t.push(TraceEvt{
Type: traceRemoveStreamEvt,
Type: TraceRemoveStreamEvt,
Name: scope,
DeltaIn: -deltaIn,
DeltaOut: -deltaOut,
@ -538,7 +569,7 @@ func (t *trace) AddConn(scope string, dir network.Direction, usefd bool, nconnsI
}
t.push(TraceEvt{
Type: traceAddConnEvt,
Type: TraceAddConnEvt,
Name: scope,
DeltaIn: deltaIn,
DeltaOut: deltaOut,
@ -565,7 +596,7 @@ func (t *trace) BlockAddConn(scope string, dir network.Direction, usefd bool, nc
}
t.push(TraceEvt{
Type: traceBlockAddConnEvt,
Type: TraceBlockAddConnEvt,
Name: scope,
DeltaIn: deltaIn,
DeltaOut: deltaOut,
@ -592,7 +623,7 @@ func (t *trace) RemoveConn(scope string, dir network.Direction, usefd bool, ncon
}
t.push(TraceEvt{
Type: traceRemoveConnEvt,
Type: TraceRemoveConnEvt,
Name: scope,
DeltaIn: deltaIn,
DeltaOut: deltaOut,
@ -613,7 +644,7 @@ func (t *trace) AddConns(scope string, deltaIn, deltaOut, deltafd, nconnsIn, nco
}
t.push(TraceEvt{
Type: traceAddConnEvt,
Type: TraceAddConnEvt,
Name: scope,
DeltaIn: deltaIn,
DeltaOut: deltaOut,
@ -634,7 +665,7 @@ func (t *trace) BlockAddConns(scope string, deltaIn, deltaOut, deltafd, nconnsIn
}
t.push(TraceEvt{
Type: traceBlockAddConnEvt,
Type: TraceBlockAddConnEvt,
Name: scope,
DeltaIn: deltaIn,
DeltaOut: deltaOut,
@ -655,7 +686,7 @@ func (t *trace) RemoveConns(scope string, deltaIn, deltaOut, deltafd, nconnsIn,
}
t.push(TraceEvt{
Type: traceRemoveConnEvt,
Type: TraceRemoveConnEvt,
Name: scope,
DeltaIn: -deltaIn,
DeltaOut: -deltaOut,

Write Preview
Loading…
Cancel
Save