fix: 修复请求头转发与任务状态流转问题

移除 util.ForwardHeaders，改为从原始请求精确提取 Authorization 或全部请求头；
任务创建时直接设为 Running 状态，避免二次更新与查询；
模型调用使用独立超时上下文，防止外层取消影响回调；
增加 OSS 上传耗时日志，调整数据库连接池参数。

service/task/task_service.go

@@ -65,20 +65,20 @@ func (s *taskService) Create(ctx context.Context, req *dto.CreateTaskReq) (res *
 		Body:    req.RequestPayload,
 		Headers: util.ParseHeadMsgHeaders(model.HeadMsg),
 	}
-	id, err := dao.ModelGatewayTask.Insert(ctx, &entity.ModelGatewayTask{
-		ModelName:      req.ModelName,
-		TaskID:         taskID,
-		State:          public.TaskStatusPending,
-		BizName:        req.BizName,
-		CallbackURL:    req.CallbackUrl,
-		RequestPayload: &requestPayload,
-		EpicycleId:     req.EpicycleId,
-	})
+	task := new(entity.ModelGatewayTask)
+	task.ModelName = model.ModelName
+	task.TaskID = taskID
+	task.State = public.TaskStatusRunning
+	task.BizName = req.BizName
+	task.CallbackURL = req.CallbackUrl
+	task.RequestPayload = &requestPayload
+	task.EpicycleId = req.EpicycleId
+	id, err := dao.ModelGatewayTask.Insert(ctx, task)
 	if err != nil { // 入库失败：回滚闸门占位
 		queue.ReleaseQueueSlot(ctx, req.ModelName, taskID)
 		return nil, err
 	}
-
+	task.Id = id
 	// 4) 写操作日志（不影响主流程，失败忽略）
 	ip := ""
 	ua := ""
@@ -107,25 +107,25 @@ func (s *taskService) Create(ctx context.Context, req *dto.CreateTaskReq) (res *
 		},
 	})
 
-	// 5) 抢占任务：改为执行中
-	rows, err := dao.ModelGatewayTask.Update(ctx, &entity.ModelGatewayTask{
-		SQLBaseDO: beans.SQLBaseDO{Id: id},
-		State:     public.TaskStatusRunning,
-	})
-	if err != nil {
-		return nil, err
-	}
-	if rows == 0 {
-		return nil, fmt.Errorf("任务不存在: id=%d", id)
-	}
+	//// 5) 抢占任务：改为执行中
+	//rows, err := dao.ModelGatewayTask.Update(ctx, &entity.ModelGatewayTask{
+	//	SQLBaseDO: beans.SQLBaseDO{Id: id},
+	//	State:     public.TaskStatusRunning,
+	//})
+	//if err != nil {
+	//	return nil, err
+	//}
+	//if rows == 0 {
+	//	return nil, fmt.Errorf("任务不存在: id=%d", id)
+	//}
 
 	// 6) 查询任务信息
-	task, err := dao.ModelGatewayTask.Get(ctx, &entity.ModelGatewayTask{
-		SQLBaseDO: beans.SQLBaseDO{Id: id},
-	})
-	if err != nil {
-		return nil, err
-	}
+	//task, err := dao.ModelGatewayTask.Get(ctx, &entity.ModelGatewayTask{
+	//	SQLBaseDO: beans.SQLBaseDO{Id: id},
+	//})
+	//if err != nil {
+	//	return nil, err
+	//}
 
 	// 7) 创建成功后立即异步尝试执行当前任务
 	go AsyncWorker.handleOne(util.AsyncCtx(ctx), task, model, req)

service/task/worker.go

@@ -129,10 +129,14 @@ func (w *asyncWorker) handleOne(ctx context.Context, task *entity.ModelGatewayTa
 		if attempt > 0 {
 			g.Log().Infof(ctx, "[执行任务][重试] OSS上传 第%d/%d次 taskId=%s", attempt, maxRetry, task.TaskID)
 		}
+		startUpload := time.Now()
 		oss, err = gateway.UploadByTask(ctx, gjson.New(result).MustToJson(), "json")
 		if err == nil {
 			break
 		}
+		cost := time.Since(startUpload)
+		g.Log().Infof(ctx, "本次上传耗时:%s", cost)
+
 		g.Log().Errorf(ctx, "[执行任务][失败] OSS上传失败 taskId=%s attempt=%d/%d err=%v", task.TaskID, attempt, maxRetry, err)
 		if attempt == maxRetry {
 			task.State = public.TaskStatusFailed
@@ -161,9 +165,9 @@ func (w *asyncWorker) handleOne(ctx context.Context, task *entity.ModelGatewayTa
 	}
 
 	queue.ReleaseQueueSlot(ctx, task.ModelName, task.TaskID)
-	go gateway.TriggerCallback(context.WithoutCancel(ctx), task)
+	go gateway.TriggerCallback(util.AsyncCtx(ctx), task)
 	if req.EpicycleId != 0 {
-		go gateway.TriggerPromptsCallback(context.WithoutCancel(ctx), task, req.EpicycleId)
+		go gateway.TriggerPromptsCallback(util.AsyncCtx(ctx), task, req.EpicycleId)
 	}
 
 	g.Log().Infof(ctx, "[执行任务][成功] taskId=%s duration=%ds fileType=%s",
@@ -420,7 +424,7 @@ func injectErrorMessage(payload map[string]any, err error) map[string]any {
 // modelKey 用于覆盖/补充模型配置 head_msg（例如每次请求携带不同的 X-API-Key）
 func InvokeModel(ctx context.Context, model *entity.ModelGatewayModel, body map[string]any) ([]byte, error) {
 	// 1) 记录模型调用次数
-	_ = dao.ModelGatewayLogsStat.IncRequestCount(ctx, time.Now(), model.TenantId, model.Creator, model.ModelName)
+	//_ = dao.ModelGatewayLogsStat.IncRequestCount(ctx, time.Now(), model.TenantId, model.Creator, model.ModelName)
 
 	// 2）请求参数映射：将标准 payload 按模型配置的 requestMapping 转为模型需要的格式
 	//—— 请求映射实际处理为提示词构建请求，因为有附加字段及其他字段的拼接。这里不方便做请求映射
@@ -447,13 +451,20 @@ func InvokeModel(ctx context.Context, model *entity.ModelGatewayModel, body map[
 				baseURL = baseURL + "?" + q.Encode()
 			}
 		}
-		req, err = http.NewRequestWithContext(ctx, http.MethodGet, baseURL, nil)
+		// 改用独立超时ctx，隔绝外层截止
+		reqCtx, reqCancel := context.WithTimeout(context.Background(), timeout)
+		defer reqCancel()
+		req, err = http.NewRequestWithContext(reqCtx, http.MethodGet, baseURL, nil)
+		//req, err = http.NewRequestWithContext(ctx, http.MethodGet, baseURL, nil)
 	default:
 		bodyBytes, err := json.Marshal(body)
 		if err != nil {
 			return nil, err
 		}
-		req, err = http.NewRequestWithContext(ctx, http.MethodPost, baseURL, bytes.NewReader(bodyBytes))
+		reqCtx, reqCancel := context.WithTimeout(context.Background(), timeout)
+		defer reqCancel()
+		req, err = http.NewRequestWithContext(reqCtx, http.MethodPost, baseURL, bytes.NewReader(bodyBytes))
+		//req, err = http.NewRequestWithContext(ctx, http.MethodPost, baseURL, bytes.NewReader(bodyBytes))
 	}
 
 	// 5）注入请求头：先模型静态配置，再动态 modelKey（后者可覆盖前者）
@@ -552,5 +563,5 @@ func (w *asyncWorker) failTask(ctx context.Context, t *entity.ModelGatewayTask,
 		g.Log().Warningf(ctx, "[执行任务][更新数据库失败] taskId=%s err=%v", t.TaskID, err)
 	}
 	queue.ReleaseQueueSlot(ctx, t.ModelName, t.TaskID)
-	go gateway.TriggerCallback(context.WithoutCancel(ctx), t)
+	go gateway.TriggerCallback(util.AsyncCtx(ctx), t)
 }