mirror of https://github.com/v2ray/v2ray-core
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package buf
|
|
|
|
import (
|
|
"io"
|
|
)
|
|
|
|
// BufferedReader is a reader with internal cache.
|
|
type BufferedReader struct {
|
|
reader io.Reader
|
|
buffer *Buffer
|
|
buffered bool
|
|
}
|
|
|
|
// NewBufferedReader creates a new BufferedReader based on an io.Reader.
|
|
func NewBufferedReader(rawReader io.Reader) *BufferedReader {
|
|
return &BufferedReader{
|
|
reader: rawReader,
|
|
buffer: NewLocal(1024),
|
|
buffered: true,
|
|
}
|
|
}
|
|
|
|
// IsBuffered returns true if the internal cache is effective.
|
|
func (r *BufferedReader) IsBuffered() bool {
|
|
return r.buffered
|
|
}
|
|
|
|
// SetBuffered is to enable or disable internal cache. If cache is disabled,
|
|
// Read() calls will be delegated to the underlying io.Reader directly.
|
|
func (r *BufferedReader) SetBuffered(cached bool) {
|
|
r.buffered = cached
|
|
}
|
|
|
|
// Read implements io.Reader.Read().
|
|
func (r *BufferedReader) Read(b []byte) (int, error) {
|
|
if !r.buffered || r.buffer == nil {
|
|
if !r.buffer.IsEmpty() {
|
|
return r.buffer.Read(b)
|
|
}
|
|
return r.reader.Read(b)
|
|
}
|
|
if r.buffer.IsEmpty() {
|
|
if err := r.buffer.Reset(ReadFrom(r.reader)); err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
|
|
if r.buffer.IsEmpty() {
|
|
return 0, nil
|
|
}
|
|
|
|
return r.buffer.Read(b)
|
|
}
|