1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-23 02:26:49 -05:00
v2fly/common/buf/multi_buffer.go

174 lines
3.1 KiB
Go
Raw Normal View History

2017-04-15 15:07:23 -04:00
package buf
2017-11-03 20:33:28 -04:00
import (
"io"
"net"
"v2ray.com/core/common/errors"
)
2017-04-15 15:07:23 -04:00
type MultiBufferWriter interface {
2017-04-23 07:41:52 -04:00
WriteMultiBuffer(MultiBuffer) error
2017-04-15 15:07:23 -04:00
}
2017-04-15 16:22:29 -04:00
type MultiBufferReader interface {
ReadMultiBuffer() (MultiBuffer, error)
}
2017-11-03 20:33:28 -04:00
func ReadAllToMultiBuffer(reader io.Reader) (MultiBuffer, error) {
mb := NewMultiBuffer()
for {
b := New()
err := b.AppendSupplier(ReadFrom(reader))
if !b.IsEmpty() {
mb.Append(b)
}
if err != nil {
if errors.Cause(err) == io.EOF {
return mb, nil
}
mb.Release()
return nil, err
}
}
}
func ReadAllToBytes(reader io.Reader) ([]byte, error) {
mb, err := ReadAllToMultiBuffer(reader)
if err != nil {
return nil, err
}
b := make([]byte, mb.Len())
mb.Read(b)
mb.Release()
return b, nil
}
2017-04-23 13:16:56 -04:00
// MultiBuffer is a list of Buffers. The order of Buffer matters.
2017-04-15 15:07:23 -04:00
type MultiBuffer []*Buffer
2017-04-23 13:16:56 -04:00
// NewMultiBuffer creates a new MultiBuffer instance.
2017-04-15 15:07:23 -04:00
func NewMultiBuffer() MultiBuffer {
2017-04-16 16:33:20 -04:00
return MultiBuffer(make([]*Buffer, 0, 128))
2017-04-15 15:07:23 -04:00
}
2017-04-23 13:16:56 -04:00
// NewMultiBufferValue wraps a list of Buffers into MultiBuffer.
2017-04-15 15:07:23 -04:00
func NewMultiBufferValue(b ...*Buffer) MultiBuffer {
return MultiBuffer(b)
}
2017-04-23 13:16:56 -04:00
func (mb *MultiBuffer) Append(buf *Buffer) {
*mb = append(*mb, buf)
2017-04-15 15:07:23 -04:00
}
2017-04-23 13:16:56 -04:00
func (mb *MultiBuffer) AppendMulti(buf MultiBuffer) {
*mb = append(*mb, buf...)
2017-04-15 15:07:23 -04:00
}
2017-04-24 19:56:08 -04:00
func (mb MultiBuffer) Copy(b []byte) int {
total := 0
for _, bb := range mb {
nBytes := copy(b[total:], bb.Bytes())
total += nBytes
if nBytes < bb.Len() {
break
}
}
return total
}
2017-04-15 15:07:23 -04:00
func (mb *MultiBuffer) Read(b []byte) (int, error) {
endIndex := len(*mb)
totalBytes := 0
for i, bb := range *mb {
2017-04-19 15:27:21 -04:00
nBytes, _ := bb.Read(b)
2017-04-15 15:07:23 -04:00
totalBytes += nBytes
b = b[nBytes:]
if bb.IsEmpty() {
bb.Release()
} else {
endIndex = i
break
}
}
*mb = (*mb)[endIndex:]
return totalBytes, nil
}
2017-05-01 18:28:16 -04:00
func (mb *MultiBuffer) Write(b []byte) {
n := len(*mb)
if n > 0 && !(*mb)[n-1].IsFull() {
nBytes, _ := (*mb)[n-1].Write(b)
b = b[nBytes:]
}
for len(b) > 0 {
bb := New()
nBytes, _ := bb.Write(b)
b = b[nBytes:]
mb.Append(bb)
}
}
2017-04-23 13:16:56 -04:00
// Len returns the total number of bytes in the MultiBuffer.
2017-04-15 15:07:23 -04:00
func (mb MultiBuffer) Len() int {
size := 0
for _, b := range mb {
size += b.Len()
}
return size
}
2017-04-23 13:16:56 -04:00
// IsEmpty return true if the MultiBuffer has no content.
2017-04-15 15:07:23 -04:00
func (mb MultiBuffer) IsEmpty() bool {
for _, b := range mb {
if !b.IsEmpty() {
return false
}
}
return true
}
2017-04-23 13:16:56 -04:00
// Release releases all Buffers in the MultiBuffer.
2017-04-15 15:07:23 -04:00
func (mb MultiBuffer) Release() {
2017-04-16 07:12:58 -04:00
for i, b := range mb {
2017-04-15 15:07:23 -04:00
b.Release()
2017-04-16 07:12:58 -04:00
mb[i] = nil
2017-04-15 15:07:23 -04:00
}
}
2017-04-16 16:30:29 -04:00
2017-04-23 13:16:56 -04:00
// ToNetBuffers converts this MultiBuffer to net.Buffers. The return net.Buffers points to the same content of the MultiBuffer.
2017-04-16 16:30:29 -04:00
func (mb MultiBuffer) ToNetBuffers() net.Buffers {
bs := make([][]byte, len(mb))
for i, b := range mb {
bs[i] = b.Bytes()
}
return bs
}
func (mb *MultiBuffer) SliceBySize(size int) MultiBuffer {
slice := NewMultiBuffer()
sliceSize := 0
endIndex := len(*mb)
for i, b := range *mb {
if b.Len()+sliceSize > size {
endIndex = i
break
}
2017-04-19 15:27:21 -04:00
sliceSize += b.Len()
slice.Append(b)
}
*mb = (*mb)[endIndex:]
return slice
}
2017-05-01 18:28:16 -04:00
func (mb *MultiBuffer) SplitFirst() *Buffer {
if len(*mb) == 0 {
return nil
}
b := (*mb)[0]
*mb = (*mb)[1:]
return b
}