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 · a4b837b · a4b837b
2 parents a14197c + 5d9ec02
commit a4b837b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/routing/query.go b/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
+}