a073ce498d
* feat(video-storage): added zstd library to the machine * feat(video-storage): added new method to pack screenshots into compressed tar arch * feat(video-storage): try to split command into 2 * feat(video-storage): try a new approad to avoid no file error * feat(api): added support to a new pre-signed url for screenshots archive * feat(video-storage): fixed an issue in extension check * feat(video-storage): correct file name * feat(backend): removed video-storage and splitted logic * feat(backend): removed video-storage from helm charts * feat(backend): split canvas and screenshot handlers * feat(canvas): clean up canvas-handler * feat(api): changed mobile replay url (screenshots instead of video) * feat(backend): removed msg.SessID() call * feat(backend): clean up code in imagestorage main
110 lines
2.7 KiB
Go
110 lines
2.7 KiB
Go
package video_maker
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
config "openreplay/backend/internal/config/videostorage"
|
|
"openreplay/backend/pkg/objectstorage"
|
|
)
|
|
|
|
type Task struct {
|
|
path string
|
|
name string
|
|
isBreakTask bool
|
|
}
|
|
|
|
func NewBreakTask() *Task {
|
|
return &Task{isBreakTask: true}
|
|
}
|
|
|
|
type VideoStorage struct {
|
|
cfg *config.Config
|
|
objStorage objectstorage.ObjectStorage
|
|
sendToS3Tasks chan *Task
|
|
workersStopped chan struct{}
|
|
}
|
|
|
|
func New(cfg *config.Config, objStorage objectstorage.ObjectStorage) (*VideoStorage, error) {
|
|
switch {
|
|
case cfg == nil:
|
|
return nil, fmt.Errorf("config is empty")
|
|
case objStorage == nil:
|
|
return nil, fmt.Errorf("object storage is empty")
|
|
}
|
|
newStorage := &VideoStorage{
|
|
cfg: cfg,
|
|
objStorage: objStorage,
|
|
sendToS3Tasks: make(chan *Task, 1),
|
|
workersStopped: make(chan struct{}),
|
|
}
|
|
go newStorage.runWorker()
|
|
return newStorage, nil
|
|
}
|
|
|
|
func (v *VideoStorage) Process(sessID uint64, filesPath, canvasMix string) error {
|
|
name := strings.TrimSuffix(canvasMix, "-list")
|
|
mixList := fmt.Sprintf("%s%s", filesPath, canvasMix)
|
|
// check that mixList exists
|
|
if _, err := os.ReadFile(mixList); err != nil {
|
|
return err
|
|
}
|
|
videoPath := fmt.Sprintf("%s%s.mp4", filesPath, name)
|
|
|
|
// Run ffmpeg to build video
|
|
start := time.Now()
|
|
sessionID := strconv.FormatUint(sessID, 10)
|
|
|
|
cmd := exec.Command("ffmpeg", "-y", "-f", "concat", "-safe", "0", "-i", mixList, "-vf", "pad=ceil(iw/2)*2:ceil(ih/2)*2", "-vsync", "vfr",
|
|
"-pix_fmt", "yuv420p", "-preset", "ultrafast", videoPath)
|
|
|
|
var stdout, stderr bytes.Buffer
|
|
cmd.Stdout = &stdout
|
|
cmd.Stderr = &stderr
|
|
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
log.Printf("Failed to execute command: %v, stderr: %v", err, stderr.String())
|
|
return err
|
|
}
|
|
log.Printf("made video replay in %v", time.Since(start))
|
|
v.sendToS3Tasks <- &Task{path: videoPath, name: sessionID + "/" + name + ".mp4"}
|
|
return nil
|
|
}
|
|
|
|
func (v *VideoStorage) sendToS3(task *Task) {
|
|
start := time.Now()
|
|
video, err := os.ReadFile(task.path)
|
|
if err != nil {
|
|
log.Fatalf("Failed to read video file: %v", err)
|
|
}
|
|
if err := v.objStorage.Upload(bytes.NewReader(video), task.name, "video/mp4", objectstorage.NoCompression); err != nil {
|
|
log.Fatalf("Storage: start uploading replay failed. %s", err)
|
|
}
|
|
log.Printf("Viode file (size: %d) uploaded successfully in %v", len(video), time.Since(start))
|
|
return
|
|
}
|
|
|
|
func (v *VideoStorage) Wait() {
|
|
v.sendToS3Tasks <- NewBreakTask()
|
|
<-v.workersStopped
|
|
}
|
|
|
|
func (v *VideoStorage) runWorker() {
|
|
for {
|
|
select {
|
|
case task := <-v.sendToS3Tasks:
|
|
if task.isBreakTask {
|
|
v.workersStopped <- struct{}{}
|
|
continue
|
|
}
|
|
v.sendToS3(task)
|
|
}
|
|
}
|
|
}
|