1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-22 18:17:52 -05:00
v2fly/common/net/transport.go

58 lines
1.3 KiB
Go
Raw Normal View History

2015-09-13 14:01:50 -04:00
package net
import (
"io"
"github.com/v2ray/v2ray-core/common/alloc"
)
2015-10-11 08:46:12 -04:00
// ReadFrom reads from a reader and put all content to a buffer.
// If buffer is nil, ReadFrom creates a new normal buffer.
func ReadFrom(reader io.Reader, buffer *alloc.Buffer) (*alloc.Buffer, error) {
if buffer == nil {
buffer = alloc.NewBuffer()
}
nBytes, err := reader.Read(buffer.Value)
buffer.Slice(0, nBytes)
return buffer, err
}
2015-09-21 06:15:25 -04:00
// ReaderToChan dumps all content from a given reader to a chan by constantly reading it until EOF.
func ReaderToChan(stream chan<- *alloc.Buffer, reader io.Reader) error {
allocate := alloc.NewBuffer
large := false
2015-09-13 14:01:50 -04:00
for {
buffer, err := ReadFrom(reader, allocate())
if buffer.Len() > 0 {
stream <- buffer
} else {
buffer.Release()
2015-09-14 12:19:17 -04:00
}
2015-09-13 14:01:50 -04:00
if err != nil {
return err
}
if buffer.IsFull() && !large {
allocate = alloc.NewLargeBuffer
large = true
} else if !buffer.IsFull() {
allocate = alloc.NewBuffer
large = false
}
2015-09-13 14:01:50 -04:00
}
}
2015-09-21 06:15:25 -04:00
// ChanToWriter dumps all content from a given chan to a writer until the chan is closed.
func ChanToWriter(writer io.Writer, stream <-chan *alloc.Buffer) error {
2015-09-13 14:01:50 -04:00
for buffer := range stream {
nBytes, err := writer.Write(buffer.Value)
2015-10-09 10:48:11 -04:00
if nBytes < buffer.Len() {
_, err = writer.Write(buffer.Value[nBytes:])
}
buffer.Release()
2015-09-13 14:01:50 -04:00
if err != nil {
return err
}
}
return nil
}