chunk_streamer_reader.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
}