mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-09-06 03:54:22 -04:00
48 lines
765 B
Go
48 lines
765 B
Go
package http
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
|
)
|
|
|
|
type ChanReader struct {
|
|
stream <-chan *alloc.Buffer
|
|
current *alloc.Buffer
|
|
eof bool
|
|
}
|
|
|
|
func NewChanReader(stream <-chan *alloc.Buffer) *ChanReader {
|
|
this := &ChanReader{
|
|
stream: stream,
|
|
}
|
|
this.fill()
|
|
return this
|
|
}
|
|
|
|
func (this *ChanReader) fill() {
|
|
b, open := <-this.stream
|
|
this.current = b
|
|
if !open {
|
|
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
|
|
}
|