emmm

main.py

@@ -89,14 +89,14 @@ async def get_tasks():
 
 @app.post("/tasks", response_model=Task)
 async def create_task(task: Task):
-    return task_manager.add(task)
+    return task_manager.add_task(task)
 
 
 '''
 监听任务进度
 可能有多个客户端监听同一个任务(向任务的观察者列表中添加websocket连接)
 可能有多个任务被同一客户端监听(向客户端的观察目标列表中添加任务)
-应检查目标任务是否存在
+应检查目标任务是否存在, 连接断开时应从观察者列表中移除
 '''
 
 
@@ -105,24 +105,19 @@ async def task_endpoint(websocket: WebSocket, task_id: str):
     await websocket.accept()
     if not task_manager.has_task(task_id):
         await websocket.close()
-        print(f"close websocket: {task_id}")
         return
     task_manager.add_observer(task_id, websocket)
     try:
         while True:
             data = await websocket.receive_text()
-            print(f"Client #says: {data}")
+            print(f"收到客户端消息: {data}")
     except WebSocketDisconnect:
         task_manager.remove_observer(task_id, websocket)
         print(f"close websocket: {task_id}")
 
 
-'''
-维护一个任务队列, 任务队列中的任务会被分发给worker节点
-任务状态变化时通知对应的客户端
-'''
-
 # 启动服务
 if __name__ == '__main__':
     port = 8000 if len(sys.argv) < 2 else int(sys.argv[1])
-    uvicorn.run(app='main:app', host='0.0.0.0', port=port, reload=True, workers=1)
+    uvicorn.run(app='main:app', host='0.0.0.0',
+                port=port, reload=True, workers=1)