v2ray-core/proxy/http/chan_reader.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
}