mirror of
https://github.com/borbann-platform/backend-api.git
synced 2025-12-18 20:24:05 +01:00
27 lines
901 B
Python
27 lines
901 B
Python
from fastapi import APIRouter, Request, HTTPException
|
|
from fastapi.responses import StreamingResponse
|
|
from log.logging_utils import RUN_LOG_QUEUES
|
|
from queue import Empty
|
|
import asyncio
|
|
|
|
router = APIRouter()
|
|
|
|
@router.get("/pipelines/{pipeline_id}/runs/{run_id}/logs/stream")
|
|
async def stream_logs(request: Request, pipeline_id: str, run_id: str):
|
|
log_queue = RUN_LOG_QUEUES.get(run_id)
|
|
if not log_queue:
|
|
raise HTTPException(status_code=404, detail="No logs for this run.")
|
|
|
|
async def event_generator():
|
|
while True:
|
|
if await request.is_disconnected():
|
|
break
|
|
try:
|
|
log_line = log_queue.get(timeout=1)
|
|
yield f"data: {log_line}\n\n"
|
|
except Empty:
|
|
await asyncio.sleep(0.2)
|
|
continue
|
|
|
|
return StreamingResponse(event_generator(), media_type="text/event-stream")
|