70 lines
2.4 KiB
Go
70 lines
2.4 KiB
Go
package app
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"errors"
|
||
"strings"
|
||
|
||
"yingce/backend/internal/model"
|
||
)
|
||
|
||
// processTask 是任务执行阶段唯一的类型分派入口。
|
||
// 任务已经在 CreateTask 阶段完成 admission；这里不把无效视频任务降级成内部工作流成功。
|
||
func (s *Service) processTask(ctx context.Context, task model.Task) (map[string]interface{}, []map[string]interface{}, error) {
|
||
if err := validateTaskType(task.Type); err != nil {
|
||
return nil, nil, err
|
||
}
|
||
decryptedInput, err := s.decryptTaskInputJSON(task.InputJSON)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
task.InputJSON = decryptedInput
|
||
ctx = withTaskExecutionID(ctx, task.ID)
|
||
ctx = withProviderAnalytics(ctx, s, task)
|
||
ctx = context.WithValue(ctx, mediaExecutionTaskKey{}, task)
|
||
|
||
if task.Type == "canvas_text" || task.Type == "canvas_image" || task.Type == "canvas_video" || task.Type == "canvas_audio" {
|
||
result, err := s.processCanvasGenerationTask(ctx, task.UserID, task.ProjectID, task.Type, task.Prompt, task.InputJSON)
|
||
return result, nil, err
|
||
}
|
||
if strings.HasPrefix(task.Type, "video_") {
|
||
if !canRunProviderTask(task) {
|
||
return nil, nil, errors.New("视频任务缺少可执行的模型配置")
|
||
}
|
||
result, err := s.processCanvasGenerationTask(ctx, task.UserID, task.ProjectID, task.Type, task.Prompt, task.InputJSON)
|
||
return result, nil, err
|
||
}
|
||
return nil, nil, errors.New("任务类型没有可用的执行分支")
|
||
}
|
||
|
||
func canRunProviderTask(task model.Task) bool {
|
||
if !strings.HasPrefix(task.Type, "video_") || strings.TrimSpace(task.InputJSON) == "" {
|
||
return false
|
||
}
|
||
var input map[string]any
|
||
if err := json.Unmarshal([]byte(task.InputJSON), &input); err != nil {
|
||
return false
|
||
}
|
||
return hasExecutableProviderVideoConfig(input)
|
||
}
|
||
|
||
func hasExecutableProviderVideoConfig(input map[string]any) bool {
|
||
mode, _ := input["mode"].(string)
|
||
config, ok := input["config"].(map[string]any)
|
||
if mode != "video" || !ok {
|
||
return false
|
||
}
|
||
interfaceType := stringValue(config["interfaceType"])
|
||
if isRunningHubInterface(interfaceType) {
|
||
if stringValue(config["workflowId"]) == "" && stringValue(config["webappId"]) == "" && stringValue(config["model"]) == "" {
|
||
return false
|
||
}
|
||
return stringValue(config["baseUrl"]) != "" && stringValue(config["apiKey"]) != ""
|
||
}
|
||
if stringValue(config["model"]) == "" {
|
||
return false
|
||
}
|
||
return stringValue(config["channelId"]) != "" || (stringValue(config["baseUrl"]) != "" && stringValue(config["apiKey"]) != "")
|
||
}
|