attempting proxying
This commit is contained in:
@@ -2,6 +2,7 @@ import json
|
|||||||
import queue
|
import queue
|
||||||
import uuid
|
import uuid
|
||||||
from typing import Dict, Any
|
from typing import Dict, Any
|
||||||
|
import logging
|
||||||
|
|
||||||
from flask import Response
|
from flask import Response
|
||||||
|
|
||||||
@@ -9,6 +10,7 @@ from events.types.event import Event
|
|||||||
|
|
||||||
# Maps user_id → dict of {connection_id: queue}
|
# Maps user_id → dict of {connection_id: queue}
|
||||||
user_queues: Dict[str, Dict[str, queue.Queue]] = {}
|
user_queues: Dict[str, Dict[str, queue.Queue]] = {}
|
||||||
|
logging.basicConfig(level=logging.INFO)
|
||||||
|
|
||||||
|
|
||||||
def get_queue(user_id: str, connection_id: str) -> queue.Queue:
|
def get_queue(user_id: str, connection_id: str) -> queue.Queue:
|
||||||
@@ -53,6 +55,8 @@ def sse_response_for_user(user_id: str):
|
|||||||
while True:
|
while True:
|
||||||
# Get message from queue (blocks until available)
|
# Get message from queue (blocks until available)
|
||||||
message = user_queue.get()
|
message = user_queue.get()
|
||||||
|
logging.info(
|
||||||
|
f"Sending SSE message to user {user_id} (connection {connection_id}): {message.decode('utf-8').strip()}")
|
||||||
yield message
|
yield message
|
||||||
except GeneratorExit:
|
except GeneratorExit:
|
||||||
# Clean up when client disconnects
|
# Clean up when client disconnects
|
||||||
|
|||||||
Reference in New Issue
Block a user