revid: simplified mtsSender to remove load and send

mtsSender has been simplified such that load and send are no longer called in Write. Load and Send have removed and logic is now in Write. The logic has been simplified such that it does not
try to send again. On next PR when ringbuffers are added to senders, we will add logic to deal with this.

revid/senders.go

@@ -185,17 +185,10 @@ type mtsSender struct {
 	buf      []byte
 	next     []byte
 	pkt      packet.Packet
-	failed    bool
-	discarded bool
 	repairer *mts.DiscontinuityRepairer
 	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{
@@ -204,9 +197,8 @@ func newMtsSender(s Sender, log func(lvl int8, msg string, args ...interface{}))
 	}
 }
 
-// load takes a *ring.Chunk and assigns to s.next, also grabbing it's pid and
-// assigning to s.curPid. s.next if exists is also appended to the sender buf.
-func (s *mtsSender) load(d []byte) error {
+// Write implements io.Writer.
+func (s *mtsSender) Write(d []byte) (int, error) {
 	if s.next != nil {
 		s.buf = append(s.buf, s.next...)
 	}
@@ -215,49 +207,23 @@ func (s *mtsSender) load(d []byte) error {
 	s.next = bytes
 	copy(s.pkt[:], bytes)
 	s.curPid = s.pkt.PID()
-	return nil
-}
-
-// send checks the currently loaded paackets PID; if it is a PAT then what is in
-// the mtsSenders buffer is fixed and sent.
-func (ms *mtsSender) send() error {
-	if ms.curPid == mts.PatPid && len(ms.buf) > 0 {
-		err := ms.fixAndSend()
-		if err != nil {
-			return err
-		}
-		ms.buf = ms.buf[:0]
-	}
-	return nil
-}
-
-// fixAndSend checks for discontinuities in the senders buffer and then sends.
-// If a discontinuity is found the PAT packet at the start of the clip has it's
-// discontintuity indicator set to true.
-func (ms *mtsSender) fixAndSend() error {
-	err := ms.repairer.Repair(ms.buf)
+	if s.curPid == mts.PatPid && len(s.buf) > 0 {
+		err := s.repairer.Repair(s.buf)
 		if err == nil {
-		err = ms.sender.send(ms.buf)
+			err = s.sender.send(s.buf)
 			if err == nil {
-			return nil
+				goto done
 			}
 		}
-	ms.failed = true
-	ms.repairer.Failed()
-	return err
+		s.repairer.Failed()
+	done:
+		s.buf = s.buf[:0]
+	}
+	return len(d), nil
 }
 
 func (s *mtsSender) close() error { return nil }
 
-// release will set the s.fail flag to false and clear the buffer if
-// the previous send was a fail. The currently loaded chunk is also closed.
-func (s *mtsSender) release() {
-	if s.failed {
-		s.buf = s.buf[:0]
-		s.failed = false
-	}
-}
-
 // httpSender implements loadSender for posting HTTP to NetReceiver
 type httpSender struct {
 	client *netsender.Sender

revid/senders_test.go

@@ -69,6 +69,7 @@ type sender struct {
 // send takes d and neglects if testDiscontinuities is true, returning an error,
 // otherwise d is appended to senders buf.
 func (ts *sender) send(d []byte) error {
+	//fmt.Println("sending")
 	if ts.testDiscontinuities && ts.currentPkt == ts.discontinuityAt {
 		ts.currentPkt++
 		return errSendFailed
@@ -131,16 +132,10 @@ func TestMtsSenderSegment(t *testing.T) {
 				break
 			}
 
-			err = loadSender.load(next.Bytes())
+			_, err = loadSender.Write(next.Bytes())
 			if err != nil {
 				t.Fatalf("Unexpected err: %v\n", err)
 			}
-
-			err = loadSender.send()
-			if err != nil {
-				t.Fatalf("Unexpected err: %v\n", err)
-			}
-			loadSender.release()
 			next.Close()
 			next = nil
 		}
@@ -220,13 +215,10 @@ func TestMtsSenderDiscontinuity(t *testing.T) {
 				break
 			}
 
-			err = loadSender.load(next.Bytes())
+			_, err = loadSender.Write(next.Bytes())
 			if err != nil {
 				t.Fatalf("Unexpected err: %v\n", err)
 			}
-
-			loadSender.send()
-			loadSender.release()
 			next.Close()
 			next = nil
 		}