cerc-io/lotus
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

Extend peer score event with score components

Browse Source
This commit is contained in:
Matija Petrunic 2021-09-29 16:20:43 +02:00
parent 7b19beaf5d
commit 181b35dc5e
1 changed files with 33 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
node/modules/tracer/tracer.go
View File

@ -37,9 +37,21 @@ type LotusTraceEvent struct {
SourceAuth string `json:"sourceAuth,omitempty"` SourceAuth string `json:"sourceAuth,omitempty"`
} }
type TopicScore struct {
Topic string `json:"topic"`
TimeInMesh time.Duration `json:"timeInMesh"`
FirstMessageDeliveries float64 `json:"firstMessageDeliveries"`
MeshMessageDeliveries float64 `json:"meshMessageDeliveries"`
InvalidMessageDeliveries float64 `json:"invalidMessageDeliveries"`
}
type TraceEventPeerScore struct { type TraceEventPeerScore struct {
PeerID string `json:"peerID"` PeerID string `json:"peerID"`
Score float32 `json:"score"` Score float64 `json:"score"`
AppSpecificScore float64 `json:"appSpecificScore"`
IPColocationFactor float64 `json:"ipColocationFactor"`
BehaviourPenalty float64 `json:"behaviourPenalty"`
Topics []TopicScore `json:"topics"`
} }
type LotusTracer interface { type LotusTracer interface {
@ -52,12 +64,30 @@ type LotusTracer interface {
func (lt *lotusTracer) PeerScores(scores map[peer.ID]*pubsub.PeerScoreSnapshot) { func (lt *lotusTracer) PeerScores(scores map[peer.ID]*pubsub.PeerScoreSnapshot) {
now := time.Now().UnixNano() now := time.Now().UnixNano()
for pid, score := range scores { for pid, score := range scores {
var topics []TopicScore
for topic, snapshot := range score.Topics {
topics = append(topics, TopicScore{
Topic: topic,
TimeInMesh: snapshot.TimeInMesh,
FirstMessageDeliveries: snapshot.FirstMessageDeliveries,
MeshMessageDeliveries: snapshot.MeshMessageDeliveries,
InvalidMessageDeliveries: snapshot.InvalidMessageDeliveries,
})
}
evt := &LotusTraceEvent{ evt := &LotusTraceEvent{
Type: *TraceEventPeerScores.Enum(), Type: *TraceEventPeerScores.Enum(),
PeerID: lt.pid.Pretty(), PeerID: lt.pid.Pretty(),
Timestamp: &now, Timestamp: &now,
SourceAuth: lt.sa, SourceAuth: lt.sa,
PeerScore: TraceEventPeerScore{PeerID: pid.Pretty(), Score: float32(score.Score)}, PeerScore: TraceEventPeerScore{
PeerID: pid.Pretty(),
Score: score.Score,
AppSpecificScore: score.AppSpecificScore,
IPColocationFactor: score.IPColocationFactor,
BehaviourPenalty: score.BehaviourPenalty,
Topics: topics,
},
} }
lt.TraceLotusEvent(evt) lt.TraceLotusEvent(evt)