lnbits-exts/nostrclient
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: do not double re-connect

Browse source
This commit is contained in:
Vlad Stan 2023-06-21 17:12:26 +03:00
parent 322679e7c5
commit d08e91b2c7
1 changed files with 18 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

32
nostr/relay_manager.py
View file

@ -22,6 +22,8 @@ class RelayManager:
def add_relay( def add_relay(
self, url: str, read: bool = True, write: bool = True, subscriptions={} self, url: str, read: bool = True, write: bool = True, subscriptions={}
): ):
if url in self.relays:
return
policy = RelayPolicy(read, write) policy = RelayPolicy(read, write)
relay = Relay(url, policy, self.message_pool, subscriptions.copy()) relay = Relay(url, policy, self.message_pool, subscriptions.copy())
self.relays[url] = relay self.relays[url] = relay
@ -42,21 +44,23 @@ class RelayManager:
def open_connections(self, ssl_options: dict = None, proxy: dict = None): def open_connections(self, ssl_options: dict = None, proxy: dict = None):
for relay in self.relays.values(): for relay in self.relays.values():
self.threads[relay.url] = threading.Thread( if relay.url not in self.threads:
target=relay.connect, self.threads[relay.url] = threading.Thread(
args=(ssl_options, proxy), target=relay.connect,
name=f"{relay.url}-thread", args=(ssl_options, proxy),
daemon=True, name=f"{relay.url}-thread",
) daemon=True,
self.threads[relay.url].start() )
self.queue_threads[relay.url] = threading.Thread( self.threads[relay.url].start()
target=relay.queue_worker,
args=(lambda: relay.shutdown,), if relay.url not in self.queue_threads:
name=f"{relay.url}-queue", self.queue_threads[relay.url] = threading.Thread(
daemon=True, target=relay.queue_worker,
) name=f"{relay.url}-queue",
self.queue_threads[relay.url].start() daemon=True,
)
self.queue_threads[relay.url].start()
def close_connections(self): def close_connections(self):
for relay in self.relays.values(): for relay in self.relays.values():