Track bytesReadSuccesfully in oggreader

pion · Jun 25, 2024 · fc3521e · fc3521e
1 parent 1bb9fa9
commit fc3521e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/pkg/media/oggreader/oggreader.go b/pkg/media/oggreader/oggreader.go
@@ -189,6 +189,8 @@ func (o *OggReader) ParseNextPage() ([]byte, *OggPageHeader, error) {
 		}
 	}
 
+	o.bytesReadSuccesfully += int64(len(h) + len(sizeBuffer) + len(payload))
+
 	return payload, pageHeader, nil
 }
 

diff --git a/pkg/media/oggreader/oggreader_test.go b/pkg/media/oggreader/oggreader_test.go
@@ -46,10 +46,12 @@ func TestOggReader_ParseNextPage(t *testing.T) {
 	reader, _, err := NewWith(ogg)
 	assert.NoError(t, err)
 	assert.NotNil(t, reader)
+	assert.Equal(t, int64(47), reader.bytesReadSuccesfully)
 
 	payload, _, err := reader.ParseNextPage()
 	assert.Equal(t, []byte{0x98, 0x36, 0xbe, 0x88, 0x9e}, payload)
 	assert.NoError(t, err)
+	assert.Equal(t, int64(80), reader.bytesReadSuccesfully)
 
 	_, _, err = reader.ParseNextPage()
 	assert.Equal(t, err, io.EOF)