ghostream/stream/forwarding/forwarding.go

101 lines
2.2 KiB
Go
Raw Normal View History

2020-09-30 13:07:36 +00:00
package forwarding
2020-09-29 19:31:53 +00:00
import (
"bufio"
"io"
"log"
"os/exec"
)
2020-09-30 13:07:36 +00:00
// Options to configure the stream forwarding.
// For each stream name, user can provide several URL to forward stream to
type Options map[string][]string
2020-09-29 19:31:53 +00:00
var (
cfg Options
ffmpegInstances = make(map[string]*exec.Cmd)
ffmpegInputStreams = make(map[string]*io.WriteCloser)
2020-09-29 19:31:53 +00:00
)
// New Load configuration
func New(c Options) {
cfg = c
log.Printf("Stream forwarding initialized")
}
// RegisterStream Declare a new open stream and create ffmpeg instances
func RegisterStream(name string) error {
streams, exist := cfg[name]
if !exist || len(streams) == 0 {
// Nothing to do, not configured
return nil
}
2020-09-29 19:31:53 +00:00
// Launch FFMPEG instance
params := []string{"-re", "-i", "pipe:0"}
for _, stream := range streams {
params = append(params, "-f", "flv", "-preset", "ultrafast", "-tune", "zerolatency",
"-c", "copy", stream)
}
ffmpeg := exec.Command("ffmpeg", params...)
// Open pipes
input, err := ffmpeg.StdinPipe()
if err != nil {
return err
}
output, err := ffmpeg.StdoutPipe()
if err != nil {
return err
}
errOutput, err := ffmpeg.StderrPipe()
if err != nil {
return err
}
2020-09-30 13:10:13 +00:00
ffmpegInstances[name] = ffmpeg
ffmpegInputStreams[name] = &input
2020-09-29 19:31:53 +00:00
// Start FFMpeg
if err := ffmpeg.Start(); err != nil {
return err
2020-09-29 19:31:53 +00:00
}
// Log ffmpeg output
go func() {
scanner := bufio.NewScanner(output)
for scanner.Scan() {
log.Printf("[FFMPEG %s] %s", name, scanner.Text())
}
}()
2020-09-30 13:10:13 +00:00
// Log also error output
go func() {
scanner := bufio.NewScanner(errOutput)
for scanner.Scan() {
log.Printf("[FFMPEG ERR %s] %s", name, scanner.Text())
}
}()
return nil
2020-09-29 19:31:53 +00:00
}
// SendPacket forward data to all FFMpeg instances related to the stream name
2020-09-30 13:10:13 +00:00
func SendPacket(name string, data []byte) {
stdin := ffmpegInputStreams[name]
_, err := (*stdin).Write(data)
if err != nil {
2020-09-30 13:10:13 +00:00
log.Printf("Error while sending a packet to external streaming server for key %s: %s", name, err)
2020-09-29 19:31:53 +00:00
}
}
// CloseConnection When the stream is ended, close FFMPEG instances
func CloseConnection(name string) error {
2020-09-30 13:10:13 +00:00
ffmpeg := ffmpegInstances[name]
if err := ffmpeg.Process.Kill(); err != nil {
return err
2020-09-29 19:31:53 +00:00
}
2020-09-30 13:10:13 +00:00
delete(ffmpegInstances, name)
delete(ffmpegInputStreams, name)
return nil
2020-09-29 19:31:53 +00:00
}