remove async
This commit is contained in:
@@ -1,11 +1,10 @@
|
||||
import asyncio
|
||||
import json
|
||||
import json
|
||||
import time
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from fastapi import APIRouter, HTTPException
|
||||
from fastapi.responses import StreamingResponse
|
||||
|
||||
from app import schemas
|
||||
from app.core.config import settings
|
||||
from app.core.security import verify_token
|
||||
from app.helper.message import MessageHelper
|
||||
from app.helper.progress import ProgressHelper
|
||||
@@ -14,7 +13,7 @@ router = APIRouter()
|
||||
|
||||
|
||||
@router.get("/progress/{process_type}", summary="实时进度")
|
||||
async def get_progress(process_type: str, token: str):
|
||||
def get_progress(process_type: str, token: str):
|
||||
"""
|
||||
实时获取处理进度,返回格式为SSE
|
||||
"""
|
||||
@@ -26,17 +25,17 @@ async def get_progress(process_type: str, token: str):
|
||||
|
||||
progress = ProgressHelper()
|
||||
|
||||
async def event_generator():
|
||||
def event_generator():
|
||||
while True:
|
||||
detail = progress.get(process_type)
|
||||
yield 'data: %s\n\n' % json.dumps(detail)
|
||||
await asyncio.sleep(0.2)
|
||||
time.sleep(0.2)
|
||||
|
||||
return StreamingResponse(event_generator(), media_type="text/event-stream")
|
||||
|
||||
|
||||
@router.get("/message", summary="实时消息")
|
||||
async def get_progress(token: str):
|
||||
def get_progress(token: str):
|
||||
"""
|
||||
实时获取系统消息,返回格式为SSE
|
||||
"""
|
||||
@@ -48,10 +47,10 @@ async def get_progress(token: str):
|
||||
|
||||
message = MessageHelper()
|
||||
|
||||
async def event_generator():
|
||||
def event_generator():
|
||||
while True:
|
||||
detail = message.get()
|
||||
yield 'data: %s\n\n' % (detail or '')
|
||||
await asyncio.sleep(3)
|
||||
time.sleep(3)
|
||||
|
||||
return StreamingResponse(event_generator(), media_type="text/event-stream")
|
||||
|
Reference in New Issue
Block a user