Merge pull request #147 from libp2p/feat/subscribes-to-events

feat: add a function to tell if a context subscribes to query events
libp2p · Apr 29, 2020 · 7b406c1 · 7b406c1
2 parents d511b7a + 6be155f
commit 7b406c1
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/core/routing/query.go b/core/routing/query.go
@@ -102,3 +102,10 @@ func PublishQueryEvent(ctx context.Context, ev *QueryEvent) {
 	ech := ich.(*eventChannel)
 	ech.send(ctx, ev)
 }
+
+// SubscribesToQueryEvents returns true if the context subscribes to query
+// events. If this function returns falls, calling `PublishQueryEvent` on the
+// context will be a no-op.
+func SubscribesToQueryEvents(ctx context.Context) bool {
+	return ctx.Value(routingQueryKey{}) != nil
+}