视频剪辑上传

service/asr/task_service.go

@@ -19,6 +19,7 @@ import (
 	serviceScene "media/service/scene"
 
 	"gitea.com/red-future/common/beans"
+	"gitea.com/red-future/common/utils"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/util/gconv"
 	"github.com/gogf/gf/v2/util/guid"
@@ -79,16 +80,19 @@ func (s *audioTaskService) Create(ctx context.Context, params *CreateTaskParams)
 	g.Log().Infof(ctx, "[创建任务 %s] 文件数=%d, 模型=%s, 语言=%s, 回调=%s",
 		taskID, len(params.InputData), params.Model, params.Language, params.CallbackURL)
 
+	// 提取调用方用户信息，传给 goroutine
+	user := getUserFromCtx(ctx)
+
 	// 异步处理
-	go s.processTask(taskID, params.InputData, params.Model, params.Language, params.Threshold, params.CallbackURL)
+	go s.processTask(user, taskID, params.InputData, params.Model, params.Language, params.Threshold, params.CallbackURL)
 
 	return &dto.CreateTaskRes{TaskID: taskID}, nil
 }
 
 // processTask 异步处理所有URL，每个文件生成一条明细
-func (s *audioTaskService) processTask(taskID string, urls []string, model, language string, threshold float64, callbackURL string) {
+func (s *audioTaskService) processTask(user *beans.User, taskID string, urls []string, model, language string, threshold float64, callbackURL string) {
 	ctx := context.Background()
-	ctx = context.WithValue(ctx, "user", &beans.User{UserName: "admin", TenantId: 1})
+	ctx = context.WithValue(ctx, "user", user)
 
 	defer func() {
 		if r := recover(); r != nil {
@@ -231,7 +235,8 @@ func (s *audioTaskService) callback(ctx context.Context, taskID, status, errMsg,
 	g.Log().Debugf(ctx, "[回调 %s] 回调载荷长度=%d字节, 明细条数=%d",
 		taskID, len(body), len(detailItems))
 	// 透传调用方的用户信息，供回调方 GetUserInfo 从 X-User-Info 头获取
-	userJSON, _ := json.Marshal(beans.User{UserName: "admin", TenantId: 1})
+	cbUser := getUserFromCtx(ctx)
+	userJSON, _ := json.Marshal(cbUser)
 	g.Log().Infof(ctx, "[回调 %s] curl -X POST '%s' -H 'Content-Type: application/json' -H 'X-User-Info: %s' -d '%s'",
 		taskID, callbackURL, string(userJSON), strings.ReplaceAll(string(body), "'", "'\\''"))
 
@@ -316,6 +321,21 @@ func (s *audioTaskService) processSingleVideo(ctx context.Context, taskID, saveP
 	}
 }
 
+// getUserFromCtx 从 context 提取用户信息，没有则返回默认 admin
+func getUserFromCtx(ctx context.Context) *beans.User {
+	if u := ctx.Value("user"); u != nil {
+		if user, ok := u.(*beans.User); ok {
+			return user
+		}
+	}
+	// 尝试用 common 库解析
+	user, err := utils.GetUserInfo(ctx)
+	if err == nil && user != nil {
+		return user
+	}
+	return &beans.User{UserName: "admin", TenantId: 1}
+}
+
 // saveDetail 保存单文件明细到 transcribe_task_detail
 func (s *audioTaskService) saveDetail(ctx context.Context, taskID string, fileIndex int, fileName, text, scenes string, audioSize int64, audioDuration, model, language, errMsg string) {
 	detail := &entity.TranscribeTaskDetail{