monibuca/pkg/util/buddy_disable.go

//go:build !enable_buddy

package util

import (
	"sync"
	"unsafe"
)

var pool0, pool1, pool2 sync.Pool

func init() {
	pool0.New = func() any {
		ret := createMemoryAllocator(defaultBufSize)
		ret.recycle = func() {
			pool0.Put(ret)
		}
		return ret
	}
	pool1.New = func() any {
		ret := createMemoryAllocator(1 << MinPowerOf2)
		ret.recycle = func() {
			pool1.Put(ret)
		}
		return ret
	}
	pool2.New = func() any {
		ret := createMemoryAllocator(1 << (MinPowerOf2 + 2))
		ret.recycle = func() {
			pool2.Put(ret)
		}
		return ret
	}
}

func createMemoryAllocator(size int) *MemoryAllocator {
	memory := make([]byte, size)
	ret := &MemoryAllocator{
		allocator: NewAllocator(size),
		Size:      size,
		memory:    memory,
		start:     int64(uintptr(unsafe.Pointer(&memory[0]))),
	}
	ret.allocator.Init(size)
	return ret
}

func GetMemoryAllocator(size int) (ret *MemoryAllocator) {
	switch size {
	case defaultBufSize:
		ret = pool0.Get().(*MemoryAllocator)
		ret.allocator.Init(size)
	case 1 << MinPowerOf2:
		ret = pool1.Get().(*MemoryAllocator)
		ret.allocator.Init(size)
	case 1 << (MinPowerOf2 + 2):
		ret = pool2.Get().(*MemoryAllocator)
		ret.allocator.Init(size)
	default:
		ret = createMemoryAllocator(size)
	}
	return
}