apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 04:31:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Fixing incomplete porting

Browse source
This commit is contained in:
Cyber MacGeddon 2025-12-17 15:58:51 +00:00
parent 57c01665c5
commit 3f98e1e0ba
1 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
trustgraph-flow/trustgraph/rev_gateway/dispatcher.py
View file

@ -26,19 +26,19 @@ class WebSocketResponder:
self.completed = True self.completed = True
class MessageDispatcher: class MessageDispatcher:
def __init__(self, max_workers: int = 10, config_receiver=None, pulsar_client=None): def __init__(self, max_workers: int = 10, config_receiver=None, backend=None):
self.max_workers = max_workers self.max_workers = max_workers
self.semaphore = asyncio.Semaphore(max_workers) self.semaphore = asyncio.Semaphore(max_workers)
self.active_tasks = set() self.active_tasks = set()
self.pulsar_client = pulsar_client self.backend = backend
# Use DispatcherManager for flow and service management # Use DispatcherManager for flow and service management
if pulsar_client and config_receiver: if backend and config_receiver:
self.dispatcher_manager = DispatcherManager(pulsar_client, config_receiver, prefix="rev-gateway") self.dispatcher_manager = DispatcherManager(backend, config_receiver, prefix="rev-gateway")
else: else:
self.dispatcher_manager = None self.dispatcher_manager = None
logger.warning("No pulsar_client or config_receiver provided - using fallback mode") logger.warning("No backend or config_receiver provided - using fallback mode")
# Service name mapping from websocket protocol to translator registry # Service name mapping from websocket protocol to translator registry
self.service_mapping = { self.service_mapping = {
@ -78,7 +78,7 @@ class MessageDispatcher:
try: try:
if not self.dispatcher_manager: if not self.dispatcher_manager:
raise RuntimeError("DispatcherManager not available - pulsar_client and config_receiver required") raise RuntimeError("DispatcherManager not available - backend and config_receiver required")
# Use DispatcherManager for flow-based processing # Use DispatcherManager for flow-based processing
responder = WebSocketResponder() responder = WebSocketResponder()