package pool // Inspired by https://github.com/xtaci/smux/blob/master/alloc.go import ( "errors" "math/bits" "sync" ) var defaultAllocator = NewAllocator() // Allocator for incoming frames, optimized to prevent overwriting after zeroing type Allocator struct { buffers []sync.Pool } // NewAllocator initiates a []byte allocator for frames less than 65536 bytes, // the waste(memory fragmentation) of space allocation is guaranteed to be // no more than 50%. func NewAllocator() *Allocator { alloc := new(Allocator) alloc.buffers = make([]sync.Pool, 17) // 1B -> 64K for k := range alloc.buffers { i := k alloc.buffers[k].New = func() interface{} { return make([]byte, 1< 65536 { return nil } bits := msb(size) if size == 1< 65536 || cap(buf) != 1<