mirror of https://github.com/v2ray/v2ray-core
69 lines
1.1 KiB
Go
69 lines
1.1 KiB
Go
package io
|
|
|
|
import (
|
|
"io"
|
|
"sync"
|
|
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
|
)
|
|
|
|
type BufferedReader struct {
|
|
sync.Mutex
|
|
reader io.Reader
|
|
buffer *alloc.Buffer
|
|
cached bool
|
|
}
|
|
|
|
func NewBufferedReader(rawReader io.Reader) *BufferedReader {
|
|
return &BufferedReader{
|
|
reader: rawReader,
|
|
buffer: alloc.NewBuffer().Clear(),
|
|
cached: true,
|
|
}
|
|
}
|
|
|
|
func (this *BufferedReader) Release() {
|
|
this.Lock()
|
|
defer this.Unlock()
|
|
|
|
this.buffer.Release()
|
|
this.buffer = nil
|
|
this.reader = nil
|
|
}
|
|
|
|
func (this *BufferedReader) Cached() bool {
|
|
return this.cached
|
|
}
|
|
|
|
func (this *BufferedReader) SetCached(cached bool) {
|
|
this.cached = cached
|
|
}
|
|
|
|
func (this *BufferedReader) Read(b []byte) (int, error) {
|
|
this.Lock()
|
|
defer this.Unlock()
|
|
|
|
if this.reader == nil {
|
|
return 0, io.EOF
|
|
}
|
|
|
|
if !this.cached {
|
|
if !this.buffer.IsEmpty() {
|
|
return this.buffer.Read(b)
|
|
}
|
|
return this.reader.Read(b)
|
|
}
|
|
if this.buffer.IsEmpty() {
|
|
_, err := this.buffer.FillFrom(this.reader)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
|
|
if this.buffer.IsEmpty() {
|
|
return 0, nil
|
|
}
|
|
|
|
return this.buffer.Read(b)
|
|
}
|