clientrtmp: perform rtsp->rtmp conversion entirely inside the writer routine

internal/clientrtmp/client.go

@@ -56,6 +56,11 @@ func pathNameAndQuery(inURL *url.URL) (string, url.Values) {
 	return pathName, ur.Query()
 }
 
+type trackIDBufPair struct {
+	trackID int
+	buf     []byte
+}
+
 // Parent is implemented by clientman.ClientMan.
 type Parent interface {
 	Log(logger.Level, string, ...interface{})
@@ -313,18 +318,28 @@ func (c *Client) runRead() {
 					return fmt.Errorf("terminated")
 				}
 
+				pair := data.(trackIDBufPair)
+
 				now := time.Now()
 
-				switch tdata := data.(type) {
-				case *rtph264.NALUAndTimestamp:
+				if c.videoTrack != nil && pair.trackID == c.videoTrack.ID {
+					nts, err := c.h264Decoder.Decode(pair.buf)
+					if err != nil {
+						if err != rtph264.ErrMorePacketsNeeded {
+							c.log(logger.Debug, "ERR while decoding video track: %v", err)
+						}
+						continue
+					}
+
+					for _, nt := range nts {
 						if !videoInitialized {
 							videoInitialized = true
 							videoStartDTS = now
-						videoPTS = tdata.Timestamp
+							videoPTS = nt.Timestamp
 						}
 
 						// aggregate NALUs by PTS
-					if tdata.Timestamp != videoPTS {
+						if nt.Timestamp != videoPTS {
 							pkt := av.Packet{
 								Type: av.H264,
 								Data: h264.FillNALUsAVCC(videoBuf),
@@ -340,14 +355,24 @@ func (c *Client) runRead() {
 							videoBuf = nil
 						}
 
-					videoPTS = tdata.Timestamp
-					videoBuf = append(videoBuf, tdata.NALU)
+						videoPTS = nt.Timestamp
+						videoBuf = append(videoBuf, nt.NALU)
+					}
+					continue
+				}
+
+				if c.audioTrack != nil && pair.trackID == c.audioTrack.ID {
+					ats, err := c.aacDecoder.Decode(pair.buf)
+					if err != nil {
+						c.log(logger.Debug, "ERR while decoding audio track: %v", err)
+						continue
+					}
 
-				case *rtpaac.AUAndTimestamp:
+					for _, at := range ats {
 						pkt := av.Packet{
 							Type: av.AAC,
-						Data: tdata.AU,
-						Time: tdata.Timestamp,
+							Data: at.AU,
+							Time: at.Timestamp,
 						}
 
 						c.conn.NetConn().SetWriteDeadline(time.Now().Add(c.writeTimeout))
@@ -356,6 +381,8 @@ func (c *Client) runRead() {
 							return err
 						}
 					}
+					continue
+				}
 			}
 		}()
 	}()
@@ -616,36 +643,6 @@ func (c *Client) Authenticate(authMethods []headers.AuthMethod,
 // OnIncomingFrame implements path.Reader.
 func (c *Client) OnIncomingFrame(trackID int, streamType gortsplib.StreamType, buf []byte) {
 	if streamType == gortsplib.StreamTypeRTP {
-		if c.videoTrack != nil {
-			if trackID == c.videoTrack.ID {
-				nts, err := c.h264Decoder.Decode(buf)
-				if err != nil {
-					if err != rtph264.ErrMorePacketsNeeded {
-						c.log(logger.Debug, "ERR while decoding video track: %v", err)
-					}
-					return
-				}
-
-				for _, nt := range nts {
-					c.ringBuffer.Push(nt)
-				}
-				return
-			}
-		}
-
-		if c.audioTrack != nil {
-			if trackID == c.audioTrack.ID {
-				ats, err := c.aacDecoder.Decode(buf)
-				if err != nil {
-					c.log(logger.Debug, "ERR while decoding audio track: %v", err)
-					return
-				}
-
-				for _, at := range ats {
-					c.ringBuffer.Push(at)
-				}
-				return
-			}
-		}
+		c.ringBuffer.Push(trackIDBufPair{trackID, buf})
 	}
 }