Prepare for githubv1.0.0

1 files changed, 35 insertions, 0 deletions

diff --git a/ttun/pubsub.py b/ttun/pubsub.py
new file mode 100644
index 0000000..5d155ea
--- /dev/null
+++ b/ttun/pubsub.py
@@ -0,0 +1,35 @@
+import asyncio
+from contextlib import contextmanager
+from typing import Any, Generator, Iterator
+
+
+class PubSub:
+    _instance = None
+
+    def __init__(self):
+        self.queues: list[asyncio.Queue] = []
+
+    @classmethod
+    def instance(cls) -> 'PubSub':
+        if cls._instance == None:
+            cls._instance = cls()
+        return cls._instance
+
+    @classmethod
+    @contextmanager
+    def subscribe(cls) -> Iterator[asyncio.Queue]:
+        queue = asyncio.Queue()
+
+        instance = cls.instance()
+        instance.queues.append(queue)
+
+        try:
+            yield queue
+        finally:
+            instance.queues.remove(queue)
+            del queue
+
+    @classmethod
+    async def publish(cls, msg: Any):
+        for queue in cls.instance().queues:
+            await queue.put(msg)