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

from ... schema import RowsQueryRequest, RowsQueryResponse
from ... messaging import TranslatorRegistry

from . requestor import ServiceRequestor

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

        super(RowsQueryRequestor, self).__init__(
            backend=backend,
            request_queue=request_queue,
            response_queue=response_queue,
            request_schema=RowsQueryRequest,
            response_schema=RowsQueryResponse,
            subscription = subscriber,
            consumer_name = consumer,
            timeout=timeout,
        )

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

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

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