mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
1.8 KiB
103 lines
1.8 KiB
package buf
|
|
|
|
import (
|
|
"io"
|
|
"sync"
|
|
)
|
|
|
|
// BytesToBufferReader is a Reader that adjusts its reading speed automatically.
|
|
type BytesToBufferReader struct {
|
|
reader io.Reader
|
|
largeBuffer *Buffer
|
|
highVolumn bool
|
|
}
|
|
|
|
// Read implements Reader.Read().
|
|
func (v *BytesToBufferReader) Read() (*Buffer, error) {
|
|
if v.highVolumn && v.largeBuffer.IsEmpty() {
|
|
if v.largeBuffer == nil {
|
|
v.largeBuffer = NewLocal(32 * 1024)
|
|
}
|
|
err := v.largeBuffer.AppendSupplier(ReadFrom(v.reader))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if v.largeBuffer.Len() < Size {
|
|
v.highVolumn = false
|
|
}
|
|
}
|
|
|
|
buffer := New()
|
|
if !v.largeBuffer.IsEmpty() {
|
|
err := buffer.AppendSupplier(ReadFrom(v.largeBuffer))
|
|
return buffer, err
|
|
}
|
|
|
|
err := buffer.AppendSupplier(ReadFrom(v.reader))
|
|
if err != nil {
|
|
buffer.Release()
|
|
return nil, err
|
|
}
|
|
|
|
if buffer.IsFull() {
|
|
v.highVolumn = true
|
|
}
|
|
|
|
return buffer, nil
|
|
}
|
|
|
|
// Release implements Releasable.Release().
|
|
func (v *BytesToBufferReader) Release() {
|
|
v.reader = nil
|
|
}
|
|
|
|
type BufferToBytesReader struct {
|
|
sync.Mutex
|
|
stream Reader
|
|
current *Buffer
|
|
eof bool
|
|
}
|
|
|
|
// Fill fills in the internal buffer.
|
|
// Private: Visible for testing.
|
|
func (v *BufferToBytesReader) Fill() {
|
|
b, err := v.stream.Read()
|
|
v.current = b
|
|
if err != nil {
|
|
v.eof = true
|
|
v.current = nil
|
|
}
|
|
}
|
|
|
|
func (v *BufferToBytesReader) Read(b []byte) (int, error) {
|
|
if v.eof {
|
|
return 0, io.EOF
|
|
}
|
|
|
|
v.Lock()
|
|
defer v.Unlock()
|
|
if v.current == nil {
|
|
v.Fill()
|
|
if v.eof {
|
|
return 0, io.EOF
|
|
}
|
|
}
|
|
nBytes, err := v.current.Read(b)
|
|
if v.current.IsEmpty() {
|
|
v.current.Release()
|
|
v.current = nil
|
|
}
|
|
return nBytes, err
|
|
}
|
|
|
|
// Release implements Releasable.Release().
|
|
func (v *BufferToBytesReader) Release() {
|
|
v.Lock()
|
|
defer v.Unlock()
|
|
|
|
v.eof = true
|
|
v.current.Release()
|
|
v.current = nil
|
|
v.stream = nil
|
|
}
|