From a8d528b14afc49e8412b659764eae9c8ae6e15b4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AE=87=E7=BF=94?= <11123448@vivo.xyz>
Date: Fri, 9 Jul 2021 11:54:21 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=89=A9=E7=BC=A9bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ring.go        |  5 +++++
 video_track.go | 30 +++++++++++++++++++-----------
 2 files changed, 24 insertions(+), 11 deletions(-)

diff --git a/ring.go b/ring.go
index 9eb188f..1fcc861 100644
--- a/ring.go
+++ b/ring.go
@@ -108,6 +108,11 @@ func (r *RingBuffer) CurrentValue() interface{} {
 	return r.Current().Value
 }
 
+func (r *RingBuffer) NextValue() interface{} {
+	return r.Next().Value.(*RingItem).Value
+}
+
+
 func (r *RingBuffer) Current() *RingItem {
 	return r.Ring.Value.(*RingItem)
 }
diff --git a/video_track.go b/video_track.go
index 832514c..b607ad9 100644
--- a/video_track.go
+++ b/video_track.go
@@ -44,6 +44,7 @@ type VideoTrack struct {
 	PushNalu        func(ts uint32, cts uint32, nalus ...[]byte) `json:"-"`
 	UsingDonlField  bool
 	writeByteStream func(pack *VideoPack)
+	idrCount        int //处于缓冲中的关键帧数量
 }
 
 func (vt *VideoTrack) initVideoRing(v interface{}) {
@@ -60,20 +61,15 @@ func (s *Stream) NewVideoTrack(codec byte) (vt *VideoTrack) {
 			vt.IDRing = vt.Ring
 			cancel()
 			idrSequence := vt.current().Sequence
-			l := vt.Ring.Len()
+			// l := vt.Ring.Len()
+			vt.idrCount++
 			vt.revIDR = func() {
+				vt.idrCount++
 				current := vt.current()
-				if vt.GOP = current.Sequence - idrSequence; vt.GOP > l-1 {
-					//缓冲环不够大，导致IDR被覆盖
-					exRing := NewRingBuffer(vt.GOP - l + 5).Ring
-					exRing.Do(vt.initVideoRing)
-					vt.Link(exRing) // 扩大缓冲环
-					l = vt.Ring.Len()
-					utils.Printf("%s ring grow to %d", s.StreamPath, l)
-				} else if vt.GOP < l-5 {
+				l := vt.Ring.Len()
+				if vt.GOP = current.Sequence - idrSequence; vt.GOP < l-5 {
 					vt.Unlink(l - vt.GOP - 5) //缩小缓冲环节省内存
-					l = vt.Ring.Len()
-					utils.Printf("%s ring atrophy to %d", s.StreamPath, l)
+					//utils.Printf("%s gop:%d ring atrophy to %d", s.StreamPath, vt.GOP, l)
 				}
 				vt.IDRing = vt.Ring
 				idrSequence = current.Sequence
@@ -529,5 +525,17 @@ func (vt *VideoTrack) push(pack *VideoPack) {
 		vt.revIDR()
 	}
 	vt.lastTs = pack.Timestamp
+	nextPack := vt.NextValue().(*VideoPack)
+	if nextPack.IDR {
+		if vt.idrCount == 1 {
+			exRing := NewRingBuffer(5).Ring
+			exRing.Do(vt.initVideoRing)
+			vt.Link(exRing) // 扩大缓冲环
+			//l := vt.Ring.Len()
+			//utils.Printf("%s ring grow to %d", vt.Stream.StreamPath, l)
+		} else {
+			vt.idrCount--
+		}
+	}
 	vt.Step()
 }