Go基础之--反射

Posted 2020-10-24 coder

反射：可以在运行时动态获取变量的相关信息

反射需要导入reflect

反射中重要函数的演示

反射有几下几个重要的函数：
reflect.TypeOf ：获取变量的类型，返回reflect.Type类型
reflect.ValueOf：获取变量的值，返回reflect.Value类型
reflect.Value.Kind：获取变量的类别，返回一个常量
reflect.Value.Interface()：转换成interface{}类型

通过一个小例子来理解：

package main

import (
    "reflect"
    "fmt"
)

type Student struct{
    Name string
    Age int
}

func (s*Student) SetName(name string){
    s.Name="coders"
}

func (s*Student)SetAge(age int){
    s.Age = 23
}


func getTypeInfo(a interface{}){
    // 用于获取一个数据的数据类型
    typeInfo := reflect.TypeOf(a)
    kind := typeInfo.Kind()
    fmt.Println("kind of a :",kind)

    num := typeInfo.NumMethod() //获取当前数据有多少个方法
    fmt.Println("method num:",num)

    method,ok:=typeInfo.MethodByName("SetName") //获取是否有某个方法
    if !ok{
        fmt.Println("not have method SetName")
    }else{
        fmt.Println(method)
    }
}

func getAllMethod(a interface{}){
    // 用于获取变量下的所有方法
    typeInfo := reflect.TypeOf(a)
    num := typeInfo.NumMethod()
    for i:=0;i<num;i++ {
        method:= typeInfo.Method(i)
        fmt.Println(method)
    }
}

func testGetAllMethod()  {
    var stu Student
    getAllMethod(&stu)
}

func testGetTypeInfo(){
    var i int
    getTypeInfo(i) //获取的结果就是int

    var stu Student
    getTypeInfo(&stu) //获取的结果就是struct
    getAllMethod(&stu)

    var s []int
    getTypeInfo(s)  //获取的结果就是slice

    var a [5]int
    getTypeInfo(a)  //获取的结果就是array


}

func testGetValueInfo(){
    var i = 100
    valueInfo := reflect.ValueOf(i)
    tmp := valueInfo.Interface()  //转换成interface类型
    val := tmp.(int) //这里我是知道是int所以直接转换了
    fmt.Println("val:",val) //这里获取的还是100
    fmt.Println("val of valueInfo:",valueInfo.Int()) // 这里打印的也是100
    fmt.Println("type:",valueInfo.Type())
    fmt.Println("kind:",valueInfo.Kind())
}

func main(){
    testGetTypeInfo()
    testGetAllMethod()
    testGetValueInfo()
}

上面这个例子中演示了reflect.Value.Kind()可以返回int,struct,slice,array,当然这里可以返回的类型还有很多如下：
Bool
Int
Int8
Int16
Int32
Int64
Uint
Uint8
Uint16
Uint32 
Uint64 
Uintptr 
Float32 
Float64 
Complex64 
Complex128 
Array
Chan 
Func 
Interface 
Map
Ptr
Slice
String
Struct 
UnsafePointer

获取变量的值

reflect.ValueOf(x).Float()
reflect.ValueOf(x).Int()
reflect.ValueOf(x).String()
reflect.ValueOf(x).Bool()

这个功能在上面的代码中也有演示

通过反射来改变变量的值

reflect.Value.SetXX相关方法，如：
reflect.Value.SetFloat()：设置浮点数
reflect.Value.SetInt()：设置整数
reflect.Value.SetString()：设置字符串

通过下面一个简单的例子来演示：

package main

import (
    "reflect"
    "fmt"
)

func main() {
    var a float64
    fmt.Println(a)
    fv := reflect.ValueOf(a)
    fv.SetFloat(3.14)
    fmt.Println(a)
}

上面这段代码会提示如下错误：

这里需要知道的是我们的变量a是一个值类型的变量，我们通过reflect.valueOf传入的时候其实是传入的变量的拷贝，所以我们如果通过SetFloat给变量设置值的时候其实并不会生效，go这里已经替我考虑到了，所以给我们提示了上面这个错误信息，那是不是我们在reflect.Value的传入地址就可以了呢，我把上述代码中更改为：reflect.Value(&a),当我们运行后发现还是报了和上面相同的错误，这是为什么呢？

我们应该还记得如果是一个指针的时候我们赋值的时候是需要在指针的左边写个*符号，但是这是在反射里面我们怎么写星号，所以go在这里提供给我们另外一个方法，当我们通过调用SetFloat的时候用：
fv.Elem().SetFloat(3.14)这种方式调用就ok了，完整的正确代码为：

package main

import (
    "reflect"
    "fmt"
)

func main() {
    var a float64
    fmt.Println(a)
    fv := reflect.ValueOf(&a)
    fv.Elem().SetFloat(3.14)
    fmt.Println(a)
}

反射操作结构体

reflect.Value.NumField():获取结构体中字段的个数
reflect.Value.Method(n).Call()：调用结构体中的方法

package main

import (
    "reflect"
    "fmt"
)

type Student struct{
    Name string
    Age int
    Sex int
}

func (s *Student) Set(name string,age int,sex int){
    s.Name = name
    s.Age = age
    s.Sex = sex
}

func testStruct()  {
    var stu *Student = &Student{}
    stu.Set("coder",23,1)
    valueInfo := reflect.ValueOf(stu)

    fieldNum := valueInfo.Elem().NumField()
    fmt.Println("filed num:",fieldNum) //这里返回的结果是3

    sexValueInfo := valueInfo.Elem().FieldByName("Sex")
    fmt.Println("sex=",sexValueInfo.Int())
    sexValueInfo.SetInt(0) //这里是更改值
    fmt.Println(stu)
    setMethod := valueInfo.MethodByName("Set") //获取Set方法
    var params []reflect.Value
    name := "tom"
    age := 18
    sex:=2
    params = append(params,reflect.ValueOf(name),reflect.ValueOf(age),reflect.ValueOf(sex))
    setMethod.Call(params) //调用Set方法
    fmt.Println(stu) //将最开始的值已经更改了


}

func main() {
    testStruct()
}