mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-20 22:51:35 +03:00
50 lines
840 B
Go
50 lines
840 B
Go
package tint
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
type buffer []byte
|
|
|
|
var bufPool = sync.Pool{
|
|
New: func() any {
|
|
b := make(buffer, 0, 1024)
|
|
return (*buffer)(&b)
|
|
},
|
|
}
|
|
|
|
func newBuffer() *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) Write(bytes []byte) (int, error) {
|
|
*b = append(*b, bytes...)
|
|
return len(bytes), nil
|
|
}
|
|
|
|
func (b *buffer) WriteByte(char byte) error {
|
|
*b = append(*b, char)
|
|
return nil
|
|
}
|
|
|
|
func (b *buffer) WriteString(str string) (int, error) {
|
|
*b = append(*b, str...)
|
|
return len(str), nil
|
|
}
|
|
|
|
func (b *buffer) WriteStringIf(ok bool, str string) (int, error) {
|
|
if !ok {
|
|
return 0, nil
|
|
}
|
|
return b.WriteString(str)
|
|
}
|