No panic with reflectutils

sunfmin · Sep 11, 2021 · 63fb18f · 63fb18f
1 parent cdeab1a
commit 63fb18f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/get.go b/get.go
@@ -1,6 +1,7 @@
 package reflectutils
 
 import (
+	"errors"
 	"fmt"
 	"reflect"
 	"strings"
@@ -30,6 +31,11 @@ func IsNil(i interface{}) bool {
 // Get value of a struct by path using reflect.
 func Get(i interface{}, name string) (value interface{}, err error) {
 	// printv(i, name)
+	defer func() {
+		if r := recover(); r != nil {
+			err = errors.New(fmt.Sprint(r))
+		}
+	}()
 
 	if IsNil(i) {
 		return

diff --git a/set.go b/set.go
@@ -31,6 +31,11 @@ var NoSuchFieldError = errors.New("no such field")
 
 // Set value of a struct by path using reflect.
 func Set(i interface{}, name string, value interface{}) (err error) {
+	defer func() {
+		if r := recover(); r != nil {
+			err = errors.New(fmt.Sprint(r))
+		}
+	}()
 
 	v := reflect.ValueOf(i)