From 5a3055d33ad1ba46b75b425620586884066d47f4 Mon Sep 17 00:00:00 2001
From: vyzo <vyzo@hackzen.org>
Date: Fri, 31 Jul 2020 10:01:32 +0300
Subject: [PATCH] track topics in Reject/Duplicate/Deliver events

---
 trace.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/trace.go b/trace.go
index 7c8b86d8..f7b4d5a0 100644
--- 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,
 		},
 	}