Better error handling in stream forwarder

stream/forwarding/forwarding.go

@@ -12,56 +12,59 @@ import (
 type Options map[string][]string
 
 var (
-	options            Options
+	cfg                Options
 	ffmpegInstances    = make(map[string]*exec.Cmd)
 	ffmpegInputStreams = make(map[string]*io.WriteCloser)
 )
 
 // New Load configuration
-func New(cfg *Options) error {
-	options = *cfg
-	return nil
+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) {
-	if len(options[name]) == 0 {
-		return
+func RegisterStream(name string) error {
+	streams, exist := cfg[name]
+	if !exist || len(streams) == 0 {
+		// Nothing to do, not configured
+		return nil
 	}
 
+	// Launch FFMPEG instance
 	params := []string{"-re", "-i", "pipe:0"}
-	for _, stream := range options[name] {
+	for _, stream := range streams {
 		params = append(params, "-f", "flv", "-preset", "ultrafast", "-tune", "zerolatency",
 			"-c", "copy", stream)
 	}
-	// Launch FFMPEG instance
 	ffmpeg := exec.Command("ffmpeg", params...)
 
 	// Open pipes
 	input, err := ffmpeg.StdinPipe()
 	if err != nil {
-		panic(err)
+		return err
 	}
 	output, err := ffmpeg.StdoutPipe()
 	if err != nil {
-		panic(err)
+		return err
 	}
 	errOutput, err := ffmpeg.StderrPipe()
 	if err != nil {
-		panic(err)
+		return err
 	}
 	ffmpegInstances[name] = ffmpeg
 	ffmpegInputStreams[name] = &input
 
+	// Start FFMpeg
 	if err := ffmpeg.Start(); err != nil {
-		panic(err)
+		return err
 	}
 
 	// Log ffmpeg output
 	go func() {
 		scanner := bufio.NewScanner(output)
 		for scanner.Scan() {
-			log.Println("[FFMPEG " + name + "] " + scanner.Text())
+			log.Printf("[FFMPEG %s] %s", name, scanner.Text())
 		}
 	}()
 
@@ -69,12 +72,14 @@ func RegisterStream(name string) {
 	go func() {
 		scanner := bufio.NewScanner(errOutput)
 		for scanner.Scan() {
-			log.Println("[FFMPEG ERROR " + name + "] " + scanner.Text())
+			log.Printf("[FFMPEG ERR %s] %s", name, scanner.Text())
 		}
 	}()
+
+	return nil
 }
 
-// SendPacket When a SRT packet is received, transmit it to all FFMPEG instances related to the stream key
+// SendPacket forward data to all FFMpeg instances related to the stream name
 func SendPacket(name string, data []byte) {
 	stdin := ffmpegInputStreams[name]
 	_, err := (*stdin).Write(data)
@@ -84,11 +89,12 @@ func SendPacket(name string, data []byte) {
 }
 
 // CloseConnection When the stream is ended, close FFMPEG instances
-func CloseConnection(name string) {
+func CloseConnection(name string) error {
 	ffmpeg := ffmpegInstances[name]
 	if err := ffmpeg.Process.Kill(); err != nil {
-		panic(err)
+		return err
 	}
 	delete(ffmpegInstances, name)
 	delete(ffmpegInputStreams, name)
+	return nil
 }