feat: add OnGuard

Author: yankeguo

guard.go

@@ -4,9 +4,15 @@ import (
 	"fmt"
 )
 
+// OnGuard is a global hook for Guard
+var OnGuard func(r any)
+
 // Guard recover from panic and set err
 func Guard(err *error) {
 	if r := recover(); r != nil {
+		if fn := OnGuard; fn != nil {
+			fn(r)
+		}
 		if re, ok := r.(error); ok {
 			*err = re
 		} else {

guard_test.go

@@ -2,8 +2,9 @@ package rg
 
 import (
 	"errors"
-	"github.com/stretchr/testify/require"
 	"testing"
+
+	"github.com/stretchr/testify/require"
 )
 
 func TestGuard(t *testing.T) {
@@ -16,6 +17,24 @@ func TestGuard(t *testing.T) {
 	require.Equal(t, "hello", err.Error())
 }
 
+func TestOnGuard(t *testing.T) {
+	var err error
+	var err2 any
+	OnGuard = func(e any) {
+		err2 = e
+	}
+	defer func() {
+		OnGuard = nil
+	}()
+	func() {
+		defer Guard(&err)
+		panic(errors.New("hello"))
+	}()
+	require.Error(t, err)
+	require.Equal(t, err, err2)
+	require.Equal(t, "hello", err.Error())
+}
+
 func TestGuardNotError(t *testing.T) {
 	var err error
 	func() {