trustgraph/trustgraph-flow/trustgraph/gateway/dispatch/objects_query.py

from ... schema import ObjectsQueryRequest, ObjectsQueryResponse
from ... messaging import TranslatorRegistry

from . requestor import ServiceRequestor

class ObjectsQueryRequestor(ServiceRequestor):
    def __init__(
            self, backend, request_queue, response_queue, timeout,
            consumer, subscriber,
    ):

        super(ObjectsQueryRequestor, self).__init__(
            backend=backend,
            request_queue=request_queue,
            response_queue=response_queue,
            request_schema=ObjectsQueryRequest,
            response_schema=ObjectsQueryResponse,
            subscription = subscriber,
            consumer_name = consumer,
            timeout=timeout,
        )

        self.request_translator = TranslatorRegistry.get_request_translator("objects-query")
        self.response_translator = TranslatorRegistry.get_response_translator("objects-query")

    def to_request(self, body):
        return self.request_translator.to_pulsar(body)

    def from_response(self, message):
        return self.response_translator.from_response_with_completion(message)