fix: withRecover cannot recover panic (#1169)

* fix: withRecover cannot recover panic * fix: withRecover cannot recover panic
apache · Oct 15, 2024 · ac4e3d6 · ac4e3d6
1 parent 9462ea8
commit ac4e3d6
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/primitive/base.go b/primitive/base.go
@@ -85,14 +85,22 @@ func verifyIP(ip string) error {
 	return nil
 }
 
-var PanicHandler func(interface{})
+type PanicHandler func(interface{})
 
-func WithRecover(fn func()) {
+func DefaultPanicHandler(interface{}) {
+	return
+}
+
+func WithRecover(fn func(), handlers ...PanicHandler) {
 	defer func() {
-		handler := PanicHandler
-		if handler != nil {
-			if err := recover(); err != nil {
-				handler(err)
+		if len(handlers) == 0 {
+			handlers = append(handlers, DefaultPanicHandler)
+		}
+		for _, handler := range handlers {
+			if handler != nil {
+				if err := recover(); err != nil {
+					handler(err)
+				}
 			}
 		}
 	}()