zishuo/goffmpeg
1
0
Fork 0
mirror of https://github.com/xfrr/goffmpeg.git synced 2025-10-05 22:56:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implemented pipe protocol

Browse Source
This commit is contained in:
muhammadharis
2019-11-12 13:04:58 -05:00
parent c49093f2d9
commit f96bd1d964
3 changed files with 175 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
transcoder/transcoder.go
View File

@@ -7,7 +7,6 @@ import (
"errors"
"fmt"
"io"
"os"
"os/exec"
"regexp"
"strconv"
@@ -103,25 +102,47 @@ func (t *Transcoder) InitializeEmptyTranscoder() error {
// SetInputPath sets the input path for transcoding
func (t *Transcoder) SetInputPath(inputPath string) error {
if t.mediafile.InputPipeCommand() != nil {
if t.mediafile.InputPipe() {
return errors.New("cannot set an input path when an input pipe command has been set")
}
t.mediafile.SetInputPath(inputPath)
return nil
}
// CreateInputPipe creates an input pipe for the transcoding process
func (t *Transcoder) CreateInputPipe(cmd *exec.Cmd) error {
if t.mediafile.InputPath() != "" {
return errors.New("cannot set an input pipe when an input path exists")
// SetOutputPath sets the output path for transcoding
func (t *Transcoder) SetOutputPath(inputPath string) error {
if t.mediafile.OutputPipe() {
return errors.New("cannot set an input path when an input pipe command has been set")
}
t.mediafile.SetInputPipeCommand(cmd)
t.mediafile.SetOutputPath(inputPath)
return nil
}
// SetOutputPath sets the output path for transcoding
func (t *Transcoder) SetOutputPath(outputPath string) {
t.mediafile.SetOutputPath(outputPath)
// CreateInputPipe creates an input pipe for the transcoding process
func (t *Transcoder) CreateInputPipe() (*io.PipeWriter, error) {
if t.mediafile.InputPath() != "" {
return nil, errors.New("cannot set an input pipe when an input path exists")
}
inputPipeReader, inputPipeWriter := io.Pipe()
t.mediafile.SetInputPipe(true)
t.mediafile.SetInputPipeReader(inputPipeReader)
t.mediafile.SetInputPipeWriter(inputPipeWriter)
return inputPipeWriter, nil
}
// CreateOutputPipe creates an output pipe for the transcoding process
func (t *Transcoder) CreateOutputPipe(containerFormat string) (*io.PipeReader, error) {
if t.mediafile.OutputPath() != "" {
return nil, errors.New("cannot set an output pipe when an output path exists")
}
t.mediafile.SetOutputFormat(containerFormat)
t.mediafile.SetMovFlags("frag_keyframe")
outputPipeReader, outputPipeWriter := io.Pipe()
t.mediafile.SetOutputPipe(true)
t.mediafile.SetOutputPipeReader(outputPipeReader)
t.mediafile.SetOutputPipeWriter(outputPipeWriter)
return outputPipeReader, nil
}
// Initialize Init the transcoding process
@@ -190,6 +211,7 @@ func (t *Transcoder) Run(progress bool) <-chan error {
}
}
// Set the stdinPipe in case we need to stop the transcoding
stdin, err := proc.StdinPipe()
if nil != err {
fmt.Println("Stdin not available: " + err.Error())
@@ -197,15 +219,20 @@ func (t *Transcoder) Run(progress bool) <-chan error {
t.stdStdinPipe = stdin
// If the user has requested progress, we send it to them on a Buffer
out := &bytes.Buffer{}
if progress {
proc.Stdout = out
}
// If an input pipe has been set, we get the command and set it as stdin for the transcoding
if t.mediafile.InputPipeCommand() != nil {
proc.Stdin, err = t.mediafile.InputPipeCommand().StdoutPipe()
proc.Stdout = os.Stdout
// If an input pipe has been set, we set it as stdin for the transcoding
if t.mediafile.InputPipe() {
proc.Stdin = t.mediafile.InputPipeReader()
}
// If an output pipe has been set, we set it as stdout for the transcoding
if t.mediafile.OutputPipe() {
proc.Stdout = t.mediafile.OutputPipeWriter()
}
err = proc.Start()
@@ -217,16 +244,8 @@ func (t *Transcoder) Run(progress bool) <-chan error {
close(done)
return
}
// Run the pipe-in command if it has been set
if t.mediafile.InputPipeCommand() != nil {
err = t.mediafile.InputPipeCommand().Run()
if err != nil {
err = fmt.Errorf("Failed execution of pipe-in command (%s) with %s", t.mediafile.InputPipeCommand().Args, err)
}
}
err = proc.Wait()
go t.closePipes()
if err != nil {
err = fmt.Errorf("Failed Finish FFMPEG (%s) with %s message %s", command, err, out.String())
}
@@ -344,3 +363,12 @@ func (t Transcoder) Output() <-chan models.Progress {
return out
}
func (t *Transcoder) closePipes() {
if t.mediafile.InputPipe() {
t.mediafile.InputPipeReader().Close()
}
if t.mediafile.OutputPipe() {
t.mediafile.OutputPipeWriter().Close()
}
}