40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
package video
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
dto "media/model/dto/video"
|
|
service "media/service/video"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
type caption struct{}
|
|
|
|
var Caption = new(caption)
|
|
|
|
// CreateCaption 创建字幕叠加任务 POST /video/caption
|
|
func (c *caption) CreateCaption(ctx context.Context, req *dto.CreateCaptionTaskReq) (res *dto.CreateCaptionTaskRes, err error) {
|
|
ctx = withUser(ctx)
|
|
g.Log().Infof(ctx, "[字幕叠加] 收到请求 入参: video_count=%d, element_count=%d, callback=%s",
|
|
len(req.VideoURLs), len(req.Elements), req.CallbackURL)
|
|
|
|
if len(req.VideoURLs) < 1 {
|
|
return nil, fmt.Errorf("至少需要1个视频")
|
|
}
|
|
|
|
taskID, taskErr := service.Caption.CreateAsyncTask(ctx, req.VideoURLs, req.AudioURL, req.Subtitles, req.SubtitleStyle, req.Elements, req.CallbackURL)
|
|
if taskErr != nil {
|
|
return nil, taskErr
|
|
}
|
|
|
|
return &dto.CreateCaptionTaskRes{TaskID: taskID}, nil
|
|
}
|
|
|
|
// GetCaptionTask 查询字幕任务结果 GET /video/caption/{taskId}
|
|
func (c *caption) GetCaptionTask(ctx context.Context, req *dto.GetCaptionTaskReq) (res *dto.GetCaptionTaskRes, err error) {
|
|
ctx = withUser(ctx)
|
|
return service.Caption.GetTaskResult(ctx, req.TaskID)
|
|
}
|