From 35730637dc5f23bfc9325d2d7155fdafa496475b Mon Sep 17 00:00:00 2001 From: Ian Davis Date: Tue, 14 Feb 2023 16:17:03 +0000 Subject: [PATCH] fix: tracer: emit raw peer ids for compatibility with libp2p tracer --- node/modules/tracer/tracer.go | 9 ++++----- node/modules/tracer/tracer_test.go | 6 ++---- 2 files changed, 6 insertions(+), 9 deletions(-) diff --git a/node/modules/tracer/tracer.go b/node/modules/tracer/tracer.go index 0d0a156d9..72f068d12 100644 --- a/node/modules/tracer/tracer.go +++ b/node/modules/tracer/tracer.go @@ -31,7 +31,7 @@ const ( type LotusTraceEvent struct { Type pubsub_pb.TraceEvent_Type `json:"type,omitempty"` - PeerID string `json:"peerID,omitempty"` + PeerID []byte `json:"peerID,omitempty"` Timestamp *int64 `json:"timestamp,omitempty"` PeerScore TraceEventPeerScore `json:"peerScore,omitempty"` SourceAuth string `json:"sourceAuth,omitempty"` @@ -46,7 +46,7 @@ type TopicScore struct { } type TraceEventPeerScore struct { - PeerID string `json:"peerID"` + PeerID []byte `json:"peerID"` Score float64 `json:"score"` AppSpecificScore float64 `json:"appSpecificScore"` IPColocationFactor float64 `json:"ipColocationFactor"` @@ -77,11 +77,11 @@ func (lt *lotusTracer) PeerScores(scores map[peer.ID]*pubsub.PeerScoreSnapshot) evt := &LotusTraceEvent{ Type: *TraceEventPeerScores.Enum(), - PeerID: lt.pid.Pretty(), + PeerID: []byte(lt.pid), Timestamp: &now, SourceAuth: lt.sa, PeerScore: TraceEventPeerScore{ - PeerID: pid.Pretty(), + PeerID: []byte(pid), Score: score.Score, AppSpecificScore: score.AppSpecificScore, IPColocationFactor: score.IPColocationFactor, @@ -104,7 +104,6 @@ func (lt *lotusTracer) TraceLotusEvent(evt *LotusTraceEvent) { log.Errorf("error while transporting peer scores: %s", err) } } - } func (lt *lotusTracer) Trace(evt *pubsub_pb.TraceEvent) { diff --git a/node/modules/tracer/tracer_test.go b/node/modules/tracer/tracer_test.go index 7ade67861..f0d7b2c0b 100644 --- a/node/modules/tracer/tracer_test.go +++ b/node/modules/tracer/tracer_test.go @@ -30,16 +30,15 @@ func (ttt *testTracerTransport) Transport(evt TracerTransportEvent) error { } func TestTracer_PeerScores(t *testing.T) { - testTransport := NewTestTraceTransport(t, func(t *testing.T, evt TracerTransportEvent) { - require.Equal(t, peerIDA.Pretty(), evt.lotusTraceEvent.PeerID) + require.Equal(t, []byte(peerIDA), evt.lotusTraceEvent.PeerID) require.Equal(t, "source-auth-token-test", evt.lotusTraceEvent.SourceAuth) require.Equal(t, float64(32), evt.lotusTraceEvent.PeerScore.Score) n := time.Now().UnixNano() require.LessOrEqual(t, *evt.lotusTraceEvent.Timestamp, n) - require.Equal(t, peerIDA.Pretty(), evt.lotusTraceEvent.PeerScore.PeerID) + require.Equal(t, []byte(peerIDA), evt.lotusTraceEvent.PeerScore.PeerID) require.Equal(t, 1, len(evt.lotusTraceEvent.PeerScore.Topics)) topic := evt.lotusTraceEvent.PeerScore.Topics[0] @@ -85,7 +84,6 @@ func TestTracer_PubSubTrace(t *testing.T) { PeerID: []byte(peerIDA), Timestamp: &n, }) - } func TestTracer_MultipleTransports(t *testing.T) {