revid: revid output check and destination setting cleaner. Also changed destination release to single chunk.Close() so that chunks aren't closed more than once

revid/revid.go

@@ -43,7 +43,6 @@ import (
 	"bitbucket.org/ausocean/av/stream/flv"
 	"bitbucket.org/ausocean/av/stream/lex"
 	"bitbucket.org/ausocean/av/stream/mts"
-	"bitbucket.org/ausocean/av/stream/rtp"
 	"bitbucket.org/ausocean/iot/pi/netsender"
 	"bitbucket.org/ausocean/utils/ring"
 	"bitbucket.org/ausocean/utils/smartlogger"
@@ -75,6 +74,11 @@ const (
 	pkg                     = "revid:"
 )
 
+const (
+	out1 = 0
+	out2 = 1
+)
+
 // Log Types
 const (
 	Error   = "Error"
@@ -209,68 +213,50 @@ func (r *Revid) reset(config Config) error {
 		}
 	}
 
+	n := 1
 	if r.config.Output2 != 0 {
-		r.destination = make([]loadSender, 2)
+		n = 2
-	} else {
-		r.destination = make([]loadSender, 1)
 	}
+	r.destination = make([]loadSender, n)
 
-	switch r.config.Output1 {
+	for outNo, outType := range []uint8{r.config.Output1, r.config.Output2} {
+		switch outType {
 		case File:
 			s, err := newFileSender(config.OutputFileName)
 			if err != nil {
 				return err
 			}
-		r.destination[0] = s
+			r.destination[outNo] = s
 		case FfmpegRtmp:
 			s, err := newFfmpegSender(config.RtmpUrl, r.config.FrameRate)
 			if err != nil {
 				return err
 			}
-		r.destination[0] = s
+			r.destination[outNo] = s
 		case Rtmp:
 			s, err := newRtmpSender(config.RtmpUrl, rtmpConnectionTimout, rtmpConnectionMaxTries, r.config.Logger.Log)
 			if err != nil {
 				return err
 			}
-		r.destination[0] = s
+			r.destination[outNo] = s
 		case Http:
-		r.destination[0] = newHttpSender(r.ns, r.config.Logger.Log)
+			r.destination[outNo] = newHttpSender(r.ns, r.config.Logger.Log)
-	case Rtp, Udp:
+		case Udp:
 			s, err := newUdpSender(r.config.RtpAddress, r.config.Logger.Log)
 			if err != nil {
 				return err
 			}
-		r.destination[0] = s
+			r.destination[outNo] = s
-	}
+		case Rtp:
-
+			// TODO: framerate in config should probably be an int, make conversions early
-	switch r.config.Output2 {
+			// when setting config fields in revid-cli
-	case File:
+			fps, _ := strconv.Atoi(r.config.FrameRate)
-		s, err := newFileSender(config.OutputFileName)
+			s, err := newRtpSender(r.config.RtpAddress, r.config.Logger.Log, fps)
 			if err != nil {
 				return err
 			}
-		r.destination[1] = s
+			r.destination[outNo] = s
-	case FfmpegRtmp:
-		s, err := newFfmpegSender(config.RtmpUrl, r.config.FrameRate)
-		if err != nil {
-			return err
 		}
-		r.destination[1] = s
-	case Rtmp:
-		s, err := newRtmpSender(config.RtmpUrl, rtmpConnectionTimout, rtmpConnectionMaxTries, r.config.Logger.Log)
-		if err != nil {
-			return err
-		}
-		r.destination[1] = s
-	case Http:
-		r.destination[1] = newHttpSender(r.ns, r.config.Logger.Log)
-	case Rtp, Udp:
-		s, err := newUdpSender(r.config.RtpAddress, r.config.Logger.Log)
-		if err != nil {
-			return err
-		}
-		r.destination[1] = s
 	}
 
 	switch r.config.Input {
@@ -316,10 +302,6 @@ func (r *Revid) reset(config Config) error {
 		if err != nil {
 			return err
 		}
-	case MpegtsRtp:
-		r.config.Logger.Log(smartlogger.Info, pkg+"using RTP packetisation")
-		frameRate, _ := strconv.Atoi(r.config.FrameRate)
-		r.encoder = mts.NewEncoder(rtp.NewEncoder(&r.packer, frameRate), float64(frameRate))
 	}
 
 	return nil
@@ -413,8 +395,10 @@ loop:
 								r.config.Logger.Log(smartlogger.Error, pkg+"failed to restart rtmp session, trying again", "error", err.Error())
 								err = rs.restart()
 							}
+
 							r.config.Logger.Log(smartlogger.Info, pkg+"restarted rtmp session")
 						}
+
 						err = dest.send()
 						if err != nil {
 							r.config.Logger.Log(smartlogger.Error, pkg+"send failed again, with error", "error", err.Error())
@@ -424,10 +408,8 @@ loop:
 			}
 		}
 
-		for _, dest := range r.destination {
+		// Release the chunk back to the ring buffer for use
-			dest.release()
+		chunk.Close()
-		}
-
 		r.config.Logger.Log(smartlogger.Debug, pkg+"done reading that clip from ring buffer")
 
 		// Log some information regarding bitrate and ring buffer size if it's time

revid/senders.go

@@ -292,7 +292,7 @@ func (s *rtmpSender) close() error {
 	return s.sess.Close()
 }
 
-// rtpSender implements loadSender for a native udp destination.
+// udpSender implements loadSender for a native udp destination.
 type udpSender struct {
 	conn  net.Conn
 	log   func(lvl int8, msg string, args ...interface{})
@@ -327,7 +327,8 @@ func (s *udpSender) release() {
 
 func (s *udpSender) close() error { return nil }
 
-// rtpSender implements loadSender for a native udp destination.
+// TODO: Write restart func for rtpSender
+// rtpSender implements loadSender for a native udp destination with rtp packetization.
 type rtpSender struct {
 	log     func(lvl int8, msg string, args ...interface{})
 	chunk   *ring.Chunk