1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2026-06-05 18:49:14 -04:00

fix timeout reader usage

This commit is contained in:
Darien Raymond
2017-03-27 11:14:55 +02:00
parent 8ce6e741f1
commit 14824c8451
3 changed files with 12 additions and 6 deletions

View File

@@ -1,13 +1,15 @@
package buf
type MergingReader struct {
reader Reader
leftover *Buffer
reader Reader
timeoutReader TimeoutReader
leftover *Buffer
}
func NewMergingReader(reader Reader) Reader {
return &MergingReader{
reader: reader,
reader: reader,
timeoutReader: reader.(TimeoutReader),
}
}
@@ -25,7 +27,11 @@ func (r *MergingReader) Read() (*Buffer, error) {
return b, nil
}
b2, err := r.reader.Read()
if r.timeoutReader == nil {
return b, nil
}
b2, err := r.timeoutReader.ReadTimeout(0)
if err != nil {
return b, nil
}