Selaa lähdekoodia

fix(decode_task): 修复获取解码结果时未返回错误原因的问题

修改SQL查询条件,移除task_status限制并返回error_reason字段
调整接口返回数据结构,包含错误原因信息
max_liu 3 päivää sitten
vanhempi
commit
6f0c890336
2 muutettua tiedostoa jossa 15 lisäystä ja 8 poistoa
  1. 2 2
      decode_task/decodeTask.py
  2. 13 6
      main.py

+ 2 - 2
decode_task/decodeTask.py

@@ -29,12 +29,12 @@ def invoke_decode_workflow(task_params: Dict[str, Any]):
 
 
 def get_decode_result_by_id(task_id:str):
-    sql = "SELECT * FROM decode_videos WHERE task_id = %s AND task_status = 2 "
+    sql = "SELECT * FROM decode_videos WHERE task_id = %s"
     tasks = mysql.fetchone(sql, (task_id,))
     if not tasks:
         logger.info(f"task_id = {task_id} , 任务不存在")
         return None
-    return tasks['result'], tasks['status']
+    return tasks['result'], tasks['status'],tasks['error_reason']
 
 
 def  decode_task_status_handler():

+ 13 - 6
main.py

@@ -8,7 +8,7 @@ from utils.params import TaskStatusParam, DecodeListParam
 from dotenv import load_dotenv
 
 
-from decode_task.decodeTask import get_decode_result_by_id
+from decode_task.decodeTask import get_decode_result_by_id as get_decode_result_by_id_db
 
 
 
@@ -73,15 +73,23 @@ def decode_video(param:DecodeListParam):
 
 
 @app.post("/decode/result")
-def get_decode_result_by_id(param: TaskStatusParam):
-    result, status = get_decode_result_by_id(param.task_id)
+def get_decode_result(param: TaskStatusParam):
+    db_res = get_decode_result_by_id_db(param.task_id)
+    if not db_res:
+        return {
+            "code": -1,
+            "message": "error",
+            "data": None
+        }
+    result, status,error_reason = db_res
     if result:
         return {
             "code": 0,
             "message": "success",
             "data": {
-                "decode_result": result,
-                "task_status": status
+                "result": result,
+                "status": status,
+                "error":error_reason
             }
         }
     else:
@@ -91,4 +99,3 @@ def get_decode_result_by_id(param: TaskStatusParam):
             "data": None
         }
 
-