93 lines
2.7 KiB
Go
93 lines
2.7 KiB
Go
|
package wakuv2
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/btcsuite/btcutil/base58"
|
||
|
|
pubsub_pb "github.com/libp2p/go-libp2p-pubsub/pb"
|
||
|
|
|
||
|
|
"go.uber.org/zap"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Trace implements EventTracer interface.
|
||
|
|
// We use custom logging, because we want to base58-encode the peerIDs. And also make the messageIDs readable.
|
||
|
|
func (w *Waku) Trace(evt *pubsub_pb.TraceEvent) {
|
||
|
|
|
||
|
|
f := []zap.Field{
|
||
|
|
zap.String("type", evt.Type.String()),
|
||
|
|
}
|
||
|
|
|
||
|
|
encode := func(peerID []byte) string {
|
||
|
|
return base58.Encode(peerID)
|
||
|
|
}
|
||
|
|
|
||
|
|
encodeMeta := func(meta *pubsub_pb.TraceEvent_RPCMeta) []zap.Field {
|
||
|
|
if meta == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
var out []zap.Field
|
||
|
|
for i, msg := range meta.Messages {
|
||
|
|
out = append(out, zap.String(fmt.Sprintf("MessageID[%d]", i), encode(msg.GetMessageID())))
|
||
|
|
out = append(out, zap.Stringp(fmt.Sprintf("Topic[%d]", i), msg.Topic))
|
||
|
|
}
|
||
|
|
for i, sub := range meta.Subscription {
|
||
|
|
out = append(out, zap.Boolp(fmt.Sprintf("Subscribe[%d]", i), sub.Subscribe))
|
||
|
|
out = append(out, zap.Stringp(fmt.Sprintf("Topic[%d]", i), sub.Topic))
|
||
|
|
}
|
||
|
|
if meta.Control != nil {
|
||
|
|
out = append(out, zap.Any("Control", meta.Control))
|
||
|
|
}
|
||
|
|
return out
|
||
|
|
}
|
||
|
|
|
||
|
|
if evt.PublishMessage != nil {
|
||
|
|
f = append(f, zap.String("MessageID", encode(evt.PublishMessage.MessageID)))
|
||
|
|
}
|
||
|
|
if evt.RejectMessage != nil {
|
||
|
|
f = append(f, zap.String("MessageID", encode(evt.RejectMessage.MessageID)))
|
||
|
|
}
|
||
|
|
if evt.DuplicateMessage != nil {
|
||
|
|
f = append(f, zap.String("MessageID", encode(evt.DuplicateMessage.MessageID)))
|
||
|
|
}
|
||
|
|
if evt.DeliverMessage != nil {
|
||
|
|
f = append(f, zap.String("MessageID", encode(evt.DeliverMessage.MessageID)))
|
||
|
|
}
|
||
|
|
|
||
|
|
if evt.AddPeer != nil {
|
||
|
|
f = append(f, zap.String("PeerID", encode(evt.AddPeer.GetPeerID())))
|
||
|
|
}
|
||
|
|
if evt.RemovePeer != nil {
|
||
|
|
f = append(f, zap.String("PeerID", encode(evt.RemovePeer.GetPeerID())))
|
||
|
|
}
|
||
|
|
|
||
|
|
if evt.RecvRPC != nil {
|
||
|
|
f = append(f, zap.String("ReceivedFrom", encode(evt.RecvRPC.GetReceivedFrom())))
|
||
|
|
f = append(f, encodeMeta(evt.RecvRPC.Meta)...)
|
||
|
|
}
|
||
|
|
if evt.SendRPC != nil {
|
||
|
|
f = append(f, zap.String("SendTo", encode(evt.SendRPC.GetSendTo())))
|
||
|
|
f = append(f, encodeMeta(evt.SendRPC.Meta)...)
|
||
|
|
}
|
||
|
|
if evt.DropRPC != nil {
|
||
|
|
f = append(f, zap.String("SendTo", encode(evt.DropRPC.GetSendTo())))
|
||
|
|
f = append(f, encodeMeta(evt.DropRPC.Meta)...)
|
||
|
|
}
|
||
|
|
|
||
|
|
if evt.Join != nil {
|
||
|
|
f = append(f, zap.String("Topic", evt.Join.GetTopic()))
|
||
|
|
}
|
||
|
|
if evt.Leave != nil {
|
||
|
|
f = append(f, zap.String("Topic", evt.Leave.GetTopic()))
|
||
|
|
}
|
||
|
|
if evt.Graft != nil {
|
||
|
|
f = append(f, zap.String("PeerID", encode(evt.Graft.GetPeerID())))
|
||
|
|
f = append(f, zap.String("Topic", evt.Graft.GetTopic()))
|
||
|
|
}
|
||
|
|
if evt.Prune != nil {
|
||
|
|
f = append(f, zap.String("PeerID", encode(evt.Prune.GetPeerID())))
|
||
|
|
f = append(f, zap.String("Topic", evt.Prune.GetTopic()))
|
||
|
|
}
|
||
|
|
|
||
|
|
w.logger.With(f...).Debug("pubsub trace")
|
||
|
|
}
|