Go setPointerFieldToNil()

1 개요

Go setPointerFieldToNil()

package main

import (
	"fmt"
	"reflect"
)

type Person struct {
	Age  int
	Name *string
}

func setPointerFieldToNil(obj interface{}) {
	v := reflect.ValueOf(obj).Elem()
	for i := 0; i < v.NumField(); i++ {
		f := v.Field(i)
		if f.Kind() == reflect.Ptr {
			f.Set(reflect.Zero(f.Type()))
		}
	}
}

func main() {
	name1 := "John"
	p1 := Person{11, &name1}
	fmt.Printf("%+v\n", p1) // {Age:11 Name:0xc000104020}
	setPointerFieldToNil(&p1)
	fmt.Printf("%+v\n", p1) // {Age:11 Name:<nil>}
}

package main

import (
	"fmt"
	"reflect"
)

type Person struct {
	Age  int
	Name *string
}

type Response struct {
	Status string
	Data1  Person
	Data2  *Person
}

func setPointerFieldToNil(s interface{}) {
	v := reflect.ValueOf(s).Elem()
	for i := 0; i < v.NumField(); i++ {
		f := v.Field(i)
		switch f.Kind() {
		case reflect.Struct:
			setPointerFieldToNil(f.Addr().Interface())
		case reflect.Ptr:
			f.Set(reflect.Zero(f.Type()))
		}
	}
}

func main() {
	name1 := "John"
	person1 := Person{11, &name1}
	r1 := Response{Status: "success", Data1: person1, Data2: &person1}
	fmt.Printf("%#v\n", r1)
	setPointerFieldToNil(&r1)
	fmt.Printf("%#v\n", r1)
}

2 같이 보기