1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-12-27 12:35:21 -05:00

get rid off ioutil

This commit is contained in:
Darien Raymond
2017-11-04 01:33:28 +01:00
parent ba16987a07
commit 4412d73b28
2 changed files with 43 additions and 4 deletions

View File

@@ -1,6 +1,11 @@
package buf
import "net"
import (
"io"
"net"
"v2ray.com/core/common/errors"
)
type MultiBufferWriter interface {
WriteMultiBuffer(MultiBuffer) error
@@ -10,6 +15,36 @@ type MultiBufferReader interface {
ReadMultiBuffer() (MultiBuffer, error)
}
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
}
// MultiBuffer is a list of Buffers. The order of Buffer matters.
type MultiBuffer []*Buffer