Merged in senders-are-writers (pull request #176)

revid: senders are now io.Writers

Approved-by: kortschak <dan@kortschak.io>
Approved-by: Alan Noble <anoble@gmail.com>

revid/revid.go

@@ -184,35 +184,35 @@ func (r *Revid) setupPipeline(mtsEnc func(io.Writer, int) io.Writer, flvEnc func
 
 	// mtsSenders will hold the senders the require MPEGTS encoding, and flvSenders
 	// will hold senders that require FLV encoding.
-	var mtsSenders, flvSenders []loadSender
+	var mtsSenders, flvSenders []io.Writer
 
 	// We will go through our outputs and create the corresponding senders to add
 	// to mtsSenders if the output requires MPEGTS encoding, or flvSenders if the
 	// output requires FLV encoding.
-	var sender loadSender
+	var w io.Writer
 	for _, out := range r.config.Outputs {
 		switch out {
 		case Http:
-			sender = newMtsSender(newMinimalHttpSender(r.ns, r.config.Logger.Log), nil)
-			mtsSenders = append(mtsSenders, sender)
+			w = newMtsSender(newMinimalHttpSender(r.ns, r.config.Logger.Log), nil)
+			mtsSenders = append(mtsSenders, w)
 		case Rtp:
-			sender, err := newRtpSender(r.config.RtpAddress, r.config.Logger.Log, r.config.FrameRate)
+			w, err := newRtpSender(r.config.RtpAddress, r.config.Logger.Log, r.config.FrameRate)
 			if err != nil {
 				r.config.Logger.Log(logger.Warning, pkg+"rtp connect error", "error", err.Error())
 			}
-			mtsSenders = append(mtsSenders, sender)
+			mtsSenders = append(mtsSenders, w)
 		case File:
-			sender, err := newFileSender(r.config.OutputPath)
+			w, err := newFileSender(r.config.OutputPath)
 			if err != nil {
 				return err
 			}
-			mtsSenders = append(mtsSenders, sender)
+			mtsSenders = append(mtsSenders, w)
 		case Rtmp:
-			sender, err := newRtmpSender(r.config.RtmpUrl, rtmpConnectionTimeout, rtmpConnectionMaxTries, r.config.Logger.Log)
+			w, err := newRtmpSender(r.config.RtmpUrl, rtmpConnectionTimeout, rtmpConnectionMaxTries, r.config.Logger.Log)
 			if err != nil {
 				r.config.Logger.Log(logger.Warning, pkg+"rtmp connect error", "error", err.Error())
 			}
-			flvSenders = append(flvSenders, sender)
+			flvSenders = append(flvSenders, w)
 		}
 	}
 

revid/revid_test.go

@@ -1,7 +1,6 @@
 package revid
 
 import (
-	"errors"
 	"fmt"
 	"io"
 	"os"
@@ -100,11 +99,11 @@ func (e *tstFlvEncoder) Write(d []byte) (int, error) { return 0, nil }
 func TestResetEncoderSenderSetup(t *testing.T) {
 	// We will use these to indicate types after assertion.
 	const (
-		mtsSenderStr  = "revid.mtsSender"
-		rtpSenderStr  = "revid.rtpSender"
-		rtmpSenderStr = "revid.RtmpSender"
-		mtsEncoderStr = "mts.Encoder"
-		flvEncoderStr = "flv.Encoder"
+		mtsSenderStr  = "*revid.mtsSender"
+		rtpSenderStr  = "*revid.rtpSender"
+		rtmpSenderStr = "*revid.rtmpSender"
+		mtsEncoderStr = "*revid.tstMtsEncoder"
+		flvEncoderStr = "*revid.tstFlvEncoder"
 	)
 
 	// Struct that will be used to format test cases nicely below.
@@ -185,31 +184,6 @@ func TestResetEncoderSenderSetup(t *testing.T) {
 		},
 	}
 
-	// typeOfEncoder will return the type of encoder implementing stream.Encoder.
-	typeOfEncoder := func(i io.Writer) (string, error) {
-		if _, ok := i.(*tstMtsEncoder); ok {
-			return mtsEncoderStr, nil
-		}
-		if _, ok := i.(*tstFlvEncoder); ok {
-			return flvEncoderStr, nil
-		}
-		return "", errors.New("unknown Encoder type")
-	}
-
-	// typeOfSender will return the type of sender implementing loadSender.
-	typeOfSender := func(s loadSender) (string, error) {
-		if _, ok := s.(*mtsSender); ok {
-			return mtsSenderStr, nil
-		}
-		if _, ok := s.(*rtpSender); ok {
-			return rtpSenderStr, nil
-		}
-		if _, ok := s.(*rtmpSender); ok {
-			return rtmpSenderStr, nil
-		}
-		return "", errors.New("unknown loadSender type")
-	}
-
 	rv, err := New(Config{Logger: &testLogger{}}, nil)
 	if err != nil {
 		t.Fatalf("unexpected err: %v", err)
@@ -241,10 +215,7 @@ func TestResetEncoderSenderSetup(t *testing.T) {
 		// Now check the correctness of encoders and their destinations.
 		for _, e := range rv.encoder {
 			// Get e's type.
-			encoderType, err := typeOfEncoder(e)
-			if err != nil {
-				t.Fatalf("could not get encoders type for test %v, failed with err: %v", testNum, err)
-			}
+			encoderType := fmt.Sprintf("%T", e)
 
 			// Check that we expect this encoder to be here.
 			idx := -1
@@ -266,7 +237,7 @@ func TestResetEncoderSenderSetup(t *testing.T) {
 				ms = e.(*tstFlvEncoder).dst
 			}
 
-			senders := ms.(*multiSender).senders
+			senders := ms.(*multiSender).dst
 			got = len(senders)
 			want = len(test.encoders[idx].destinations)
 			if got != want {
@@ -278,10 +249,7 @@ func TestResetEncoderSenderSetup(t *testing.T) {
 				ok := false
 				for _, dst := range senders {
 					// Get type of sender.
-					senderType, err := typeOfSender(dst)
-					if err != nil {
-						t.Fatalf("could not get encoders type for test %v, failed with err: %v", testNum, err)
-					}
+					senderType := fmt.Sprintf("%T", dst)
 
 					// If it's one we want, indicate.
 					if senderType == expectDst {

revid/senders.go

@@ -31,6 +31,7 @@ package revid
 import (
 	"errors"
 	"fmt"
+	"io"
 	"net"
 	"os"
 	"strconv"
@@ -58,30 +59,33 @@ type Log func(level int8, message string, params ...interface{})
 // multiSender implements io.Writer. It provides the capacity to send to multiple
 // senders from a single Write call.
 type multiSender struct {
-	senders []loadSender
-	log     Log
+	dst []io.Writer
+	log Log
 }
 
 // newMultiSender returns a pointer to a new multiSender.
-func newMultiSender(senders []loadSender, log Log) *multiSender {
+func newMultiSender(senders []io.Writer, log Log) *multiSender {
 	return &multiSender{
-		senders: senders,
-		log:     log,
+		dst: senders,
+		log: log,
 	}
 }
 
 // Write implements io.Writer. This will call load (with the passed slice), send
 // and release on all senders of multiSender.
 func (s *multiSender) Write(d []byte) (int, error) {
-	for i, sender := range s.senders {
-		sender.load(d)
-		err := sender.send()
+	for i, sender := range s.dst {
+		_, err := sender.Write(d)
 		if err != nil {
 			s.log(logger.Warning, pkg+"send failed", "sender", i, "error", err)
 		}
 	}
-	for _, sender := range s.senders {
-		sender.release()
+	for _, sender := range s.dst {
+		s, ok := sender.(loadSender)
+		if !ok {
+			panic("sender is not a loadSender")
+		}
+		s.release()
 	}
 	return len(d), nil
 }
@@ -137,7 +141,20 @@ type fileSender struct {
 	data []byte
 }
 
-func newFileSender(path string) (*fileSender, error) {
+// Write implements io.Writer.
+func (s *fileSender) Write(d []byte) (int, error) {
+	err := s.load(d)
+	if err != nil {
+		return 0, err
+	}
+	err = s.send()
+	if err != nil {
+		return len(d), err
+	}
+	return len(d), nil
+}
+
+func newFileSender(path string) (io.Writer, error) {
 	f, err := os.Create(path)
 	if err != nil {
 		return nil, err
@@ -174,6 +191,11 @@ type mtsSender struct {
 	curPid    int
 }
 
+// Write implements io.Writer.
+func (s *mtsSender) Write(d []byte) (int, error) {
+	return write(s, d)
+}
+
 // newMtsSender returns a new mtsSender.
 func newMtsSender(s Sender, log func(lvl int8, msg string, args ...interface{})) *mtsSender {
 	return &mtsSender{
@@ -357,6 +379,11 @@ func newRtmpSender(url string, timeout uint, retries int, log func(lvl int8, msg
 	return s, err
 }
 
+// Write implements io.Writer.
+func (s *rtmpSender) Write(d []byte) (int, error) {
+	return write(s, d)
+}
+
 func (s *rtmpSender) load(d []byte) error {
 	s.data = d
 	return nil
@@ -406,6 +433,11 @@ type rtpSender struct {
 	data    []byte
 }
 
+// Write implements io.Writer.
+func (s *rtpSender) Write(d []byte) (int, error) {
+	return write(s, d)
+}
+
 func newRtpSender(addr string, log func(lvl int8, msg string, args ...interface{}), fps uint) (*rtpSender, error) {
 	conn, err := net.Dial("udp", addr)
 	if err != nil {
@@ -432,3 +464,16 @@ func (s *rtpSender) send() error {
 	_, err := s.encoder.Write(s.data)
 	return err
 }
+
+// write wraps the load and send method for loadSenders.
+func write(s loadSender, d []byte) (int, error) {
+	err := s.load(d)
+	if err != nil {
+		return 0, err
+	}
+	err = s.send()
+	if err != nil {
+		return len(d), err
+	}
+	return len(d), nil
+}

revid/senders_test.go

@@ -31,6 +31,7 @@ package revid
 import (
 	"errors"
 	"fmt"
+	"io"
 	"sync"
 	"testing"
 	"time"
@@ -271,6 +272,10 @@ func newDummyLoadSender(fail bool, retry bool) *dummyLoadSender {
 	return &dummyLoadSender{failOnSend: fail, failHandled: true, retry: retry}
 }
 
+func (s *dummyLoadSender) Write(d []byte) (int, error) {
+	return write(s, d)
+}
+
 // load takes a byte slice and assigns it to the dummyLoadSenders data slice.
 func (s *dummyLoadSender) load(d []byte) error {
 	s.data = d
@@ -315,7 +320,7 @@ func (s *dummyLoadSender) retrySend() bool { return s.retry }
 // TestMultiSenderWrite checks that we can do basic writing to multiple senders
 // using the multiSender.
 func TestMultiSenderWrite(t *testing.T) {
-	senders := []loadSender{
+	senders := []io.Writer{
 		newDummyLoadSender(false, false),
 		newDummyLoadSender(false, false),
 		newDummyLoadSender(false, false),
@@ -330,7 +335,7 @@ func TestMultiSenderWrite(t *testing.T) {
 
 	// Check that the senders got the data correctly from the writes.
 	for i := byte(0); i < noOfWrites; i++ {
-		for j, dest := range ms.senders {
+		for j, dest := range ms.dst {
 			got := dest.(*dummyLoadSender).buf[i][0]
 			if got != i {
 				t.Errorf("Did not get expected result for sender: %v. \nGot: %v\nWant: %v\n", j, got, i)