remove async

This commit is contained in:
jxxghp
2023-07-09 11:51:29 +08:00
parent ae10cd2a6b
commit 7f029d7389
15 changed files with 150 additions and 152 deletions

View File

@@ -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")