lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-02-20 23:47:21 -05:00
Code Issues Releases Wiki Activity

grow buffer size slower

Browse Source
This commit is contained in:
Darien Raymond 2017-12-08 13:23:15 +01:00
parent a42a1cba47
commit 4c7c42d39f
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
common/buf/reader.go
View File

@ -18,11 +18,14 @@ func NewBytesToBufferReader(reader io.Reader) Reader {
} }
} }
const mediumSize = 8 * 1024
const largeSize = 64 * 1024
func (r *BytesToBufferReader) readSmall() (MultiBuffer, error) { func (r *BytesToBufferReader) readSmall() (MultiBuffer, error) {
b := New() b := New()
err := b.Reset(ReadFrom(r.Reader)) err := b.Reset(ReadFrom(r.Reader))
if b.IsFull() { if b.IsFull() {
r.buffer = make([]byte, 32*1024) r.buffer = make([]byte, mediumSize)
} }
if !b.IsEmpty() { if !b.IsEmpty() {
return NewMultiBufferValue(b), nil return NewMultiBufferValue(b), nil
@ -41,6 +44,9 @@ func (r *BytesToBufferReader) ReadMultiBuffer() (MultiBuffer, error) {
if nBytes > 0 { if nBytes > 0 {
mb := NewMultiBufferCap(nBytes/Size + 1) mb := NewMultiBufferCap(nBytes/Size + 1)
mb.Write(r.buffer[:nBytes]) mb.Write(r.buffer[:nBytes])
if nBytes == len(r.buffer) && len(r.buffer) == mediumSize {
r.buffer = make([]byte, largeSize)
}
return mb, nil return mb, nil
} }
return nil, err return nil, err

6
common/buf/reader_test.go
View File

@ -21,7 +21,11 @@ func TestAdaptiveReader(t *testing.T) {
b, err = reader.ReadMultiBuffer() b, err = reader.ReadMultiBuffer()
assert(err, IsNil) assert(err, IsNil)
assert(b.Len(), Equals, 32*1024) assert(b.Len(), Equals, 8*1024)
b, err = reader.ReadMultiBuffer()
assert(err, IsNil)
assert(b.Len(), Equals, 64*1024)
} }
func TestBytesReaderWriteTo(t *testing.T) { func TestBytesReaderWriteTo(t *testing.T) {