1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-22 18:17:52 -05:00
v2fly/common/buf/reader.go

80 lines
1.4 KiB
Go
Raw Normal View History

2016-12-09 07:17:34 -05:00
package buf
2016-12-27 14:53:29 -05:00
import "io"
2016-12-09 07:17:34 -05:00
// 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() {
2016-12-21 09:37:16 -05:00
err := buffer.AppendSupplier(ReadFrom(v.largeBuffer))
return buffer, err
2016-12-09 07:17:34 -05:00
}
err := buffer.AppendSupplier(ReadFrom(v.reader))
if err != nil {
buffer.Release()
return nil, err
}
if buffer.IsFull() {
v.highVolumn = true
}
return buffer, nil
}
2017-02-20 04:33:35 -05:00
type bufferToBytesReader struct {
2016-12-09 07:17:34 -05:00
stream Reader
current *Buffer
2017-02-06 18:39:07 -05:00
err error
2016-12-09 07:17:34 -05:00
}
2017-02-20 04:33:35 -05:00
// fill fills in the internal buffer.
func (v *bufferToBytesReader) fill() {
2016-12-09 07:17:34 -05:00
b, err := v.stream.Read()
v.current = b
if err != nil {
2017-02-06 18:39:07 -05:00
v.err = err
2016-12-09 07:17:34 -05:00
v.current = nil
}
}
2017-02-20 04:33:35 -05:00
func (v *bufferToBytesReader) Read(b []byte) (int, error) {
2017-02-06 18:39:07 -05:00
if v.err != nil {
return 0, v.err
2016-12-09 07:17:34 -05:00
}
if v.current == nil {
2017-02-20 04:33:35 -05:00
v.fill()
2017-02-06 18:39:07 -05:00
if v.err != nil {
return 0, v.err
2016-12-09 07:17:34 -05:00
}
}
nBytes, err := v.current.Read(b)
if v.current.IsEmpty() {
v.current.Release()
v.current = nil
}
return nBytes, err
}