Merge pull request #205 from Flared/mathgl/batching_topic_public

batching topic: Redirect publish to wrapped topic.
Flared · Jun 21, 2022 · 8510023 · 8510023
2 parents be4382d + f8db952
commit 8510023
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/saturn_engine/worker/topics/batching.py b/src/saturn_engine/worker/topics/batching.py
@@ -95,3 +95,6 @@ async def close(self) -> None:
         self.force_done = True
         await self.topic.close()
         await self.task_group.close()
+
+    async def publish(self, message: TopicMessage, wait: bool) -> bool:
+        return await self.topic.publish(message, wait=wait)