Revert "chore: Shrink allocator pool range"
This reverts commit 3c088b33a2
.
This commit is contained in:
parent
a974e810c2
commit
8f61b0e180
2 changed files with 57 additions and 32 deletions
|
@ -12,7 +12,7 @@ var defaultAllocator = NewAllocator()
|
||||||
|
|
||||||
// Allocator for incoming frames, optimized to prevent overwriting after zeroing
|
// Allocator for incoming frames, optimized to prevent overwriting after zeroing
|
||||||
type Allocator struct {
|
type Allocator struct {
|
||||||
buffers [11]sync.Pool
|
buffers [17]sync.Pool
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewAllocator initiates a []byte allocator for frames less than 65536 bytes,
|
// NewAllocator initiates a []byte allocator for frames less than 65536 bytes,
|
||||||
|
@ -20,7 +20,13 @@ type Allocator struct {
|
||||||
// no more than 50%.
|
// no more than 50%.
|
||||||
func NewAllocator() *Allocator {
|
func NewAllocator() *Allocator {
|
||||||
return &Allocator{
|
return &Allocator{
|
||||||
buffers: [...]sync.Pool{ // 64B -> 64K
|
buffers: [...]sync.Pool{ // 1B -> 64K
|
||||||
|
{New: func() any { return new([1]byte) }},
|
||||||
|
{New: func() any { return new([1 << 1]byte) }},
|
||||||
|
{New: func() any { return new([1 << 2]byte) }},
|
||||||
|
{New: func() any { return new([1 << 3]byte) }},
|
||||||
|
{New: func() any { return new([1 << 4]byte) }},
|
||||||
|
{New: func() any { return new([1 << 5]byte) }},
|
||||||
{New: func() any { return new([1 << 6]byte) }},
|
{New: func() any { return new([1 << 6]byte) }},
|
||||||
{New: func() any { return new([1 << 7]byte) }},
|
{New: func() any { return new([1 << 7]byte) }},
|
||||||
{New: func() any { return new([1 << 8]byte) }},
|
{New: func() any { return new([1 << 8]byte) }},
|
||||||
|
@ -46,38 +52,46 @@ func (alloc *Allocator) Get(size int) []byte {
|
||||||
case size > 65536:
|
case size > 65536:
|
||||||
return make([]byte, size)
|
return make([]byte, size)
|
||||||
default:
|
default:
|
||||||
var index uint16
|
index := msb(size)
|
||||||
if size > 64 {
|
|
||||||
index = msb(size)
|
|
||||||
if size != 1<<index {
|
if size != 1<<index {
|
||||||
index += 1
|
index += 1
|
||||||
}
|
}
|
||||||
index -= 6
|
|
||||||
}
|
|
||||||
|
|
||||||
buffer := alloc.buffers[index].Get()
|
buffer := alloc.buffers[index].Get()
|
||||||
switch index {
|
switch index {
|
||||||
case 0:
|
case 0:
|
||||||
return buffer.(*[1 << 6]byte)[:size]
|
return buffer.(*[1]byte)[:size]
|
||||||
case 1:
|
case 1:
|
||||||
return buffer.(*[1 << 7]byte)[:size]
|
return buffer.(*[1 << 1]byte)[:size]
|
||||||
case 2:
|
case 2:
|
||||||
return buffer.(*[1 << 8]byte)[:size]
|
return buffer.(*[1 << 2]byte)[:size]
|
||||||
case 3:
|
case 3:
|
||||||
return buffer.(*[1 << 9]byte)[:size]
|
return buffer.(*[1 << 3]byte)[:size]
|
||||||
case 4:
|
case 4:
|
||||||
return buffer.(*[1 << 10]byte)[:size]
|
return buffer.(*[1 << 4]byte)[:size]
|
||||||
case 5:
|
case 5:
|
||||||
return buffer.(*[1 << 11]byte)[:size]
|
return buffer.(*[1 << 5]byte)[:size]
|
||||||
case 6:
|
case 6:
|
||||||
return buffer.(*[1 << 12]byte)[:size]
|
return buffer.(*[1 << 6]byte)[:size]
|
||||||
case 7:
|
case 7:
|
||||||
return buffer.(*[1 << 13]byte)[:size]
|
return buffer.(*[1 << 7]byte)[:size]
|
||||||
case 8:
|
case 8:
|
||||||
return buffer.(*[1 << 14]byte)[:size]
|
return buffer.(*[1 << 8]byte)[:size]
|
||||||
case 9:
|
case 9:
|
||||||
return buffer.(*[1 << 15]byte)[:size]
|
return buffer.(*[1 << 9]byte)[:size]
|
||||||
case 10:
|
case 10:
|
||||||
|
return buffer.(*[1 << 10]byte)[:size]
|
||||||
|
case 11:
|
||||||
|
return buffer.(*[1 << 11]byte)[:size]
|
||||||
|
case 12:
|
||||||
|
return buffer.(*[1 << 12]byte)[:size]
|
||||||
|
case 13:
|
||||||
|
return buffer.(*[1 << 13]byte)[:size]
|
||||||
|
case 14:
|
||||||
|
return buffer.(*[1 << 14]byte)[:size]
|
||||||
|
case 15:
|
||||||
|
return buffer.(*[1 << 15]byte)[:size]
|
||||||
|
case 16:
|
||||||
return buffer.(*[1 << 16]byte)[:size]
|
return buffer.(*[1 << 16]byte)[:size]
|
||||||
default:
|
default:
|
||||||
panic("invalid pool index")
|
panic("invalid pool index")
|
||||||
|
@ -96,33 +110,44 @@ func (alloc *Allocator) Put(buf []byte) error {
|
||||||
if cap(buf) != 1<<bits {
|
if cap(buf) != 1<<bits {
|
||||||
return errors.New("allocator Put() incorrect buffer size")
|
return errors.New("allocator Put() incorrect buffer size")
|
||||||
}
|
}
|
||||||
bits -= 6
|
|
||||||
buf = buf[:cap(buf)]
|
buf = buf[:cap(buf)]
|
||||||
|
|
||||||
//nolint
|
//nolint
|
||||||
//lint:ignore SA6002 ignore temporarily
|
//lint:ignore SA6002 ignore temporarily
|
||||||
switch bits {
|
switch bits {
|
||||||
case 0:
|
case 0:
|
||||||
alloc.buffers[bits].Put((*[1 << 6]byte)(buf))
|
alloc.buffers[bits].Put((*[1]byte)(buf))
|
||||||
case 1:
|
case 1:
|
||||||
alloc.buffers[bits].Put((*[1 << 7]byte)(buf))
|
alloc.buffers[bits].Put((*[1 << 1]byte)(buf))
|
||||||
case 2:
|
case 2:
|
||||||
alloc.buffers[bits].Put((*[1 << 8]byte)(buf))
|
alloc.buffers[bits].Put((*[1 << 2]byte)(buf))
|
||||||
case 3:
|
case 3:
|
||||||
alloc.buffers[bits].Put((*[1 << 9]byte)(buf))
|
alloc.buffers[bits].Put((*[1 << 3]byte)(buf))
|
||||||
case 4:
|
case 4:
|
||||||
alloc.buffers[bits].Put((*[1 << 10]byte)(buf))
|
alloc.buffers[bits].Put((*[1 << 4]byte)(buf))
|
||||||
case 5:
|
case 5:
|
||||||
alloc.buffers[bits].Put((*[1 << 11]byte)(buf))
|
alloc.buffers[bits].Put((*[1 << 5]byte)(buf))
|
||||||
case 6:
|
case 6:
|
||||||
alloc.buffers[bits].Put((*[1 << 12]byte)(buf))
|
alloc.buffers[bits].Put((*[1 << 6]byte)(buf))
|
||||||
case 7:
|
case 7:
|
||||||
alloc.buffers[bits].Put((*[1 << 13]byte)(buf))
|
alloc.buffers[bits].Put((*[1 << 7]byte)(buf))
|
||||||
case 8:
|
case 8:
|
||||||
alloc.buffers[bits].Put((*[1 << 14]byte)(buf))
|
alloc.buffers[bits].Put((*[1 << 8]byte)(buf))
|
||||||
case 9:
|
case 9:
|
||||||
alloc.buffers[bits].Put((*[1 << 15]byte)(buf))
|
alloc.buffers[bits].Put((*[1 << 9]byte)(buf))
|
||||||
case 10:
|
case 10:
|
||||||
|
alloc.buffers[bits].Put((*[1 << 10]byte)(buf))
|
||||||
|
case 11:
|
||||||
|
alloc.buffers[bits].Put((*[1 << 11]byte)(buf))
|
||||||
|
case 12:
|
||||||
|
alloc.buffers[bits].Put((*[1 << 12]byte)(buf))
|
||||||
|
case 13:
|
||||||
|
alloc.buffers[bits].Put((*[1 << 13]byte)(buf))
|
||||||
|
case 14:
|
||||||
|
alloc.buffers[bits].Put((*[1 << 14]byte)(buf))
|
||||||
|
case 15:
|
||||||
|
alloc.buffers[bits].Put((*[1 << 15]byte)(buf))
|
||||||
|
case 16:
|
||||||
alloc.buffers[bits].Put((*[1 << 16]byte)(buf))
|
alloc.buffers[bits].Put((*[1 << 16]byte)(buf))
|
||||||
default:
|
default:
|
||||||
panic("invalid pool index")
|
panic("invalid pool index")
|
||||||
|
|
|
@ -13,8 +13,8 @@ func TestAllocGet(t *testing.T) {
|
||||||
assert.Equal(t, 1, len(alloc.Get(1)))
|
assert.Equal(t, 1, len(alloc.Get(1)))
|
||||||
assert.Equal(t, 2, len(alloc.Get(2)))
|
assert.Equal(t, 2, len(alloc.Get(2)))
|
||||||
assert.Equal(t, 3, len(alloc.Get(3)))
|
assert.Equal(t, 3, len(alloc.Get(3)))
|
||||||
assert.Equal(t, 64, cap(alloc.Get(3)))
|
assert.Equal(t, 4, cap(alloc.Get(3)))
|
||||||
assert.Equal(t, 64, cap(alloc.Get(4)))
|
assert.Equal(t, 4, cap(alloc.Get(4)))
|
||||||
assert.Equal(t, 1023, len(alloc.Get(1023)))
|
assert.Equal(t, 1023, len(alloc.Get(1023)))
|
||||||
assert.Equal(t, 1024, cap(alloc.Get(1023)))
|
assert.Equal(t, 1024, cap(alloc.Get(1023)))
|
||||||
assert.Equal(t, 1024, len(alloc.Get(1024)))
|
assert.Equal(t, 1024, len(alloc.Get(1024)))
|
||||||
|
|
Loading…
Reference in a new issue