mirror of
https://github.com/aljazceru/python-nostr.git
synced 2025-12-19 07:14:23 +01:00
create relay manager class
This commit is contained in:
39
nostr/relay_manager.py
Normal file
39
nostr/relay_manager.py
Normal file
@@ -0,0 +1,39 @@
|
||||
import threading
|
||||
from .filter import Filters
|
||||
from .message_pool import MessagePool
|
||||
from .relay import Relay, RelayPolicy
|
||||
|
||||
class RelayManager:
|
||||
def __init__(self) -> None:
|
||||
self.relays: list[Relay] = []
|
||||
self.message_pool = MessagePool()
|
||||
|
||||
def add_relay(self, url: str, read: bool, write: bool, subscriptions={}):
|
||||
policy = RelayPolicy(read, write)
|
||||
relay = Relay(url, policy, self.message_pool, subscriptions)
|
||||
self.relays.append(relay)
|
||||
|
||||
def add_subscription(self, id: str, filters: Filters):
|
||||
for relay in self.relays:
|
||||
relay.add_subscription(id, filters)
|
||||
|
||||
def close_subscription(self, id: str):
|
||||
for relay in self.relays:
|
||||
relay.close_subscription(id)
|
||||
|
||||
def open_connections(self):
|
||||
for relay in self.relays:
|
||||
threading.Thread(
|
||||
target=relay.connect,
|
||||
name=f"{relay.url}-thread"
|
||||
).start()
|
||||
|
||||
def close_connections(self):
|
||||
for relay in self.relays:
|
||||
relay.close()
|
||||
|
||||
def publish_message(self, message: str):
|
||||
for relay in self.relays:
|
||||
if relay.policy.should_write:
|
||||
relay.publish(message)
|
||||
|
||||
Reference in New Issue
Block a user