Use channels to send SRT packets to forwarded streams

stream/forwarding/forwarding.go

@@ -2,6 +2,7 @@ package forwarding
 
 import (
 	"bufio"
+	"gitlab.crans.org/nounous/ghostream/stream/srt"
 	"io"
 	"log"
 	"os/exec"
@@ -13,16 +14,43 @@ type Options map[string][]string
 
 var (
 	cfg                Options
+	forwardingChannel  chan srt.Packet
 	ffmpegInstances    = make(map[string]*exec.Cmd)
 	ffmpegInputStreams = make(map[string]*io.WriteCloser)
 )
 
-// New Load configuration
-func New(c Options) {
+// New Load configuration and initialize SRT channel
+func New(c Options, channel chan srt.Packet) {
 	cfg = c
+	forwardingChannel = channel
+	go waitForPackets()
 	log.Printf("Stream forwarding initialized")
 }
 
+func waitForPackets() {
+	for {
+		var err error = nil
+		packet := <-forwardingChannel
+		switch packet.PacketType {
+		case "register":
+			err = RegisterStream(packet.StreamName)
+			break
+		case "sendData":
+			err = SendPacket(packet.StreamName, packet.Data)
+			break
+		case "close":
+			err = CloseConnection(packet.StreamName)
+			break
+		default:
+			log.Println("Unknown SRT packet type:", packet.PacketType)
+			break
+		}
+		if err != nil {
+			log.Printf("Error occured while receiving SRT packet of type %s: %s", packet.PacketType, err)
+		}
+	}
+}
+
 // RegisterStream Declare a new open stream and create ffmpeg instances
 func RegisterStream(name string) error {
 	streams, exist := cfg[name]
@@ -80,16 +108,14 @@ func RegisterStream(name string) error {
 }
 
 // SendPacket forward data to all FFMpeg instances related to the stream name
-func SendPacket(name string, data []byte) {
+func SendPacket(name string, data []byte) error {
 	stdin := ffmpegInputStreams[name]
 	if stdin == nil {
 		// Don't need to forward stream
-		return
+		return nil
 	}
 	_, err := (*stdin).Write(data)
-	if err != nil {
-		log.Printf("Error while sending a packet to external streaming server for key %s: %s", name, err)
-	}
+	return err
 }
 
 // CloseConnection When the stream is ended, close FFMPEG instances

stream/forwarding/forwarding_test.go

@@ -52,11 +52,14 @@ func TestForwardStream(t *testing.T) {
 
 	forwardingList := make(map[string][]string)
 	forwardingList["demo"] = []string{"rtmp://127.0.0.1:1936/live/app"}
+
+	forwardingChannel = make(chan srt.Packet)
+
 	// Register forwarding stream list
-	New(forwardingList)
+	New(forwardingList, forwardingChannel)
 
 	// Serve HTTP Server
-	go srt.Serve(&srt.Options{ListenAddress: ":9711", MaxClients: 2})
+	go srt.Serve(&srt.Options{ListenAddress: ":9711", MaxClients: 2}, forwardingChannel)
 
 	ffmpeg := exec.Command("ffmpeg",
 		"-i", "http://ftp.crans.org/events/Blender%20OpenMovies/big_buck_bunny_480p_stereo.ogg",