RabbitMQ Broker

Не зависящий от фреймворков пакет для общения между микросервисами. Пакет предоставляет интерфейс для работы с брокером сообщений RabbitMQ и базовый класс цепочки обработчиков.

Пакет реализует паттерн цепочка обязанностей в синхронном и асинхронном виде.

Конфигурация

Использование

Цепочка обработчиков

Для использования цепочки обязанностей нужно импортировать базовый класс звена и унаследоваться от него. Необходимо установить параметр request_type. Обязательно должен быть переопределен абстрактный метод BaseChain.get_response_body(self, data):

from rmq_broker.chains.base import BaseChain


class LoginChain(BaseChain):
    request_type = "login"

    def get_response_body(self, data: dict) -> dict:
        ...

Также, можно унаследоваться от асинхронного варианта реализации:

from rmq_broker.async_chains.base import BaseChain


class LoginChain(BaseChain):
    request_type = "login"

    async def get_response_body(self, data: dict) -> dict:
        ...

Метод должен содержать логику обработки запроса и вызывать родительский метод BaseChain.form_response(data, body, code, message)

class LoginChain(BaseChain):
    request_type = "login"

    async def get_response_body(self, data):
        # Обработка запроса
        ...
        return super().form_response(data, response, status.HTTP_200_OK, "OK")

data: Изначальное сообщение. body: Обработанное тело запроса. code: HTTP код ответа. message: Сообщение к ответу.

Модели

Доработанные pydantic модели - могут быть использованы как для валидации, так и для генерации сообщения.

Валидация сообщения происходит так же, как и в обычной pydantic модели:

ProcessedMessage(**message_dict)

Для генерации сообщения необходимо создать объект модели, не передавая аргументы при инициализации, и вызвать метод generate. В аргументы метода generate можно передавать любой из ключей структуры сообщения, в том числе code, message, dst и src. Вложенная структура(header, status) формируется сама:

>>> ProcessedMessage().generate(dst="destination", code=201, request_type="creation")
>>> {"header": {"src": "", "dst": "destination"}, "request_type": "creation"...}

RPC брокер

Запуск прослушки и автоматического ответа на сообщения. В файле consumer.py:

from app.chains import your_chain
from rmq_broker.queues.rabbitmq import AsyncRabbitMessageQueue

async def broker_runner():
    async with AsyncRabbitMessageQueue() as provider:
        await provider.register_tasks("rpc_queue_name", your_chain.handle)
        await provider.consume()


asyncio.run(broker_runner())

Далее:

python3 consumer.py