From 83186f6e70e3ddb68958593f4b26c3dcdeb44da1 Mon Sep 17 00:00:00 2001
From: lixin <xinli0111@gmail.com>
Date: Tue, 25 Nov 2025 10:48:00 +0800
Subject: [PATCH] try fix taskId Parse

fix taskId Parse when work with runpod  comfyui worker
---
 app/handler/worker_handler.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/app/handler/worker_handler.go b/app/handler/worker_handler.go
index 0a6445d..848c42b 100644
--- a/app/handler/worker_handler.go
+++ b/app/handler/worker_handler.go
@@ -3,6 +3,7 @@ package handler
 import (
 	"fmt"
 	"net/http"
+	"strings"
 
 	"waverless/internal/model"
 	"waverless/internal/service"
@@ -172,6 +173,10 @@ func (h *WorkerHandler) SubmitResult(c *gin.Context) {
 	if req.TaskID == "" {
 		// Try URL path parameter first
 		taskID := c.Param("task_id")
+		//failed to update task result: task not found: 8edae883-00b5-4fb5-b9a2-69f20b57319e&isStream=false
+		if strings.Index(taskID, "&") > 0 {
+			taskID = strings.Split(taskID, "&")[0]
+		}
 		if taskID != "" {
 			req.TaskID = taskID
 			logger.DebugCtx(c.Request.Context(), "got task_id from URL path, task_id: %s", taskID)