Fixing API mismatch

This commit is contained in:
Cyber MacGeddon 2025-12-17 12:46:53 +00:00
parent 288b8b61d6
commit caf10ff0c1
13 changed files with 24 additions and 29 deletions

View file

@ -224,11 +224,7 @@ class Configuration:
return ConfigResponse( return ConfigResponse(
version = await self.get_version(), version = await self.get_version(),
value = None,
directory = None,
values = None,
config = config, config = config,
error = None,
) )
async def handle(self, msg): async def handle(self, msg):

View file

@ -211,7 +211,6 @@ class Processor(AsyncProcessor):
type = "config-error", type = "config-error",
message = str(e), message = str(e),
), ),
text=None,
) )
await self.config_response_producer.send( await self.config_response_producer.send(

View file

@ -5,7 +5,7 @@ from ... messaging import TranslatorRegistry
from . requestor import ServiceRequestor from . requestor import ServiceRequestor
class CollectionManagementRequestor(ServiceRequestor): class CollectionManagementRequestor(ServiceRequestor):
def __init__(self, pulsar_client, consumer, subscriber, timeout=120, def __init__(self, backend, consumer, subscriber, timeout=120,
request_queue=None, response_queue=None): request_queue=None, response_queue=None):
if request_queue is None: if request_queue is None:
@ -14,7 +14,7 @@ class CollectionManagementRequestor(ServiceRequestor):
response_queue = collection_response_queue response_queue = collection_response_queue
super(CollectionManagementRequestor, self).__init__( super(CollectionManagementRequestor, self).__init__(
pulsar_client=pulsar_client, backend=backend,
consumer_name = consumer, consumer_name = consumer,
subscription = subscriber, subscription = subscriber,
request_queue=request_queue, request_queue=request_queue,

View file

@ -7,7 +7,7 @@ from ... messaging import TranslatorRegistry
from . requestor import ServiceRequestor from . requestor import ServiceRequestor
class ConfigRequestor(ServiceRequestor): class ConfigRequestor(ServiceRequestor):
def __init__(self, pulsar_client, consumer, subscriber, timeout=120, def __init__(self, backend, consumer, subscriber, timeout=120,
request_queue=None, response_queue=None): request_queue=None, response_queue=None):
if request_queue is None: if request_queue is None:
@ -16,7 +16,7 @@ class ConfigRequestor(ServiceRequestor):
response_queue = config_response_queue response_queue = config_response_queue
super(ConfigRequestor, self).__init__( super(ConfigRequestor, self).__init__(
pulsar_client=pulsar_client, backend=backend,
consumer_name = consumer, consumer_name = consumer,
subscription = subscriber, subscription = subscriber,
request_queue=request_queue, request_queue=request_queue,

View file

@ -11,8 +11,8 @@ logger = logging.getLogger(__name__)
class CoreImport: class CoreImport:
def __init__(self, pulsar_client): def __init__(self, backend):
self.pulsar_client = pulsar_client self.backend = backend
async def process(self, data, error, ok, request): async def process(self, data, error, ok, request):
@ -20,7 +20,7 @@ class CoreImport:
user = request.query["user"] user = request.query["user"]
kr = KnowledgeRequestor( kr = KnowledgeRequestor(
pulsar_client = self.pulsar_client, backend = self.backend,
consumer = "api-gateway-core-import-" + str(uuid.uuid4()), consumer = "api-gateway-core-import-" + str(uuid.uuid4()),
subscriber = "api-gateway-core-import-" + str(uuid.uuid4()), subscriber = "api-gateway-core-import-" + str(uuid.uuid4()),
) )

View file

@ -11,10 +11,10 @@ from . sender import ServiceSender
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class DocumentLoad(ServiceSender): class DocumentLoad(ServiceSender):
def __init__(self, pulsar_client, queue): def __init__(self, backend, queue):
super(DocumentLoad, self).__init__( super(DocumentLoad, self).__init__(
pulsar_client = pulsar_client, backend = backend,
queue = queue, queue = queue,
schema = Document, schema = Document,
) )

View file

@ -7,7 +7,7 @@ from ... messaging import TranslatorRegistry
from . requestor import ServiceRequestor from . requestor import ServiceRequestor
class FlowRequestor(ServiceRequestor): class FlowRequestor(ServiceRequestor):
def __init__(self, pulsar_client, consumer, subscriber, timeout=120, def __init__(self, backend, consumer, subscriber, timeout=120,
request_queue=None, response_queue=None): request_queue=None, response_queue=None):
if request_queue is None: if request_queue is None:
@ -16,7 +16,7 @@ class FlowRequestor(ServiceRequestor):
response_queue = flow_response_queue response_queue = flow_response_queue
super(FlowRequestor, self).__init__( super(FlowRequestor, self).__init__(
pulsar_client=pulsar_client, backend=backend,
consumer_name = consumer, consumer_name = consumer,
subscription = subscriber, subscription = subscriber,
request_queue=request_queue, request_queue=request_queue,

View file

@ -10,7 +10,7 @@ from ... messaging import TranslatorRegistry
from . requestor import ServiceRequestor from . requestor import ServiceRequestor
class KnowledgeRequestor(ServiceRequestor): class KnowledgeRequestor(ServiceRequestor):
def __init__(self, pulsar_client, consumer, subscriber, timeout=120, def __init__(self, backend, consumer, subscriber, timeout=120,
request_queue=None, response_queue=None): request_queue=None, response_queue=None):
if request_queue is None: if request_queue is None:
@ -19,7 +19,7 @@ class KnowledgeRequestor(ServiceRequestor):
response_queue = knowledge_response_queue response_queue = knowledge_response_queue
super(KnowledgeRequestor, self).__init__( super(KnowledgeRequestor, self).__init__(
pulsar_client=pulsar_client, backend=backend,
consumer_name = consumer, consumer_name = consumer,
subscription = subscriber, subscription = subscriber,
request_queue=request_queue, request_queue=request_queue,

View file

@ -9,7 +9,7 @@ from ... messaging import TranslatorRegistry
from . requestor import ServiceRequestor from . requestor import ServiceRequestor
class LibrarianRequestor(ServiceRequestor): class LibrarianRequestor(ServiceRequestor):
def __init__(self, pulsar_client, consumer, subscriber, timeout=120, def __init__(self, backend, consumer, subscriber, timeout=120,
request_queue=None, response_queue=None): request_queue=None, response_queue=None):
if request_queue is None: if request_queue is None:
@ -18,7 +18,7 @@ class LibrarianRequestor(ServiceRequestor):
response_queue = librarian_response_queue response_queue = librarian_response_queue
super(LibrarianRequestor, self).__init__( super(LibrarianRequestor, self).__init__(
pulsar_client=pulsar_client, backend=backend,
consumer_name = consumer, consumer_name = consumer,
subscription = subscriber, subscription = subscriber,
request_queue=request_queue, request_queue=request_queue,

View file

@ -13,7 +13,7 @@ class ServiceRequestor:
def __init__( def __init__(
self, self,
pulsar_client, backend,
request_queue, request_schema, request_queue, request_schema,
response_queue, response_schema, response_queue, response_schema,
subscription="api-gateway", consumer_name="api-gateway", subscription="api-gateway", consumer_name="api-gateway",
@ -21,12 +21,12 @@ class ServiceRequestor:
): ):
self.pub = Publisher( self.pub = Publisher(
pulsar_client, request_queue, backend, request_queue,
schema=request_schema, schema=request_schema,
) )
self.sub = Subscriber( self.sub = Subscriber(
pulsar_client, response_queue, backend, response_queue,
subscription, consumer_name, subscription, consumer_name,
response_schema response_schema
) )

View file

@ -14,12 +14,12 @@ class ServiceSender:
def __init__( def __init__(
self, self,
pulsar_client, backend,
queue, schema, queue, schema,
): ):
self.pub = Publisher( self.pub = Publisher(
pulsar_client, queue, backend, queue,
schema=schema, schema=schema,
) )

View file

@ -13,7 +13,7 @@ class ServiceRequestor:
def __init__( def __init__(
self, self,
pulsar_client, backend,
queue, schema, queue, schema,
handler, handler,
subscription="api-gateway", consumer_name="api-gateway", subscription="api-gateway", consumer_name="api-gateway",
@ -21,7 +21,7 @@ class ServiceRequestor:
): ):
self.sub = Subscriber( self.sub = Subscriber(
pulsar_client, queue, backend, queue,
subscription, consumer_name, subscription, consumer_name,
schema schema
) )

View file

@ -11,10 +11,10 @@ from . sender import ServiceSender
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class TextLoad(ServiceSender): class TextLoad(ServiceSender):
def __init__(self, pulsar_client, queue): def __init__(self, backend, queue):
super(TextLoad, self).__init__( super(TextLoad, self).__init__(
pulsar_client = pulsar_client, backend = backend,
queue = queue, queue = queue,
schema = TextDocument, schema = TextDocument,
) )