Golang反射-下篇

共 2263字,需浏览 5分钟

 ·

2021-11-30 04:30

目录

  • 1、判断类型 interface.Type

  • 2、自定义 struct 的反射

  • 3、结构体标签和反射

  • 4、反射调用函数

  • 5、反射调用方法

  • 6、反射创建值

    • 6.1 反射创建 struct

    • 6.2 反射创建 slice

    • 6.3 反射创建 map

  • 7、反射修改值

    • 7.1 反射修改 struct

    • 7.2 反射修改 slice

    • 7.3 反射修改 map



本文是 Golang反射-上篇 的续篇内容,主要介绍反射实际的一些使用

1、判断类型 interface.Type

利用类型断言来判断数据类型的用法如下

package main

import "fmt"

func main()  {
 var s interface{} = "abc"
 switch s.(type) {
 case string:
  fmt.Println("s.type=string")
 case int:
  fmt.Println("s.type=int")
 case bool:
  fmt.Println("s.type=bool")
 default:
  fmt.Println("未知的类型")
 }
}

上述类型判断的问题

  • 类型判断会写很多,代码很长
  • 类型还会增删,不灵活

如果使用反射获取变量内部的信息

  • reflect 包提供 ValueOf 和 TypeOf
  • reflect.ValueOf:获取输入接口中数据的值,如果为空返回 0
  • reflect.TypeOf:获取输入接口中值的类型,如果为空返回 nil
  • TypeOf 能传入所有类型,是因为所有的类型都实现了空接口
package main

import (
 "fmt"
 "reflect"
)

func main()  {
 var s interface{} = "abc"
 //TypeOf会返回目标的对象
 reflectType:=reflect.TypeOf(s)
 reflectValue:=reflect.ValueOf(s)
 fmt.Printf("[typeof:%v]\n", reflectType)  // string
 fmt.Printf("[valueof:%v]\n", reflectValue)  // abc
}

2、自定义 struct 的反射

自定义 struct 的相关操作

  • 对于成员变量

    • 先获取 interface 的 reflect.Type,然后遍历 NumField
    • 再通过 reflect.Type 的 Field 获取字段名及类型
    • 最后通过 Field 的 interface 获取对应的 value
  • 对于方法

    • 先获取 interface 的 reflect.Type,然后遍历 NumMethod
    • 再通过 reflect.Type 的 t.Method 获取真实的方法名
    • 最后通过 Name 和 Type 获取方法的类型和值

注意点

  • 用于对未知类型进行遍历探测其 Field,抽象成一个函数
  • go 语言里面 struct 成员变量小写,在反射的时候直接 panic()
  • 结构体方法名小写是不会 panic 的,反射值也不会被查看到
  • 指针方法是不能被反射查看到的
package main

import (
 "fmt"
 "reflect"
)

type Person struct {
 Name string
 Age  int
}

type Student struct {
 Person     // 匿名结构体嵌套
 StudentId  int
 SchoolName string
 Graduated  bool
 Hobbies    []string
 //panic: reflect.Value.Interface: cannot return value obtained from unexported field or method
 //hobbies    []string
 Label      map[string]string
}

func (s *Student) GoHome() {
 fmt.Printf("回家了,sid:%d\n", s.StudentId)
}

//func (s Student) GoHome() {
// fmt.Printf("回家了,sid:%d\n", s.StudentId)
//}

func (s Student) GotoSchool() {
 fmt.Printf("上学了,sid:%d\n", s.StudentId)
}

func (s *Student) graduated() {
 fmt.Printf("毕业了,sid:%d\n", s.StudentId)
}

//func (s Student) Ggraduated() {
// fmt.Printf("毕业了,sid:%d\n", s.StudentId)
//}

func reflectProbeStruct(s interface{}) {
 // 获取目标对象
 t := reflect.TypeOf(s)
 fmt.Printf("对象的类型名称 %s\n", t.Name())
 // 获取目标对象的值类型
 v := reflect.ValueOf(s)
 // 遍历获取成员变量
 for i := 0; i < t.NumField(); i++ {
  // Field 代表对象的字段名
  key := t.Field(i)
  value := v.Field(i).Interface()
  // 字段
  if key.Anonymous {
   fmt.Printf("匿名字段 第 %d 个字段,字段名 %s, 字段类型 %v, 字段的值 %v\n", i+1, key.Name, key.Type, value)
  } else {
   fmt.Printf("命名字段 第 %d 个字段,字段名 %s, 字段类型 %v, 字段的值 %v\n", i+1, key.Name, key.Type, value)
  }
 }
 // 打印方法
 for i := 0; i < t.NumMethod(); i++ {
  m := t.Method(i)
  fmt.Printf("第 %d 个方法,方法名 %s, 方法类型 %v\n", i+1, m.Name, m.Type)
 }
}

func main() {
 s := Student{
  Person: Person{
   "geek",
   24,
  },
  StudentId:  123,
  SchoolName: "Beijing University",
  Graduated:  true,
  Hobbies:    []string{"唱""跳""Rap"},
  //hobbies:    []string{"唱", "跳", "Rap"},
  Label:      map[string]string{"k1""v1""k2""v2"},
 }
 p := Person{
  Name: "张三",
  Age:  100,
 }
 reflectProbeStruct(s)
 reflectProbeStruct(p)
 /*
 对象的类型名称 Student
 匿名字段 第 1 个字段,字段名 Person, 字段类型 main.Person, 字段的值 {geek 24}
 命名字段 第 2 个字段,字段名 StudentId, 字段类型 int, 字段的值 123
 命名字段 第 3 个字段,字段名 SchoolName, 字段类型 string, 字段的值 Beijing University
 命名字段 第 4 个字段,字段名 Graduated, 字段类型 bool, 字段的值 true
 命名字段 第 5 个字段,字段名 Hobbies, 字段类型 []string, 字段的值 [唱 跳 Rap]
 命名字段 第 6 个字段,字段名 Label, 字段类型 map[string]string, 字段的值 map[k1:v1 k2:v2]
 第 1 个方法,方法名 GotoSchool, 方法类型 func(main.Student)
 对象的类型名称 Person
 命名字段 第 1 个字段,字段名 Name, 字段类型 string, 字段的值 张三
 命名字段 第 2 个字段,字段名 Age, 字段类型 int, 字段的值 100
  */

}

3、结构体标签和反射

  • json 的标签解析出 json
  • yaml 的标签解析出 yaml
  • xorm、gorm 的标签标识数据库 db 字段
  • 自定义标签
  • 原理是 t.Field.Tag.Lookup("标签名")

示例

package main

import (
 "encoding/json"
 "fmt"
 "gopkg.in/yaml.v2"
 "io/ioutil"
)

type Person struct {
 Name string `json:"name" yaml:"yaml_name"`
 Age  int    `json:"age" yaml:"yaml_age"`
 City string `json:"city" yaml:"yaml_city"`
 //City string `json:"-" yaml:"yaml_city"` // 忽略json:"-"
}

// json解析
func jsonWork() {
 // 对象Marshal成字符串
 p := Person{
  Name: "geek",
  Age:  24,
  City: "Beijing",
 }
 data, err := json.Marshal(p)
 if err != nil {
  fmt.Printf("json.marshal.err: %v\n", err)
 }
 fmt.Printf("person.marshal.res: %v\n"string(data))

 // 从字符串解析成结构体
 p2str := `{
 "name": "张三",
 "age": 38,
 "city": "山东"
 }`

 var p2 Person
 err = json.Unmarshal([]byte(p2str), &p2)
 if err != nil {
  fmt.Printf("json.unmarshal.err: %v\n", err)
  return
 }
 fmt.Printf("person.unmarshal.res: %v\n", p2)
}

// yaml解析
func yamlWork() {
 filename := "a.yaml"
 content, err := ioutil.ReadFile(filename)
 if err != nil {
  fmt.Printf("ioutil.ReadFile.err: %v\n", err)
  return
 }
 p := &Person{}
 //err = yaml.Unmarshal([]byte(content), p)
 err = yaml.UnmarshalStrict([]byte(content), p)  // 解析严格,考虑多余字段,忽略字段等
 if err != nil {
  fmt.Printf("yaml.UnmarshalStrict.err: %v\n", err)
  return
 }
 fmt.Printf("yaml.UnmarshalStrict.res: %v\n", p)
}

func main() {
 jsonWork()
 /*
  person.marshal.res: {"name":"geek","age":24,"city":"Beijing"}
  person.unmarshal.res: {张三 38 山东}
 */

 yamlWork()
 /*
  yaml.UnmarshalStrict.res: &{李四 18 Shanghai}
  */

}

解析的 yaml 内容

yaml_name: 李四
yaml_age: 18
yaml_city: Shanghai
  • 自定义标签格式解析
package main

import (
 "fmt"
 "reflect"
)

type Person struct {
 Name string `aa:"name"`
 Age  int    `aa:"age"`
 City string `aa:"city"`
}

// CustomParse 自定义解析
func CustomParse(s interface{}) {
 // TypeOf type类型
 r:=reflect.TypeOf(s)
 value := reflect.ValueOf(s)
 for i:=0;i  field:=r.Field(i)
  key:=field.Name
  if tag, ok:=field.Tag.Lookup("aa");ok{
   if tag == "-"{
    continue
   }
   fmt.Printf("找到了aa标签, key: %v, value: %v, tag: %s\n", key, value.Field(i), tag)
  }
 }
}

func main() {
 p := Person{
  Name: "geek",
  Age:  24,
  City: "Beijing",
 }
 CustomParse(p)
 /*
 找到了aa标签, key: Name, value: geek, tag: name
 找到了aa标签, key: Age, value: 24, tag: age
 找到了aa标签, key: City, value: Beijing, tag: city
  */

}

4、反射调用函数

valueFunc := reflect.ValueOf(Add) //函数也是一种数据类型
typeFunc := reflect.TypeOf(Add)
argNum := typeFunc.NumIn()            //函数输入参数的个数
args := make([]reflect.Value, argNum) //准备函数的输入参数
for i := 0; i < argNum; i++ {
 if typeFunc.In(i).Kind() == reflect.Int {
  args[i] = reflect.ValueOf(3//给每一个参数都赋3
 }
}
sumValue := valueFunc.Call(args) //返回[]reflect.Value,因为go语言的函数返回可能是一个列表
if typeFunc.Out(0).Kind() == reflect.Int {
 sum := sumValue[0].Interface().(int//从Value转为原始数据类型
 fmt.Printf("sum=%d\n", sum)
}

5、反射调用方法

示例

user := User{
 Id:     7,
 Name:   "杰克逊",
 Weight: 65.5,
 Height: 1.68,
}
valueUser := reflect.ValueOf(&user)              //必须传指针,因为BMI()在定义的时候它是指针的方法
bmiMethod := valueUser.MethodByName("BMI")       //MethodByName()通过Name返回类的成员变量
resultValue := bmiMethod.Call([]reflect.Value{}) //无参数时传一个空的切片
result := resultValue[0].Interface().(float32)
fmt.Printf("bmi=%.2f\n", result)

//Think()在定义的时候用的不是指针,valueUser可以用指针也可以不用指针
thinkMethod := valueUser.MethodByName("Think")
thinkMethod.Call([]reflect.Value{})

valueUser2 := reflect.ValueOf(user)
thinkMethod = valueUser2.MethodByName("Think")
thinkMethod.Call([]reflect.Value{})

过程

  • 首先通过 reflect.ValueOf(p1) 获取得到反射类型对象
  • reflect.ValueOf(p1).MethodByName 需 要传入准确的方法名称(名称不对会 panic: reflect: call of reflect.Value.Call on zero Value),MethodByName 代表注册
  • []reflect.Value 这是最终需要调用方法的参数,无参数传空切片
  • call 调用
package main

import (
 "fmt"
 "reflect"
)

type Person struct {
 Name   string
 Age    int
 Gender string
}

func (p Person) ReflectCallFuncWithArgs(name string, age int) {
 fmt.Printf("调用的是带参数的方法, args.name: %s, args.age: %d, p.name: %s, p.age: %d\n",
  name,
  age,
  p.Name,
  p.Age,
 )
}

func (p Person) ReflectCallFuncWithNoArgs() {
 fmt.Printf("调用的是不带参数的方法\n")
}

func main() {
 p1 := Person{
  Name:   "geek",
  Age:    24,
  Gender: "男",
 }
 // 1.首先通过reflect.ValueOf(p1)获取得到反射值类型
 getValue := reflect.ValueOf(p1)
 // 2.带参数的方法调用
 methodValue1 := getValue.MethodByName("ReflectCallFuncWithArgs")
 // 参数是reflect.Value的切片
 args1 := []reflect.Value{reflect.ValueOf("张三"), reflect.ValueOf(30)}
 methodValue1.Call(args1)
 // 3.不带参数的方法调用
 methodValue2 := getValue.MethodByName("ReflectCallFuncWithNoArgs")
 // 参数是reflect.Value的切片
 args2 := make([]reflect.Value, 0)
 methodValue2.Call(args2)
 /*
 调用的是带参数的方法, args.name: 张三, args.age: 30, p.name: geek, p.age: 24
 调用的是不带参数的方法
  */

}

6、反射创建值

6.1 反射创建 struct

t := reflect.TypeOf(User{})
value := reflect.New(t) //根据reflect.Type创建一个对象,得到该对象的指针,再根据指针提到reflect.Value
value.Elem().FieldByName("Id").SetInt(10)
value.Elem().FieldByName("Name").SetString("宋江")
value.Elem().FieldByName("Weight").SetFloat(78.)
value.Elem().FieldByName("Height").SetFloat(168.4)
user := value.Interface().(*User) //把反射类型转成go原始数据类型
fmt.Printf("id=%d name=%s weight=%.1f height=%.1f\n", user.Id, user.Name, user.Weight, user.Height)

6.2 反射创建 slice

var slice []User
sliceType := reflect.TypeOf(slice)
sliceValue := reflect.MakeSlice(sliceType, 13//reflect.MakeMap、reflect.MakeSlice、reflect.MakeChan、reflect.MakeFunc
sliceValue.Index(0).Set(reflect.ValueOf(User{
 Id:     8,
 Name:   "李达",
 Weight: 80,
 Height: 180,
}))
users := sliceValue.Interface().([]User)
fmt.Printf("1st user name %s\n", users[0].Name)

6.3 反射创建 map

var userMap map[int]*User
mapType := reflect.TypeOf(userMap)
// mapValue:=reflect.MakeMap(mapType)
mapValue := reflect.MakeMapWithSize(mapType, 10//reflect.MakeMap、reflect.MakeSlice、reflect.MakeChan、reflect.MakeFunc

user := &common.User{
 Id:     7,
 Name:   "杰克逊",
 Weight: 65.5,
 Height: 1.68,
}
key := reflect.ValueOf(user.Id)
mapValue.SetMapIndex(key, reflect.ValueOf(user))                    //SetMapIndex 往map里添加一个key-value对
mapValue.MapIndex(key).Elem().FieldByName("Name").SetString("令狐一刀"//MapIndex 根据Key取出对应的map
userMap = mapValue.Interface().(map[int]*User)
fmt.Printf("user name %s %s\n", userMap[7].Name, user.Name)

7、反射修改值

反射修改值要求必须是指针类型

修改值的操作:pointer.Elem().Setxxx()

package main

import (
 "fmt"
 "reflect"
)

func main() {
 var num float64 = 3.14
 fmt.Printf("原始值 %f\n", num)
 // 通过reflect.ValueOf获取num中的value,必须是指针才可以修改值
 //pointer := reflect.ValueOf(num)  // 直接传值会panic
 pointer := reflect.ValueOf(&num)
 newValue := pointer.Elem()
 // 赋新的值
 newValue.SetFloat(5.66)
 fmt.Printf("新的值 %f\n", num)
}

7.1 反射修改 struct

user := User{
 Id:     7,
 Name:   "杰克逊",
 Weight: 65.5,
 Height: 1.68,
}
valueUser := reflect.ValueOf(&user)
// valueS.Elem().SetInt(8)//会panic
valueUser.Elem().FieldByName("Weight").SetFloat(68.0//FieldByName()通过Name返回类的成员变量。不能在指针Value上调用FieldByName
addrValue := valueUser.Elem().FieldByName("addr")
if addrValue.CanSet() {
 addrValue.SetString("北京")
else {
 fmt.Println("addr是未导出成员,不可Set"//以小写字母开头的成员相当于是私有成员
}

7.2 反射修改 slice

下面示例,间接的实现了 append 功能

users := make([]*User, 15//len=1,cap=5

sliceValue := reflect.ValueOf(&users) //准备通过Value修改users,所以传users的地址
if sliceValue.Elem().Len() > 0 {      //取得slice的长度
 sliceValue.Elem().Index(0).Elem().FieldByName("Name").SetString("哈哈哈")
 // u0 := users[0]
 fmt.Printf("1st user name change to %s\n", users[0].Name)
}

sliceValue.Elem().SetCap(3//新的cap必须位于原始的len到cap之间
sliceValue.Elem().SetLen(2)
//调用reflect.Value的Set()函数修改其底层指向的原始数据
sliceValue.Elem().Index(1).Set(reflect.ValueOf(&User{
 Id:     8,
 Name:   "geek",
 Weight: 80,
 Height: 180,
}))
fmt.Printf("2nd user name %s\n", users[1].Name)

7.3 反射修改 map

u1 := &User{
 Id:     7,
 Name:   "杰克逊",
 Weight: 65.5,
 Height: 1.68,
}
u2 := &User{
 Id:     8,
 Name:   "杰克逊",
 Weight: 65.5,
 Height: 1.68,
}
userMap := make(map[int]*User, 5)
userMap[u1.Id] = u1

mapValue := reflect.ValueOf(&userMap)                                                         //准备通过Value修改userMap,所以传userMap的地址
mapValue.Elem().SetMapIndex(reflect.ValueOf(u2.Id), reflect.ValueOf(u2))                      //SetMapIndex 往map里添加一个key-value对
mapValue.Elem().MapIndex(reflect.ValueOf(u1.Id)).Elem().FieldByName("Name").SetString("令狐一刀"//MapIndex 根据Key取出对应的map
for k, user := range userMap {
 fmt.Printf("key %d name %s\n", k, user.Name)
}

See you ~

欢迎进群一起进行技术交流

加群方式:公众号消息私信“加群或加我好友再加群均可

浏览 99
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报