go reflect详解

目录

---

本文操作对象定义如下

type User struct {
	Name   string
	Age    int
	Height int
}

func (u *User) Hello(msg string) {
	fmt.Printf("My name is %s, age %v, msg:%v\n", u.Name, u.Age, msg)
}

1.基础使用

运行中的程序变量都是由类型和值构成的，通过reflect.TypeOf和reflect.ValueOf方法获取运行时的类型和值信息，如下

// 基础使用
func CheckType(o interface{}) {
	t := reflect.TypeOf(o)

	switch t.Kind() {
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
		fmt.Println("Int number")
	case reflect.Float64, reflect.Float32:
		fmt.Println("Float number")
	case reflect.String:
		fmt.Println("String")
	case reflect.Slice:
		fmt.Println("Slice")
	case reflect.Map:
		fmt.Println("Map")
	case reflect.Struct:
		fmt.Println("Struct")
	default:
		fmt.Printf("Unknown %v\n", t)
	}
}

func Info(o interface{}) {
	//反射访问类型和值
	t := reflect.TypeOf(o)
	fmt.Printf("%v\n", t)

	v := reflect.ValueOf(o)
	fmt.Printf("%v\n", v)

	// Kind 具体分类
	if k := t.Kind(); k != reflect.Struct {
		fmt.Println("Bad Input Type")
		return
	}

	// 遍历输出类型和值
	for i := 0; i < t.NumField(); i++ {
		f := t.Field(i)
		fv := v.Field(i).Interface()
		fmt.Printf("%6s %v=%v\n", f.Name, f.Type, fv)
	}
}

func TestBase(t *testing.T) {
	u := User{Age: 16, Name: "wenzhou", Height: 18}
	Info(u)
	CheckType(u)
	CheckType(&u)
}

输出如下

reflect_test.User
{wenzhou 16 18}
  Name string=wenzhou
   Age int=16
Height int=18
Struct
Unknown *reflect_test.User

1.这里通过NumFiield遍历结构体

2.这里的Kind 和 Type的差别在于：前者是常用的golang数据类型定义(int/string/struct等)，后者是程序定义的具体文件中的具体类型名称(如reflect_test.User)

2.结构体字段读取

 结构体可以通过FieldByName来访问，也可以通过数组下标访问对应StructField，如下

// 结构的字段反射访问
func Info2(o interface{}) {
	t2 := reflect.TypeOf(o)
	v2 := reflect.ValueOf(o)

	fmt.Printf("%v \n", t2.Field(0))
	fmt.Println(v2.FieldByName("Name"))
}

func TestReflectStructField(t *testing.T) {
	u := User{Age: 16, Name: "wenzhou", Height: 18}
	Info2(u)
}

 输出

{Name  string  0 [0] false} 
wenzhou

3.方法调用

使用MethodByName 实现方法调用，注意参数以数组形式传入，如下

// 使用value的method调用
func Info4(o interface{}) {
	v := reflect.ValueOf(o)

	m := v.MethodByName("Hello")
	args := []reflect.Value{reflect.ValueOf("testall")}
	m.Call(args)
}

func TestReflectCall(t *testing.T) {
	u := User{Age: 16, Name: "wenzhou", Height: 18}
	Info4(&u)
}

4.设置值

设置值必须传入指针类型，具体的值设置需要通过Elem取得对应值类型才能设置，参考如下

// 设置值必须是pointer interface， 此时用Elem取得指针的值来操作
func Info3(o interface{}) {
	x := 3
	v1 := reflect.ValueOf(&x) //注意ptr
	v1.Elem().SetInt(100)
	fmt.Println("x=", x)

	//完整预判过程
	v2 := reflect.ValueOf(o)

	if k := v2.Kind(); k != reflect.Ptr || !v2.Elem().CanSet() {
		fmt.Println("bad input")
		return
	}

	//取出指针指向对象值，遍历或者查找名字来设置
	v2 = v2.Elem()

	n := v2.FieldByName("Name")
	if !n.IsValid() {
		fmt.Println("No Name field")
		return
	}

	if n.Kind() != reflect.String {
		fmt.Println("No Name string")
		return
	}

	n.SetString("wenwen")
	v2.Field(1).SetInt(2)
}

本文主要讲解reflect基础和常用方法，下文会讲解常见的应用场景。