Fix workflow/unsubscribe endpoint

2024-10-14 13:20:41 +02:00
parent 2fce4c56d5
commit 0b01696f5b
1 changed files with 5 additions and 2 deletions
@@ -34,8 +34,11 @@ class WorkflowExchange:
        for publisher in self._publishers.values():
            await self._notify(client_id, publisher)

-    def unsubscribe(self, client_id: str):
-        self._subscribers.remove(client_id)
+    async def unsubscribe(self, client_id: str):
+        if client_id in self._subscribers:
+            self._subscribers.remove(client_id)
+        else:
+            raise KeyError("No subscriber found with id " + client_id)

    async def _notify(self, client_id: str, publisher: Publisher):
        data = {