Be able to set nil

Author: sunfmin

example_test.go

@@ -303,7 +303,7 @@ func ExampleSet_7notexists() {
 
 	fmt.Println(err)
 	//Output:
-	// no such field.
+	// no such field
 }
 
 

go.mod

@@ -0,0 +1,3 @@
+module github.com/sunfmin/reflectutils
+
+go 1.17

map_slice_test.go

@@ -71,3 +71,60 @@ func TestSlices(t *testing.T) {
 	}
 
 }
+
+func TestMapSetNil(t *testing.T) {
+	m := make(map[string]int)
+	Set(&m, "", nil)
+	if m != nil {
+		t.Errorf("got non-nil (%p), want nil", m)
+	}
+}
+
+func TestSliceSetNil(t *testing.T) {
+	m := []int{1}
+	err := Set(&m, "", nil)
+	if err != nil {
+		panic(err)
+	}
+	if m != nil {
+		t.Errorf("got non-nil (%p), want nil", m)
+	}
+
+}
+
+func TestStructSetNil(t *testing.T) {
+	type S struct {
+		Val string
+	}
+	m := &S{Val: "123"}
+	err := Set(&m, "", nil)
+	if err != nil {
+		panic(err)
+	}
+
+
+	if m != nil {
+		t.Errorf("got non-nil (%#+v), want nil", m)
+	}
+}
+
+func TestSetFieldNil(t *testing.T) {
+
+	type S struct {
+		Value []int
+	}
+
+	var s = &S{
+		Value: []int{1, 2},
+	}
+
+	err := Set(s, "Value", nil)
+	if err != nil {
+		panic(err)
+	}
+
+	if s.Value != nil {
+		panic("s.Value is not nil")
+	}
+
+}

set.go

@@ -63,6 +63,12 @@ func Set(i interface{}, name string, value interface{}) (err error) {
 				return
 			}
 		default:
+			if value == nil {
+				vm := reflect.ValueOf(i)
+				vm.Elem().Set(reflect.Zero(vm.Elem().Type()))
+				return
+			}
+
 			valv := reflect.ValueOf(value)
 			for valv.Kind() == reflect.Ptr {
 				valv = valv.Elem()