package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-10-06 15:46:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

ringbuffer: when buffer is full, preserve queued data (#386)

Browse Source
This commit is contained in:
Alessandro Ros
2023-08-26 17:14:04 +02:00
committed by GitHub
parent e87c6d66e5
commit bf3de06b3c
3 changed files with 85 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
pkg/ringbuffer/ringbuffer_test.go
View File

@@ -18,12 +18,12 @@ func TestPushBeforePull(t *testing.T) {
require.NoError(t, err)
defer r.Close()
data := bytes.Repeat([]byte{0x01, 0x02, 0x03, 0x04}, 1024/4)
ok := r.Push(bytes.Repeat([]byte{1, 2, 3, 4}, 1024/4))
require.Equal(t, true, ok)
r.Push(data)
ret, ok := r.Pull()
require.Equal(t, true, ok)
require.Equal(t, data, ret)
require.Equal(t, bytes.Repeat([]byte{1, 2, 3, 4}, 1024/4), ret)
}
func TestPullBeforePush(t *testing.T) {
@@ -31,19 +31,19 @@ func TestPullBeforePush(t *testing.T) {
require.NoError(t, err)
defer r.Close()
data := bytes.Repeat([]byte{0x01, 0x02, 0x03, 0x04}, 1024/4)
done := make(chan struct{})
go func() {
defer close(done)
ret, ok := r.Pull()
require.Equal(t, true, ok)
require.Equal(t, data, ret)
require.Equal(t, bytes.Repeat([]byte{1, 2, 3, 4}, 1024/4), ret)
}()
time.Sleep(100 * time.Millisecond)
r.Push(data)
ok := r.Push(bytes.Repeat([]byte{1, 2, 3, 4}, 1024/4))
require.Equal(t, true, ok)
<-done
}
@@ -62,17 +62,37 @@ func TestClose(t *testing.T) {
require.Equal(t, false, ok)
}()
r.Push([]byte{0x01, 0x02, 0x03, 0x04})
ok := r.Push([]byte{1, 2, 3, 4})
require.Equal(t, true, ok)
r.Close()
<-done
r.Reset()
r.Push([]byte{0x05, 0x06, 0x07, 0x08})
_, ok := r.Pull()
ok = r.Push([]byte{5, 6, 7, 8})
require.Equal(t, true, ok)
_, ok = r.Pull()
require.Equal(t, true, ok)
}
func TestOverflow(t *testing.T) {
r, err := New(32)
require.NoError(t, err)
for i := 0; i < 32; i++ {
r.Push([]byte{1, 2, 3, 4})
}
ok := r.Push([]byte{5, 6, 7, 8})
require.Equal(t, false, ok)
for i := 0; i < 32; i++ {
data, ok := r.Pull()
require.Equal(t, true, ok)
require.Equal(t, []byte{1, 2, 3, 4}, data)
}
}
func BenchmarkPushPullContinuous(b *testing.B) {
@@ -123,7 +143,7 @@ func BenchmarkPushPullPaused5(b *testing.B) {
}
func BenchmarkPushPullPaused10(b *testing.B) {
r, _ := New(1024 * 8)
r, _ := New(128)
defer r.Close()
data := make([]byte, 1024)