status-im
/
go-libp2p-pubsub
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add traces for send/drop rpc 

forgotten!
This commit is contained in:
vyzo 2019-11-11 21:29:14 +02:00
parent 151ec25616
commit ae0fcc6b4d
1 changed files with 30 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
trace.go
View File

@ -152,15 +152,41 @@ func (t *pubsubTracer) RecvRPC(rpc *RPC) {
} }
func (t *pubsubTracer) SendRPC(rpc *RPC, p peer.ID) { func (t *pubsubTracer) SendRPC(rpc *RPC, p peer.ID) {
if t != nil { if t == nil {
// TODO return
} }
now := time.Now().UnixNano()
evt := &pb.TraceEvent{
Type: pb.TraceEvent_SEND_RPC.Enum(),
PeerID: []byte(t.pid),
Timestamp: &now,
SendRPC: &pb.TraceEvent_SendRPC{
SendTo: []byte(rpc.from),
Meta: traceRPCMeta(rpc),
},
}
t.tracer.Trace(evt)
} }
func (t *pubsubTracer) DropRPC(rpc *RPC, p peer.ID) { func (t *pubsubTracer) DropRPC(rpc *RPC, p peer.ID) {
if t != nil { if t == nil {
// TODO return
} }
now := time.Now().UnixNano()
evt := &pb.TraceEvent{
Type: pb.TraceEvent_DROP_RPC.Enum(),
PeerID: []byte(t.pid),
Timestamp: &now,
DropRPC: &pb.TraceEvent_DropRPC{
SendTo: []byte(rpc.from),
Meta: traceRPCMeta(rpc),
},
}
t.tracer.Trace(evt)
} }
func traceRPCMeta(rpc *RPC) *pb.TraceEvent_RPCMeta { func traceRPCMeta(rpc *RPC) *pb.TraceEvent_RPCMeta {