浅见池也 发表于 2018-9-19 14:11:33

golang中的reflect包用法

  最近在写一个自动生成api文档的功能,用到了reflect包来给结构体赋值,给空数组新增一个元素,这样只要定义一个input结构体和一个output的结构体,并填写一些相关tag信息,就能使用程序来生成输入和输出的相关文档。


介绍
  reflect包是golang中很重要的一个包,实现了在运行时允许程序操纵任意类型对象的功能。可以看下文档简单了解一下。
  在reflect中,最重要的是Value类,只有先获取到一个对象或者变量的Value对象后,我们才可以对这个对象或者变量进行更进一步的分析和处理。我们可以使用reflect.ValueOf()方法获取Value对象。
  

  
var i int
  
value := reflect.ValueOf(i) // 使用ValueOf()获取到变量的Value对象
  

  
type S struct {
  a string
  
}
  

  
var s S
  
value2 := reflect.ValueOf(s) // 使用ValueOf()获取到结构体的Value对象
  

  获取到对象或者变量的Value对象后,我们就可以对他们进一步的操作了。

1.获取对象或者变量的类型(Value.Type()和Value.Kind())
  Value.Type()和Value.Kind()这两个方法都可以获取对象或者变量的类型,如果是变量的话,使用这两个方法获取到的类型都是一样,差别是结构体对象,举个例子看一下:
  

var i int  
value := reflect.ValueOf(i)
  

  
log.Println(value.Type()) //输出:int
  
log.Println(value.Kind()) //输出:int
  

  
type S struct {
  a string
  
}
  

  
var s S
  
value2 := reflect.ValueOf(s) // 使用ValueOf()获取到结构体的Value对象
  

  

  
log.Println(value2.Type()) //输出:S
  
log.Println(value2.Kind()) //输出:struct
  

  变量i使用kind和type两个方法都输出了int,而结构体s的Type()方法输出了S,Kind()方法输出了struct,由此可以总结如下,如果你想拿到结构体里面定义的变量信息的时候,使用Type(f)方法。如果只是相判断是否是结构体时,就使用Kind()

2.获取变量的值和给变量赋值
  获取变量的值使用value.Interface()方法,该方法会返回一个value的值,不过类型是interface。给变量赋值需要先判断该变量的类型,使用之前提到过的Value.Kind()方法,如果变量的类型是reflect.Int,我们就可以使用Value.SetInt()方法给变量赋值。下面是一个例子:
  

var i int = 1  

  
// 获取Value,这里注意,如果你要改变这个变量的话,需要传递变量的地址
  
value := reflect.ValueOf(&i)
  

  
// value是一个指针,这里获取了该指针指向的值,相当于value.Elem()
  
value = reflect.Indirect(value)
  

  
// Interface是获取该value的值,返回的是一个interface对象
  
log.Println(value.Interface()) // 输出:1
  

  
// 把变量i的值设为2
  
if value.Kind() == reflect.Int {
  value.SetInt(2)
  
}
  

  
log.Println(value.Interface()) // 输出:2
  

  给结构体对象中的成员变量赋值的方法:
  

type S struct {  A string // 注意:只有大写开头的成员变量可以Set
  
}
  

  
s := S{"x"}
  

  
value := reflect.ValueOf(&s)
  

  
value = reflect.Indirect(value)
  

  

  
//value是结构体s,所以打印出来的是整个结构体的信息
  
log.Println(value.Interface()) //输出: {x}
  

  
f0 := value.FieldByName("A") //获取结构体s中第一个元素a
  

  
log.Println(f0) // 输出: x
  

  
if f0.Kind() == reflect.String {
  if f0.CanSet() {
  f0.SetString("y")
  }
  
}
  

  
log.Println(f0) // 输出: y
  

  
log.Println(value.Interface()) //输出: {y}
  

  

  结构体这里需要注意的是,只有公有的成员变量可以被reflect改变值,私有的变量是无法改变值得。

3.获取结构体成员变量的tag信息
  由于golang变量大小写和公有私有息息相关,所以码农门很难按照自己的意愿来定义变量名。于是golang提供了tag机制,来给变量提供一个标签,这个标签可以作为一个别名,来给一些存储结构来获取结构体变量名字使用。下面是一个获取结构体成员变量tag信息的例子:
  

type S struct {  A string `json:"tag_a"`
  
}
  

  
s := S{}
  

  
value := reflect.ValueOf(&s)
  

  
value = reflect.Indirect(value)
  

  
//获取结构体s的类型S
  
vt := value.Type()
  

  
//获取S中的A成员变量
  
f, _ := vt.FieldByName("A")
  

  
//获取成员变量A的db标签
  
log.Println(f.Tag.Get("json")) //输出: tag_a
  

  未完待续。。。


页: [1]
查看完整版本: golang中的reflect包用法