1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2026-05-16 09:29:07 -04:00

reorg chan reader

This commit is contained in:
v2ray
2016-05-25 22:36:52 +02:00
parent e610faaff6
commit 46f76e55e5
2 changed files with 26 additions and 8 deletions

View File

@@ -1,48 +0,0 @@
package http
import (
"io"
"github.com/v2ray/v2ray-core/common/alloc"
v2io "github.com/v2ray/v2ray-core/common/io"
)
type ChanReader struct {
stream v2io.Reader
current *alloc.Buffer
eof bool
}
func NewChanReader(stream v2io.Reader) *ChanReader {
this := &ChanReader{
stream: stream,
}
this.fill()
return this
}
func (this *ChanReader) fill() {
b, err := this.stream.Read()
this.current = b
if err != nil {
this.eof = true
this.current = nil
}
}
func (this *ChanReader) Read(b []byte) (int, error) {
if this.current == nil {
this.fill()
if this.eof {
return 0, io.EOF
}
}
nBytes := copy(b, this.current.Value)
if nBytes == this.current.Len() {
this.current.Release()
this.current = nil
} else {
this.current.SliceFrom(nBytes)
}
return nBytes, nil
}

View File

@@ -248,7 +248,7 @@ func (this *HttpProxyServer) handlePlainHTTP(request *http.Request, dest v2net.D
finish.Add(1)
go func() {
defer finish.Done()
responseReader := bufio.NewReader(NewChanReader(ray.InboundOutput()))
responseReader := bufio.NewReader(v2io.NewChanReader(ray.InboundOutput()))
response, err := http.ReadResponse(responseReader, request)
if err != nil {
log.Warning("HTTP: Failed to read response: ", err)