init comment-service

services/comment-service/comment_service/events/service.py

@@ -0,0 +1,67 @@
+import asyncio
+from typing import Type
+
+from comment_service.models.config import Config
+from comment_service.models.service import CommentDBRepository
+from comment_service.events.producer import CommentEventProducer
+from comment_service.events.post_consumer import PostEventConsumer
+from comment_service.events.user_consumer import UserEventConsumer
+
+
+class EventConsumersWrapper:
+    """A wrapper class for starting the event consumers.
+    
+    Attributes:
+        _post_consumer (PostEventConsumer): A wrapped consumer.
+        _user_consumer (UserEventConsumer): A wrapped consumer.
+
+    """
+
+    def __init__(self, post_consumer: PostEventConsumer, user_consumer: UserEventConsumer) -> None:
+        """Add the consumers to the wrapper
+
+        Args:
+            post_consumer (PostEventConsumer): Initialised post consumer.
+            user_consumer (UserEventConsumer): Initialised user consumer.
+
+        """
+        self._post_consumer = post_consumer
+        self._user_consumer = user_consumer
+
+    async def start(self) -> None:
+        """Begin consuming events on all the event consumers."""
+        await asyncio.gather(
+            self._post_consumer.start(),
+            self._user_consumer.start()
+        )
+
+
+def create_consumers(config: Config, db_repo: Type[CommentDBRepository]) -> EventConsumersWrapper:
+    """Initialse the event consumers and return them in a wrapper.
+    
+    Args:
+        config (Config): The app configuration instance.
+        db_repo (Type[CommentDBRepository]): The database repo to pass to the consumers.
+    
+    Returns:
+        EventConsumerWrapper
+
+    """
+    post_consumer = PostEventConsumer(config, db_repo)
+    user_consumer = UserEventConsumer(config, db_repo)
+    return EventConsumersWrapper(post_consumer=post_consumer, user_consumer=user_consumer)
+
+
+async def create_producer(config: Config) -> CommentEventProducer:
+    """Create an event producer for the service.
+    
+    Args:
+        config (Config): The app configuration instance.
+
+    Returns:
+        CommentEventProducer
+    
+    """
+    producer = CommentEventProducer(config)
+    await producer.start()
+    return producer