track topics in Reject/Duplicate/Deliver events

sstanculeanu · Jul 31, 2020 · 5a3055d · 5a3055d
1 parent fa4d16e
commit 5a3055d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/trace.go b/trace.go
@@ -95,6 +95,7 @@ func (t *pubsubTracer) RejectMessage(msg *Message, reason string) {
 			MessageID:    []byte(t.msgID(msg.Message)),
 			ReceivedFrom: []byte(msg.ReceivedFrom),
 			Reason:       &reason,
+			Topics:       msg.TopicIDs,
 		},
 	}
 
@@ -124,6 +125,7 @@ func (t *pubsubTracer) DuplicateMessage(msg *Message) {
 		DuplicateMessage: &pb.TraceEvent_DuplicateMessage{
 			MessageID:    []byte(t.msgID(msg.Message)),
 			ReceivedFrom: []byte(msg.ReceivedFrom),
+			Topics:       msg.TopicIDs,
 		},
 	}
 
@@ -152,6 +154,7 @@ func (t *pubsubTracer) DeliverMessage(msg *Message) {
 		Timestamp: &now,
 		DeliverMessage: &pb.TraceEvent_DeliverMessage{
 			MessageID: []byte(t.msgID(msg.Message)),
+			Topics:    msg.TopicIDs,
 		},
 	}