pull/314/head
Darien Raymond 2016-12-06 17:26:06 +01:00
parent e756222a7d
commit 7a641749a8
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
1 changed files with 10 additions and 1 deletions

View File

@ -9,7 +9,7 @@ const (
defaultOffset = 16 defaultOffset = 16
) )
// A Writer that writes contents into the given buffer. // BytesWriter is a writer that writes contents into the given buffer.
type BytesWriter func([]byte) int type BytesWriter func([]byte) int
// Buffer is a recyclable allocation of a byte array. Buffer.Release() recycles // Buffer is a recyclable allocation of a byte array. Buffer.Release() recycles
@ -23,6 +23,7 @@ type Buffer struct {
end int end int
} }
// CreateBuffer creates a new Buffer object based on given container and parent pool.
func CreateBuffer(container []byte, parent Pool) *Buffer { func CreateBuffer(container []byte, parent Pool) *Buffer {
b := new(Buffer) b := new(Buffer)
b.head = container b.head = container
@ -68,6 +69,7 @@ func (b *Buffer) Append(data []byte) {
b.end += nBytes b.end += nBytes
} }
// AppendFunc appends the content of a BytesWriter to the buffer.
func (b *Buffer) AppendFunc(writer BytesWriter) { func (b *Buffer) AppendFunc(writer BytesWriter) {
nBytes := writer(b.head[b.end:]) nBytes := writer(b.head[b.end:])
b.end += nBytes b.end += nBytes
@ -80,6 +82,7 @@ func (b *Buffer) Prepend(data []byte) {
copy(b.head[b.start:], data) copy(b.head[b.start:], data)
} }
// PrependBytes prepends all data in front of the buffer.
func (b *Buffer) PrependBytes(data ...byte) { func (b *Buffer) PrependBytes(data ...byte) {
b.Prepend(data) b.Prepend(data)
} }
@ -89,10 +92,12 @@ func (b *Buffer) PrependFunc(offset int, writer BytesWriter) {
writer(b.head[b.start:]) writer(b.head[b.start:])
} }
// Byte returns the bytes at index.
func (b *Buffer) Byte(index int) byte { func (b *Buffer) Byte(index int) byte {
return b.head[b.start+index] return b.head[b.start+index]
} }
// SetByte sets the byte value at index.
func (b *Buffer) SetByte(index int, value byte) { func (b *Buffer) SetByte(index int, value byte) {
b.head[b.start+index] = value b.head[b.start+index] = value
} }
@ -102,6 +107,7 @@ func (b *Buffer) Bytes() []byte {
return b.head[b.start:b.end] return b.head[b.start:b.end]
} }
// BytesRange returns a slice of this buffer with given from and to bounary.
func (b *Buffer) BytesRange(from, to int) []byte { func (b *Buffer) BytesRange(from, to int) []byte {
if from < 0 { if from < 0 {
from += b.Len() from += b.Len()
@ -166,6 +172,7 @@ func (b *Buffer) Len() int {
return b.end - b.start return b.end - b.start
} }
// IsEmpty returns true if the buffer is empty.
func (b *Buffer) IsEmpty() bool { func (b *Buffer) IsEmpty() bool {
return b.Len() == 0 return b.Len() == 0
} }
@ -217,10 +224,12 @@ func NewBuffer() *Buffer {
return mediumPool.Allocate() return mediumPool.Allocate()
} }
// NewSmallBuffer returns a buffer with 2K bytes capacity.
func NewSmallBuffer() *Buffer { func NewSmallBuffer() *Buffer {
return smallPool.Allocate() return smallPool.Allocate()
} }
// NewLocalBuffer creates and returns a buffer on current thread.
func NewLocalBuffer(size int) *Buffer { func NewLocalBuffer(size int) *Buffer {
return CreateBuffer(make([]byte, size), nil) return CreateBuffer(make([]byte, size), nil)
} }