hypercorn_config.toml

配置文件修改

applications/routes.py

@@ -2,8 +2,9 @@
 @author: luojunhui
 """
 import time
+import json
 import uuid
-from quart import Blueprint, jsonify, request
+from quart import Blueprint, jsonify, request, websocket
 
 from applications.log import logging
 from applications.process import ProcessParams
@@ -39,10 +40,25 @@ async def post_data():
         port="title_to_video",
         trace_id=trace_id
     )
-    p = ProcessParams(t_id=trace_id)
-    data = await request.get_json()
-    processed_data = p.deal(data)
-    return jsonify(processed_data)
+    p = ProcessParams(t_id="trace_id")  # 确保trace_id是有效的，这里用一个示例值代替
+    while True:
+        # 从WebSocket接收数据
+        data = await websocket.receive()
+
+        # 处理数据，假设收到的数据是JSON字符串
+        try:
+            json_data = json.loads(data)
+            processed_data = p.deal(json_data)
+
+            # 发送处理后的数据回客户端
+            await websocket.send(json.dumps(processed_data))
+        except json.JSONDecodeError:
+            # 处理错误情况，比如数据不是JSON格式
+            await websocket.send(json.dumps({'error': 'Invalid data format'}))
+    # p = ProcessParams(t_id=trace_id)
+    # data = await request.get_json()
+    # processed_data = p.deal(data)
+    # return jsonify(processed_data)
 
 
 @my_blueprint.route('/search_videos', methods=['POST'])