varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-09 22:50:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add clean-up logics into TrioSubscriptionAPI

Browse Source 
Register an `unsubscribe_fn` when initializing the TrioSubscriptionAPI.
`unsubscribe_fn` is called when subscription is unsubscribed.
This commit is contained in:
mhchia
2020-01-28 00:29:05 +08:00
parent c3ba67ea87
commit 095a848f30
6 changed files with 70 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
libp2p/pubsub/pubsub.py
View File

@ -1,3 +1,4 @@
import functools
import logging
import math
import time
@ -387,9 +388,14 @@ class Pubsub(Service, IPubsub):
if topic_id in self.topic_ids:
return self.subscribed_topics_receive[topic_id]
channels = trio.open_memory_channel[rpc_pb2.Message](math.inf)
send_channel, receive_channel = channels
subscription = TrioSubscriptionAPI(receive_channel)
send_channel, receive_channel = trio.open_memory_channel[rpc_pb2.Message](
math.inf
)
subscription = TrioSubscriptionAPI(
receive_channel,
unsubscribe_fn=functools.partial(self.unsubscribe, topic_id),
)
self.subscribed_topics_send[topic_id] = send_channel
self.subscribed_topics_receive[topic_id] = subscription