You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
705 B
43 lines
705 B
11 months ago
|
package buffer
|
||
|
|
|
||
|
|
import "sync"
|
||
|
|
|
||
|
|
const bufferSize = 1024
|
||
|
|
|
||
|
|
type Buffer []byte
|
||
|
|
|
||
|
|
// Having an initial size gives a dramatic speedup.
|
||
|
|
//
|
||
|
|
//nolint:gochecknoglobals
|
||
|
|
var bufPool = sync.Pool{
|
||
|
|
New: func() any {
|
||
|
|
b := make([]byte, 0, bufferSize)
|
||
|
|
|
||
|
|
return (*Buffer)(&b)
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
//nolint:forcetypeassert
|
||
|
|
func New() *Buffer {
|
||
|
|
return bufPool.Get().(*Buffer)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) Free() {
|
||
|
|
// To reduce peak allocation, return only smaller buffers to the pool.
|
||
|
|
const maxBufferSize = 16 << 10
|
||
|
|
if cap(*b) <= maxBufferSize {
|
||
|
|
*b = (*b)[:0]
|
||
|
|
bufPool.Put(b)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) WriteString(s string) (int, error) {
|
||
|
|
*b = append(*b, s...)
|
||
|
|
|
||
|
|
return len(s), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) String() string {
|
||
|
|
return string(*b)
|
||
|
|
}
|