feat: optimize performance of allocator helpers

joetifa2003 · joetifa2003 · commit ea9dfd7fcd17 · 2024-09-08T16:21:56.000+02:00
diff --git a/allocator/allocator.go b/allocator/allocator.go
@@ -49,21 +49,21 @@ func getSize[T any]() int {
 
 // Alloc allocates T and returns a pointer to it.
 func Alloc[T any](a Allocator) *T {
-	ptr := a.Alloc(getSize[T]())
+	ptr := a.alloc(a.allocator, getSize[T]())
 	return (*T)(unsafe.Pointer(ptr))
 }
 
 // FreeMany frees memory allocated by Alloc takes a ptr
 // CAUTION: be careful not to double free, and prefer using defer to deallocate
 func Free[T any](a Allocator, ptr *T) {
-	a.Free(unsafe.Pointer(ptr))
+	a.free(a.allocator, unsafe.Pointer(ptr))
 }
 
 // AllocMany allocates n of T and returns a slice representing the heap.
 // CAUTION: don't append to the slice, the purpose of it is to replace pointer
 // arithmetic with slice indexing
 func AllocMany[T any](a Allocator, n int) []T {
-	ptr := a.Alloc(getSize[T]() * n)
+	ptr := a.alloc(a.allocator, getSize[T]()*n)
 	return unsafe.Slice(
 		(*T)(ptr),
 		n,
@@ -73,12 +73,12 @@ func AllocMany[T any](a Allocator, n int) []T {
 // FreeMany frees memory allocated by AllocMany takes in the slice (aka the heap)
 // CAUTION: be careful not to double free, and prefer using defer to deallocate
 func FreeMany[T any](a Allocator, slice []T) {
-	a.Free(unsafe.Pointer(&slice[0]))
+	a.free(a.allocator, unsafe.Pointer(&slice[0]))
 }
 
 // Realloc reallocates memory allocated with AllocMany and doesn't change underling data
 func Realloc[T any](a Allocator, slice []T, newN int) []T {
-	ptr := a.Realloc(unsafe.Pointer(&slice[0]), getSize[T]()*newN)
+	ptr := a.realloc(a.allocator, unsafe.Pointer(&slice[0]), getSize[T]()*newN)
 	return unsafe.Slice(
 		(*T)(ptr),
 		newN,
