rogee/mp-qvyun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add buffer to utils

Browse Source
This commit is contained in:
Rogee
2024-12-17 09:50:28 +08:00
parent 7dd4d76a71
commit be19daf5fe
2 changed files with 80 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
backend/modules/commands/discover/discover_medias.go
View File

@@ -1,6 +1,7 @@
package discover
import (
"bytes"
"crypto/md5"
"fmt"
"io"
@@ -13,6 +14,7 @@ import (
"backend/modules/medias"
"backend/pkg/media_store"
"backend/pkg/path"
"backend/pkg/utils"
"github.com/pkg/errors"
log "github.com/sirupsen/logrus"
@@ -61,6 +63,12 @@ func (d *DiscoverMedias) RunE(from, to string) error {
return errors.Wrapf(err, "get file md5: %s", video)
}
name := filepath.Base(video)[0:strings.LastIndex(filepath.Base(video), ".")]
d.log.Infof("start process video: %s, md5: %s, name: %s", video, md5, name)
if name == "" {
d.log.Errorf("video name is empty: %s", video)
continue
}
if info, ok := store.HashExists(md5); ok {
info.Name = name
@@ -68,6 +76,7 @@ func (d *DiscoverMedias) RunE(from, to string) error {
if err := store.Save(to); err != nil {
return errors.Wrapf(err, "save store: %s", to)
}
d.log.Infof("video exists: %s", video)
continue
}
@@ -156,11 +165,18 @@ func (d *DiscoverMedias) extractAudio(video string, output string) error {
}
d.log.Infof("cmd: ffmpeg %s", strings.Join(args, " "))
logs, err := exec.Command("ffmpeg", args...).CombinedOutput()
if err != nil {
return errors.Wrapf(err, "extract audio: %s", video)
cmd := exec.Command("ffmpeg", args...)
var buf bytes.Buffer
logWriter := utils.NewLogBuffer(func(line string) {
d.log.Info(line)
})
cmd.Stdout = utils.NewCombinedBuffer(&buf, logWriter)
cmd.Stderr = utils.NewCombinedBuffer(&buf, logWriter)
if err := cmd.Run(); err != nil {
return errors.Wrapf(err, "extract audio failed: %s\n%s", video, buf.String())
}
d.log.Infof("extract audio: %s", logs)
return nil
}
@@ -206,12 +222,19 @@ func (d *DiscoverMedias) ffmpegVideoToM3U8(input, output, hash string) error {
}
log.Infof("cmd: ffmpeg %s", strings.Join(args, " "))
logs, err := exec.Command("ffmpeg", args...).CombinedOutput()
if err != nil {
return errors.Wrapf(err, "ffmpeg video to m3u8: %s", input)
cmd := exec.Command("ffmpeg", args...)
var buf bytes.Buffer
logWriter := utils.NewLogBuffer(func(line string) {
d.log.Info(line)
})
cmd.Stdout = utils.NewCombinedBuffer(&buf, logWriter)
cmd.Stderr = utils.NewCombinedBuffer(&buf, logWriter)
if err := cmd.Run(); err != nil {
return errors.Wrapf(err, "ffmpeg video to m3u8 failed: %s\n%s", input, buf.String())
}
d.log.Infof("ffmpeg video to m3u8: %s", logs)
return nil
}
@@ -234,12 +257,19 @@ func (d *DiscoverMedias) ffmpegAudioToM3U8(input, output, hash string) error {
}
log.Infof("cmd: ffmpeg %s", strings.Join(args, " "))
logs, err := exec.Command("ffmpeg", args...).CombinedOutput()
if err != nil {
return errors.Wrapf(err, "ffmpeg audio to m3u8: %s", input)
cmd := exec.Command("ffmpeg", args...)
var buf bytes.Buffer
logWriter := utils.NewLogBuffer(func(line string) {
d.log.Info(line)
})
cmd.Stdout = utils.NewCombinedBuffer(&buf, logWriter)
cmd.Stderr = utils.NewCombinedBuffer(&buf, logWriter)
if err := cmd.Run(); err != nil {
return errors.Wrapf(err, "ffmpeg audio to m3u8 failed: %s\n%s", input, buf.String())
}
d.log.Infof("ffmpeg audio to m3u8: %s", logs)
return nil
}
@@ -274,8 +304,8 @@ func (d *DiscoverMedias) runCleanup(to string) {
// getSnapshot get the snapshot of target seconds of a video
func (d *DiscoverMedias) ffmpegVideoToPoster(video string, output string) error {
// ffmpeg -i input_video.mp4 -ss N -vframes 1 -vf "scale=width:height" output_image.jpg
args := []string{
"-y", // 添加 -y 参数强制覆盖
"-i", video,
"-ss", "00:00:01",
"-vframes", "1",
@@ -284,12 +314,19 @@ func (d *DiscoverMedias) ffmpegVideoToPoster(video string, output string) error
}
d.log.Infof("cmd: ffmpeg %s", strings.Join(args, " "))
logs, err := exec.Command("ffmpeg", args...).CombinedOutput()
if err != nil {
return errors.Wrapf(err, "get snapshot: %s", video)
cmd := exec.Command("ffmpeg", args...)
var buf bytes.Buffer
logWriter := utils.NewLogBuffer(func(line string) {
d.log.Info(line)
})
cmd.Stdout = utils.NewCombinedBuffer(&buf, logWriter)
cmd.Stderr = utils.NewCombinedBuffer(&buf, logWriter)
if err := cmd.Run(); err != nil {
return errors.Wrapf(err, "get snapshot failed: %s\n%s", video, buf.String())
}
d.log.Infof("get snapshot: %s", logs)
return nil
}