93 lines
2.6 KiB
Go
93 lines
2.6 KiB
Go
package video
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"os"
|
||
|
||
dto "media/model/dto/video"
|
||
service "media/service/video"
|
||
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
)
|
||
|
||
type cut struct{}
|
||
|
||
var Cut = new(cut)
|
||
|
||
// Cut 视频分镜剪切（URL模式） POST /video/cut
|
||
func (c *cut) Cut(ctx context.Context, req *dto.CutReq) (res *dto.CutRes, err error) {
|
||
ctx = withUser(ctx)
|
||
g.Log().Infof(ctx, "[视频分镜剪切] 收到请求 入参: video_url=%s, total_scenes=%d, upload=%v",
|
||
req.VideoURL, len(req.Scenes), req.Upload)
|
||
|
||
// 下载视频到临时目录
|
||
tempDir := getTempDir(ctx)
|
||
os.MkdirAll(tempDir, 0755)
|
||
|
||
savePath, err := downloadFromURL(ctx, req.VideoURL, tempDir)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("下载视频失败: %v", err)
|
||
}
|
||
defer os.Remove(savePath)
|
||
|
||
// 转换分镜为 service 层类型（解析时间字符串为秒）
|
||
serviceShots, err := service.ConvertScenes(req.Scenes)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("解析分镜失败: %v", err)
|
||
}
|
||
if len(serviceShots) == 0 {
|
||
return nil, fmt.Errorf("没有有效的分镜片段")
|
||
}
|
||
|
||
// 调用服务层
|
||
svcRes, err := service.Cut.Cut(ctx, &service.CutReq{
|
||
VideoPath: savePath,
|
||
Shots: serviceShots,
|
||
Upload: req.Upload,
|
||
})
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
defer os.Remove(svcRes.OutputPath)
|
||
|
||
return &dto.CutRes{
|
||
OutputPath: svcRes.OutputPath,
|
||
FileSize: svcRes.FileSize,
|
||
Duration: svcRes.Duration,
|
||
DurationStr: svcRes.DurationStr,
|
||
ShotsCount: svcRes.ShotsCount,
|
||
FileURL: svcRes.FileURL,
|
||
}, nil
|
||
}
|
||
|
||
// CutAsync 视频分镜剪切-异步（URL模式） POST /video/cut/async
|
||
func (c *cut) CutAsync(ctx context.Context, req *dto.CutAsyncReq) (res *dto.CreateCutTaskRes, err error) {
|
||
ctx = withUser(ctx)
|
||
g.Log().Infof(ctx, "[视频分镜剪切-异步] 收到请求 入参: video_url=%s, total_scenes=%d, upload=%v, callback=%s",
|
||
req.VideoURL, len(req.Scenes), req.Upload, req.CallbackURL)
|
||
|
||
// 转换分镜为 service 层类型（解析时间字符串为秒）
|
||
serviceShots, err := service.ConvertScenes(req.Scenes)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("解析分镜失败: %v", err)
|
||
}
|
||
if len(serviceShots) == 0 {
|
||
return nil, fmt.Errorf("没有有效的分镜片段")
|
||
}
|
||
|
||
taskID, taskErr := service.Cut.CreateAsyncTask(ctx, req.VideoURL, serviceShots, req.Upload, req.CallbackURL)
|
||
if taskErr != nil {
|
||
return nil, taskErr
|
||
}
|
||
|
||
return &dto.CreateCutTaskRes{TaskID: taskID}, nil
|
||
}
|
||
|
||
// GetCutTask 查询异步剪切任务结果 GET /video/cut/task/{taskId}
|
||
func (c *cut) GetCutTask(ctx context.Context, req *dto.GetCutTaskReq) (res *dto.GetCutTaskRes, err error) {
|
||
ctx = withUser(ctx)
|
||
return service.Cut.GetTaskResult(ctx, req.TaskID)
|
||
}
|