add arpc.HandleFree()

lesismal · Aug 6, 2024 · d9ed04b · d9ed04b
1 parent b9bb0b6
commit d9ed04b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/handler.go b/handler.go
@@ -1117,6 +1117,11 @@ func HandleMalloc(f func(int) []byte) {
 	DefaultHandler.HandleMalloc(f)
 }
 
+// HandleFree registers buffer releaser.
+func HandleFree(f func([]byte)) {
+	DefaultHandler.HandleFree(f)
+}
+
 // EnablePool registers handlers for pool operation for Context and Message and Message.Buffer
 func EnablePool(enable bool) {
 	DefaultHandler.EnablePool(enable)

diff --git a/handler_test.go b/handler_test.go
@@ -170,5 +170,6 @@ func TestSetHandler(t *testing.T) {
 	Handle("nothing", func(*Context) {}, true)
 	HandleNotFound(func(*Context) {})
 	HandleMalloc(func(int) []byte { return nil })
+	HandleFree(func([]byte) {})
 	SetHandler(d)
 }