mirror of
https://github.com/encounter/go-rtmp.git
synced 2026-03-30 11:12:49 -07:00
yutopp
38 lines
817 B
Go
38 lines
817 B
Go
//
|
|
// Copyright (c) 2018- yutopp (yutopp@gmail.com)
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
package rtmp
|
|
|
|
import (
|
|
"io"
|
|
)
|
|
|
|
type ChunkStreamerReader struct {
|
|
reader io.Reader
|
|
totalReadBytes uint32 // TODO: Check overflow
|
|
fragmentReadBytes uint32
|
|
}
|
|
|
|
func (r *ChunkStreamerReader) Read(b []byte) (int, error) {
|
|
n, err := r.reader.Read(b)
|
|
r.totalReadBytes += uint32(n)
|
|
r.fragmentReadBytes += uint32(n)
|
|
return n, err
|
|
}
|
|
|
|
func (r *ChunkStreamerReader) TotalReadBytes() uint32 {
|
|
return r.totalReadBytes
|
|
}
|
|
|
|
func (r *ChunkStreamerReader) FragmentReadBytes() uint32 {
|
|
return r.fragmentReadBytes
|
|
}
|
|
|
|
func (r *ChunkStreamerReader) ResetFragmentReadBytes() {
|
|
r.fragmentReadBytes = 0
|
|
}
|