简介

反射是一种机制,在编译时不知道具体类型的情况下,可以透视结构的组成、更新值。使用反射,可以让我们编写出能统一处理所有类型的代码。甚至是编写这部分代码时还不存在的类型。一个具体的例子就是fmt.Println()方法,可以打印出我们自定义的结构类型。

虽然,一般来说都不建议在代码中使用反射。反射影响性能、不易阅读、将编译时就能检查出来的类型问题推迟到运行时以 panic 形式表现出来,这些都是反射的缺点。但是,我认为反射是一定要掌握的,原因如下:

  • 很多标准库和第三方库都用到了反射,虽然暴露的接口做了封装,不需要了解反射。但是如果要深入研究这些库,了解实现,阅读源码, 反射是绕不过去的。例如encoding/jsonencoding/xml等;
  • 如果有一个需求,编写一个可以处理所有类型的函数或方法,我们就必须会用到反射。因为 Go 的类型数量是无限的,而且可以自定义类型,所以使用类型断言是无法达成目标的。

Go 语言标准库reflect提供了反射功能。

接口

反射是建立在 Go 的类型系统之上的,并且与接口密切相关。

首先简单介绍一下接口。Go 语言中的接口约定了一组方法集合,任何定义了这组方法的类型(也称为实现了接口)的变量都可以赋值给该接口的变量。

  1. package main
  2. import "fmt"
  3. type Animal interface {
  4. Speak()
  5. }
  6. type Cat struct {
  7. }
  8. func (c Cat) Speak() {
  9. fmt.Println("Meow")
  10. }
  11. type Dog struct {
  12. }
  13. func (d Dog) Speak() {
  14. fmt.Println("Bark")
  15. }
  16. func main() {
  17. var a Animal
  18. a = Cat{}
  19. a.Speak()
  20. a = Dog{}
  21. a.Speak()
  22. }

上面代码中,我们定义了一个Animal接口,它约定了一个方法Speak()。而后定义了两个结构类型CatDog,都定义了这个方法。这样,我们就可以将CatDog对象赋值给Animal类型的变量了。

接口变量包含两部分:类型和值,即(type, value)。类型就是赋值给接口变量的值的类型,值就是赋值给接口变量的值。如果知道接口中存储的变量类型,我们也可以使用类型断言通过接口变量获取具体类型的值:

  1. type Animal interface {
  2. Speak()
  3. }
  4. type Cat struct {
  5. Name string
  6. }
  7. func (c Cat) Speak() {
  8. fmt.Println("Meow")
  9. }
  10. func main() {
  11. var a Animal
  12. a = Cat{Name: "kitty"}
  13. a.Speak()
  14. c := a.(Cat)
  15. fmt.Println(c.Name)
  16. }

上面代码中,我们知道接口a中保存的是Cat对象,直接使用类型断言a.(Cat)获取Cat对象。但是,如果类型断言的类型与实际存储的类型不符,会直接 panic。所以实际开发中,通常使用另一种类型断言形式c, ok := a.(Cat)。如果类型不符,这种形式不会 panic,而是通过将第二个返回值置为 false 来表明这种情况

有时候,一个类型定义了很多方法,而不只是接口约定的方法。通过接口,我们只能调用接口中约定的方法。当然我们也可以将其类型断言为另一个接口,然后调用这个接口约定的方法,前提是原对象实现了这个接口:

  1. var r io.Reader
  2. r = new(bytes.Buffer)
  3. w = r.(io.Writer)

io.Readerio.Writer是标准库中使用最为频繁的两个接口:

  1. // src/io/io.go
  2. type Reader interface {
  3. Read(p []byte) (n int, err error)
  4. }
  5. type Writer interface {
  6. Write(p []byte) (n int, err error)
  7. }

bytes.Buffer同时实现了这两个接口,所以byte.Buffer对象可以赋值给io.Reader变量r,然后r可以断言为io.Writer,因为接口io.Reader中存储的值也实现了io.Writer接口。

如果一个接口A包含另一个接口B的所有方法,那么接口A的变量可以直接赋值给B的变量,因为A中存储的值一定实现了A约定的所有方法,那么肯定也实现了B。此时,无须类型断言。例如标准库io中还定义了一个io.ReadCloser接口,此接口变量可以直接赋值给io.Reader

  1. // src/io/io.go
  2. type ReadCloser interface {
  3. Reader
  4. Closer
  5. }

空接口interface{}是比较特殊的一个接口,它没有约定任何方法。所有类型值都可以赋值给空接口类型的变量,因为它没有任何方法限制

有一点特别重要,接口变量之间类型断言也好,直接赋值也好,其内部存储的(type, value)类型-值对是没有变化的。只是通过不同的接口能调用的方法有所不同而已。也是由于这个原因,接口变量中存储的值一定不是接口类型

有了这些接口的基础知识,下面我们介绍反射。

反射基础

Go 语言中的反射功能由reflect包提供。reflect包定义了一个接口reflect.Type和一个结构体reflect.Value,它们定义了大量的方法用于获取类型信息,设置值等。在reflect包内部,只有类型描述符实现了reflect.Type接口。由于类型描述符是未导出类型,我们只能通过reflect.TypeOf()方法获取reflect.Type类型的值:

  1. package main
  2. import (
  3. "fmt"
  4. "reflect"
  5. )
  6. type Cat struct {
  7. Name string
  8. }
  9. func main() {
  10. var f float64 = 3.5
  11. t1 := reflect.TypeOf(f)
  12. fmt.Println(t1.String())
  13. c := Cat{Name: "kitty"}
  14. t2 := reflect.TypeOf(c)
  15. fmt.Println(t2.String())
  16. }

输出:

  1. float64
  2. main.Cat

Go 语言是静态类型的,每个变量在编译期有且只能有一个确定的、已知的类型,即变量的静态类型。静态类型在变量声明的时候就已经确定了,无法修改。一个接口变量,它的静态类型就是该接口类型。虽然在运行时可以将不同类型的值赋值给它,改变的也只是它内部的动态类型和动态值。它的静态类型始终没有改变。

reflect.TypeOf()方法就是用来取出接口中的动态类型部分,以reflect.Type返回。等等!上面代码好像并没有接口类型啊?

我们看下reflect.TypeOf()的定义:

  1. // src/reflect/type.go
  2. func TypeOf(i interface{}) Type {
  3. eface := *(*emptyInterface)(unsafe.Pointer(&i))
  4. return toType(eface.typ)
  5. }

它接受一个interface{}类型的参数,所以上面的float64Cat变量会先转为interface{}再传给方法,reflect.TypeOf()方法获取的就是这个interface{}中的类型部分。

相应地,reflect.ValueOf()方法自然就是获取接口中的值部分,返回值为reflect.Value类型。在上例基础上添加下面代码:

  1. v1 := reflect.ValueOf(f)
  2. fmt.Println(v1)
  3. fmt.Println(v1.String())
  4. v2 := reflect.ValueOf(c)
  5. fmt.Println(v2)
  6. fmt.Println(v2.String())

运行输出:

  1. 3.5
  2. <float64 Value>
  3. {kitty}
  4. <main.Cat Value>

由于fmt.Println()会对reflect.Value类型做特殊处理,打印其内部的值,所以上面显示调用了reflect.Value.String()方法获取更多信息。

获取类型如此常见,fmt提供了格式化符号%T输出参数类型:

  1. fmt.Printf("%T\n", 3) // int

Go 语言中类型是无限的,而且可以通过type定义新的类型。但是类型的种类是有限的,reflect包中定义了所有种类的枚举:

  1. // src/reflect/type.go
  2. type Kind uint
  3. const (
  4. Invalid Kind = iota
  5. Bool
  6. Int
  7. Int8
  8. Int16
  9. Int32
  10. Int64
  11. Uint
  12. Uint8
  13. Uint16
  14. Uint32
  15. Uint64
  16. Uintptr
  17. Float32
  18. Float64
  19. Complex64
  20. Complex128
  21. Array
  22. Chan
  23. Func
  24. Interface
  25. Map
  26. Ptr
  27. Slice
  28. String
  29. Struct
  30. UnsafePointer
  31. )

一共 26 种,我们可以分类如下:

  • 基础类型BoolString以及各种数值类型(有符号整数Int/Int8/Int16/Int32/Int64,无符号整数Uint/Uint8/Uint16/Uint32/Uint64/Uintptr,浮点数Float32/Float64,复数Complex64/Complex128
  • 复合(聚合)类型ArrayStruct
  • 引用类型ChanFuncPtrSliceMap(值类型和引用类型区分不明显,这里不引战,大家理解意思就行)
  • 接口类型Interface
  • 非法类型Invalid,表示它还没有任何值(reflect.Value的零值就是Invalid类型)

Go 中所有的类型(包括自定义的类型),都是上面这些类型或它们的组合。

例如:

  1. type MyInt int
  2. func main() {
  3. var i int
  4. var j MyInt
  5. i = int(j) // 必须强转
  6. ti := reflect.TypeOf(i)
  7. fmt.Println("type of i:", ti.String())
  8. tj := reflect.TypeOf(j)
  9. fmt.Println("type of j:", tj.String())
  10. fmt.Println("kind of i:", ti.Kind())
  11. fmt.Println("kind of j:", tj.Kind())
  12. }

上面两个变量的静态类型分别为intMyInt,是不同的。虽然MyInt的底层类型(underlying type)也是int。它们之间的赋值必须要强制类型转换。但是它们的种类是一样的,都是int

代码输出如下:

  1. type of i: int
  2. type of j: main.MyInt
  3. kind of i: int
  4. kind of j: int

反射用法

由于反射的内容和 API 非常多,我们结合具体用法来介绍。

透视数据组成

透视结构体组成,需要以下方法:

  • reflect.ValueOf():获取反射值对象;
  • reflect.Value.NumField():从结构体的反射值对象中获取它的字段个数;
  • reflect.Value.Field(i):从结构体的反射值对象中获取第i个字段的反射值对象;
  • reflect.Kind():从反射值对象中获取种类;
  • reflect.Int()/reflect.Uint()/reflect.String()/reflect.Bool():这些方法从反射值对象做取出具体类型。

示例:

  1. type User struct {
  2. Name string
  3. Age int
  4. Married bool
  5. }
  6. func inspectStruct(u interface{}) {
  7. v := reflect.ValueOf(u)
  8. for i := 0; i < v.NumField(); i++ {
  9. field := v.Field(i)
  10. switch field.Kind() {
  11. case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
  12. fmt.Printf("field:%d type:%s value:%d\n", i, field.Type().Name(), field.Int())
  13. case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
  14. fmt.Printf("field:%d type:%s value:%d\n", i, field.Type().Name(), field.Uint())
  15. case reflect.Bool:
  16. fmt.Printf("field:%d type:%s value:%t\n", i, field.Type().Name(), field.Bool())
  17. case reflect.String:
  18. fmt.Printf("field:%d type:%s value:%q\n", i, field.Type().Name(), field.String())
  19. default:
  20. fmt.Printf("field:%d unhandled kind:%s\n", i, field.Kind())
  21. }
  22. }
  23. }
  24. func main() {
  25. u := User{
  26. Name: "dj",
  27. Age: 18,
  28. Married: true,
  29. }
  30. inspectStruct(u)
  31. }

结合使用reflect.ValueNumField()Field()方法可以遍历结构体的每个字段。然后针对每个字段的Kind做相应的处理。

有些方法只有在原对象是某种特定类型时,才能调用。例如NumField()Field()方法只有原对象是结构体时才能调用,否则会panic

识别出具体类型后,可以调用反射值对象的对应类型方法获取具体类型的值,例如上面的field.Int()/field.Uint()/field.Bool()/field.String()。但是为了减轻处理的负担,Int()/Uint()方法对种类做了合并处理,它们只返回相应的最大范围的类型,Int()返回Int64类型,Uint()返回Uint64类型。而Int()/Uint()内部会对相应的有符号或无符号种类做处理,转为Int64/Uint64返回。下面是reflect.Value.Int()方法的实现:

  1. // src/reflect/value.go
  2. func (v Value) Int() int64 {
  3. k := v.kind()
  4. p := v.ptr
  5. switch k {
  6. case Int:
  7. return int64(*(*int)(p))
  8. case Int8:
  9. return int64(*(*int8)(p))
  10. case Int16:
  11. return int64(*(*int16)(p))
  12. case Int32:
  13. return int64(*(*int32)(p))
  14. case Int64:
  15. return *(*int64)(p)
  16. }
  17. panic(&ValueError{"reflect.Value.Int", v.kind()})
  18. }

上面代码,我们只处理了少部分种类。在实际开发中,完善的处理需要破费一番功夫,特别是字段是其他复杂类型,甚至包含循环引用的时候。

另外,我们也可以透视标准库中的结构体,并且可以透视其中的未导出字段。使用上面定义的inspectStruct()方法:

  1. inspectStruct(bytes.Buffer{})

bytes.Buffer的结构如下:

  1. type Buffer struct {
  2. buf []byte
  3. off int
  4. lastRead readOp
  5. }

都是未导出的字段,程序输出:

  1. field:0 unhandled kind:slice
  2. field:1 type:int value:0
  3. field:2 type:readOp value:0

透视map组成,需要以下方法:

  • reflect.Value.MapKeys():将每个键的reflect.Value对象组成一个切片返回;
  • reflect.Value.MapIndex(k):传入键的reflect.Value对象,返回值的reflect.Value
  • 然后可以对键和值的reflect.Value进行和上面一样的处理。

示例:

  1. func inspectMap(m interface{}) {
  2. v := reflect.ValueOf(m)
  3. for _, k := range v.MapKeys() {
  4. field := v.MapIndex(k)
  5. fmt.Printf("%v => %v\n", k.Interface(), field.Interface())
  6. }
  7. }
  8. func main() {
  9. inspectMap(map[uint32]uint32{
  10. 1: 2,
  11. 3: 4,
  12. })
  13. }

我这里偷懒了,没有针对每个Kind去做处理,直接调用键-值reflect.ValueInterface()方法。该方法以空接口的形式返回内部包含的值。程序输出:

  1. 1 => 2
  2. 3 => 4

同样地,MapKeys()MapIndex(k)方法只能在原对象是map类型时才能调用,否则会panic

透视切片或数组组成,需要以下方法:

  • reflect.Value.Len():返回数组或切片的长度;
  • reflect.Value.Index(i):返回第i个元素的reflect.Value值;
  • 然后对这个reflect.Value判断Kind()进行处理。

示例:

  1. func inspectSliceArray(sa interface{}) {
  2. v := reflect.ValueOf(sa)
  3. fmt.Printf("%c", '[')
  4. for i := 0; i < v.Len(); i++ {
  5. elem := v.Index(i)
  6. fmt.Printf("%v ", elem.Interface())
  7. }
  8. fmt.Printf("%c\n", ']')
  9. }
  10. func main() {
  11. inspectSliceArray([]int{1, 2, 3})
  12. inspectSliceArray([3]int{4, 5, 6})
  13. }

程序输出:

  1. [1 2 3 ]
  2. [4 5 6 ]

同样地Len()Index(i)方法只能在原对象是切片,数组或字符串时才能调用,其他类型会panic

透视函数类型,需要以下方法:

  • reflect.Type.NumIn():获取函数参数个数;
  • reflect.Type.In(i):获取第i个参数的reflect.Type
  • reflect.Type.NumOut():获取函数返回值个数;
  • reflect.Type.Out(i):获取第i个返回值的reflect.Type

示例:

  1. func Add(a, b int) int {
  2. return a + b
  3. }
  4. func Greeting(name string) string {
  5. return "hello " + name
  6. }
  7. func inspectFunc(name string, f interface{}) {
  8. t := reflect.TypeOf(f)
  9. fmt.Println(name, "input:")
  10. for i := 0; i < t.NumIn(); i++ {
  11. t := t.In(i)
  12. fmt.Print(t.Name())
  13. fmt.Print(" ")
  14. }
  15. fmt.Println()
  16. fmt.Println("output:")
  17. for i := 0; i < t.NumOut(); i++ {
  18. t := t.Out(i)
  19. fmt.Print(t.Name())
  20. fmt.Print(" ")
  21. }
  22. fmt.Println("\n===========")
  23. }
  24. func main() {
  25. inspectFunc("Add", Add)
  26. inspectFunc("Greeting", Greeting)
  27. }

同样地,只有在原对象是函数类型的时候才能调用NumIn()/In()/NumOut()/Out()这些方法,其他类型会panic

程序输出:

  1. Add input:
  2. int int
  3. output:
  4. int
  5. ===========
  6. Greeting input:
  7. string
  8. output:
  9. string
  10. ===========

透视结构体中定义的方法,需要以下方法:

  • reflect.Type.NumMethod():返回结构体定义的方法个数;
  • reflect.Type.Method(i):返回第i个方法的reflect.Method对象;

示例:

  1. func inspectMethod(o interface{}) {
  2. t := reflect.TypeOf(o)
  3. for i := 0; i < t.NumMethod(); i++ {
  4. m := t.Method(i)
  5. fmt.Println(m)
  6. }
  7. }
  8. type User struct {
  9. Name string
  10. Age int
  11. }
  12. func (u *User) SetName(n string) {
  13. u.Name = n
  14. }
  15. func (u *User) SetAge(a int) {
  16. u.Age = a
  17. }
  18. func main() {
  19. u := User{
  20. Name: "dj",
  21. Age: 18,
  22. }
  23. inspectMethod(&u)
  24. }

reflect.Method定义如下:

  1. // src/reflect/type.go
  2. type Method struct {
  3. Name string // 方法名
  4. PkgPath string
  5. Type Type // 方法类型(即函数类型)
  6. Func Value // 方法值(以接收器作为第一个参数)
  7. Index int // 是结构体中的第几个方法
  8. }

事实上,reflect.Value也定义了NumMethod()/Method(i)这些方法。区别在于:reflect.Type.Method(i)返回的是一个reflect.Method对象,可以获取方法名、类型、是结构体中的第几个方法等信息。如果要通过这个reflect.Method调用方法,必须使用Func字段,而且要传入接收器的reflect.Value作为第一个参数:

  1. m.Func.Call(v, ...args)

但是reflect.Value.Method(i)返回一个reflect.Value对象,它总是以调用Method(i)方法的reflect.Value作为接收器对象,不需要额外传入。而且直接使用Call()发起方法调用:

  1. m.Call(...args)

reflect.Typereflect.Value有不少同名方法,使用时需要注意甄别。

调用函数或方法

调用函数,需要以下方法:

  • reflect.Value.Call():使用reflect.ValueOf()生成每个参数的反射值对象,然后组成切片传给Call()方法。Call()方法执行函数调用,返回[]reflect.Value。其中每个元素都是原返回值的反射值对象。

示例:

  1. func Add(a, b int) int {
  2. return a + b
  3. }
  4. func Greeting(name string) string {
  5. return "hello " + name
  6. }
  7. func invoke(f interface{}, args ...interface{}) {
  8. v := reflect.ValueOf(f)
  9. argsV := make([]reflect.Value, 0, len(args))
  10. for _, arg := range args {
  11. argsV = append(argsV, reflect.ValueOf(arg))
  12. }
  13. rets := v.Call(argsV)
  14. fmt.Println("ret:")
  15. for _, ret := range rets {
  16. fmt.Println(ret.Interface())
  17. }
  18. }
  19. func main() {
  20. invoke(Add, 1, 2)
  21. invoke(Greeting, "dj")
  22. }

我们封装一个invoke()方法,以interface{}空接口接收函数对象,以interface{}可变参数接收函数调用的参数。函数内部首先调用reflect.ValueOf()方法获得函数对象的反射值对象。然后依次对每个参数调用reflect.ValueOf(),生成参数的反射值对象切片。最后调用函数反射值对象的Call()方法,输出返回值。

程序运行结果:

  1. ret:
  2. 3
  3. ret:
  4. hello dj

方法的调用也是类似的:

  1. type M struct {
  2. a, b int
  3. op rune
  4. }
  5. func (m M) Op() int {
  6. switch m.op {
  7. case '+':
  8. return m.a + m.b
  9. case '-':
  10. return m.a - m.b
  11. case '*':
  12. return m.a * m.b
  13. case '/':
  14. return m.a / m.b
  15. default:
  16. panic("invalid op")
  17. }
  18. }
  19. func main() {
  20. m1 := M{1, 2, '+'}
  21. m2 := M{3, 4, '-'}
  22. m3 := M{5, 6, '*'}
  23. m4 := M{8, 2, '/'}
  24. invoke(m1.Op)
  25. invoke(m2.Op)
  26. invoke(m3.Op)
  27. invoke(m4.Op)
  28. }

运行结果:

  1. ret:
  2. 3
  3. ret:
  4. -1
  5. ret:
  6. 30
  7. ret:
  8. 4

以上是在编译期明确知道方法名的情况下发起调用。如果只给一个结构体对象,通过参数指定具体调用哪个方法该怎么做呢?这需要以下方法:

  • reflect.Value.MethodByName(name):获取结构体中定义的名为name的方法的reflect.Value对象,这个方法默认有接收器参数,即调用MethodByName()方法的reflect.Value

示例:

  1. type Math struct {
  2. a, b int
  3. }
  4. func (m Math) Add() int {
  5. return m.a + m.b
  6. }
  7. func (m Math) Sub() int {
  8. return m.a - m.b
  9. }
  10. func (m Math) Mul() int {
  11. return m.a * m.b
  12. }
  13. func (m Math) Div() int {
  14. return m.a / m.b
  15. }
  16. func invokeMethod(obj interface{}, name string, args ...interface{}) {
  17. v := reflect.ValueOf(obj)
  18. m := v.MethodByName(name)
  19. argsV := make([]reflect.Value, 0, len(args))
  20. for _, arg := range args {
  21. argsV = append(argsV, reflect.ValueOf(arg))
  22. }
  23. rets := m.Call(argsV)
  24. fmt.Println("ret:")
  25. for _, ret := range rets {
  26. fmt.Println(ret.Interface())
  27. }
  28. }
  29. func main() {
  30. m := Math{a: 10, b: 2}
  31. invokeMethod(m, "Add")
  32. invokeMethod(m, "Sub")
  33. invokeMethod(m, "Mul")
  34. invokeMethod(m, "Div")
  35. }

我们可以在结构体的反射值对象上使用NumMethod()Method()遍历它定义的所有方法。

实战案例

使用前面介绍的方法,我们很容易实现一个简单的、基于 HTTP 的 RPC 调用。约定格式:路径名/obj/method/arg1/arg2调用obj.method(arg1, arg2)方法。

首先定义两个结构体,并为它们定义方法,我们约定可导出的方法会注册为 RPC 方法。并且方法必须返回两个值:一个结果,一个错误。

  1. type StringObject struct{}
  2. func (StringObject) Concat(s1, s2 string) (string, error) {
  3. return s1 + s2, nil
  4. }
  5. func (StringObject) ToUpper(s string) (string, error) {
  6. return strings.ToUpper(s), nil
  7. }
  8. func (StringObject) ToLower(s string) (string, error) {
  9. return strings.ToLower(s), nil
  10. }
  11. type MathObject struct{}
  12. func (MathObject) Add(a, b int) (int, error) {
  13. return a + b, nil
  14. }
  15. func (MathObject) Sub(a, b int) (int, error) {
  16. return a - b, nil
  17. }
  18. func (MathObject) Mul(a, b int) (int, error) {
  19. return a * b, nil
  20. }
  21. func (MathObject) Div(a, b int) (int, error) {
  22. if b == 0 {
  23. return 0, errors.New("divided by zero")
  24. }
  25. return a / b, nil
  26. }

接下来我们定义一个结构表示可以调用的 RPC 方法:

  1. type RpcMethod struct {
  2. method reflect.Value
  3. args []reflect.Type
  4. }

其中method是方法的反射值对象,args是各个参数的类型。我们定义一个函数从对象中提取可以 RPC 调用的方法:

  1. var (
  2. mapObjMethods map[string]map[string]RpcMethod
  3. )
  4. func init() {
  5. mapObjMethods = make(map[string]map[string]RpcMethod)
  6. }
  7. func registerMethods(objName string, o interface{}) {
  8. v := reflect.ValueOf(o)
  9. mapObjMethods[objName] = make(map[string]RpcMethod)
  10. for i := 0; i < v.NumMethod(); i++ {
  11. m := v.Method(i)
  12. if m.Type().NumOut() != 2 {
  13. // 排除不是两个返回值的
  14. continue
  15. }
  16. if m.Type().Out(1).Name() != "error" {
  17. // 排除第二个返回值不是 error 的
  18. continue
  19. }
  20. t := v.Type().Method(i)
  21. methodName := t.Name
  22. if len(methodName) <= 1 || strings.ToUpper(methodName[0:1]) != methodName[0:1] {
  23. // 排除非导出方法
  24. continue
  25. }
  26. types := make([]reflect.Type, 0, 1)
  27. for j := 0; j < m.Type().NumIn(); j++ {
  28. types = append(types, m.Type().In(j))
  29. }
  30. mapObjMethods[objName][methodName] = RpcMethod{
  31. m, types,
  32. }
  33. }
  34. }

registerMethods()函数使用reflect.Value.NumMethod()reflect.Method(i)从对象中遍历方法,排除掉不是两个返回值的、第二个返回值不是 error 的或者非导出的方法。

然后定义一个 http 处理器:

  1. func handler(w http.ResponseWriter, r *http.Request) {
  2. parts := strings.Split(r.URL.Path[1:], "/")
  3. if len(parts) < 2 {
  4. handleError(w, errors.New("invalid request"))
  5. return
  6. }
  7. m := lookupMethod(parts[0], parts[1])
  8. if m.method.IsZero() {
  9. handleError(w, fmt.Errorf("no such method:%s in object:%s", parts[0], parts[1]))
  10. return
  11. }
  12. argSs := parts[2:]
  13. if len(m.args) != len(argSs) {
  14. handleError(w, errors.New("inconsistant args num"))
  15. return
  16. }
  17. argVs := make([]reflect.Value, 0, 1)
  18. for i, t := range m.args {
  19. switch t.Kind() {
  20. case reflect.Int:
  21. value, _ := strconv.Atoi(argSs[i])
  22. argVs = append(argVs, reflect.ValueOf(value))
  23. case reflect.String:
  24. argVs = append(argVs, reflect.ValueOf(argSs[i]))
  25. default:
  26. handleError(w, fmt.Errorf("invalid arg type:%s", t.Kind()))
  27. return
  28. }
  29. }
  30. ret := m.method.Call(argVs)
  31. err := ret[1].Interface()
  32. if err != nil {
  33. handleError(w, err.(error))
  34. return
  35. }
  36. response(w, ret[0].Interface())
  37. }

我们将路径分割得到一个切片,第一个元素为对象名(即mathstring),第二个元素为方法名(即Add/Sub/Mul/Div等),后面的都是参数。接着,我们查找要调用的方法,根据注册时记录的各个参数的类型将路径中的字符串转换为对应类型。然后调用,检查第二个返回值是否为nil可以获知方法调用是否出错。成功调用则返回结果。

最后我们只需要启动一个 http 服务器即可:

  1. func main() {
  2. registerMethods("math", MathObject{})
  3. registerMethods("string", StringObject{})
  4. mux := http.NewServeMux()
  5. mux.HandleFunc("/", handler)
  6. server := &http.Server{
  7. Addr: ":8080",
  8. Handler: mux,
  9. }
  10. if err := server.ListenAndServe(); err != nil {
  11. log.Fatal(err)
  12. }
  13. }

完整代码在 Github 仓库中。运行:

  1. $ go run main.go

使用 curl 来验证:

  1. $ curl localhost:8080/math/Add/1/2
  2. {"data":3}
  3. $ curl localhost:8080/math/Sub/10/2
  4. {"data":8}
  5. $ curl localhost:8080/math/Div/10/2
  6. {"data":5}
  7. $ curl localhost:8080/math/Div/10/0
  8. {"error":"divided by zero"}
  9. $ curl localhost:8080/string/Concat/abc/def
  10. {"data":"abcdef"}

当然,这只是一个简单的实现,还有很多错误处理没有考虑,方法参数的类型目前只支持intstring,感兴趣可以去完善一下。

设置值

首先介绍一个概念:可寻址。可寻址是可以通过反射获得其地址的能力。可寻址与指针紧密相关。所有通过reflect.ValueOf()得到的reflect.Value都不可寻址。因为它们只保存了自身的值,对自身的地址一无所知。例如指针p *int保存了另一个int数据在内存中的地址,但是它自身的地址无法通过自身获取到,因为在将它传给reflect.ValueOf()时,其自身地址信息就丢失了。我们可以通过reflect.Value.CanAddr()判断是否可寻址:

  1. func main() {
  2. x := 2
  3. a := reflect.ValueOf(2)
  4. b := reflect.ValueOf(x)
  5. c := reflect.ValueOf(&x)
  6. fmt.Println(a.CanAddr()) // false
  7. fmt.Println(b.CanAddr()) // false
  8. fmt.Println(c.CanAddr()) // false
  9. }

虽然指针不可寻址,但是我们可以在其反射对象上调用Elem()获取它指向的元素的reflect.Value。这个reflect.Value就可以寻址了,因为是通过reflect.Value.Elem()获取的值,可以记录这个获取路径。因而得到的reflect.Value中保存了它的地址:

  1. d := c.Elem()
  2. fmt.Println(d.CanAddr())

另外通过切片反射对象的Index(i)方法得到的reflect.Value也是可寻址的,我们总是可以通过切片得到某个索引的地址。通过结构体的指针获取到的字段也是可寻址的:

  1. type User struct {
  2. Name string
  3. Age int
  4. }
  5. s := []int{1, 2, 3}
  6. sv := reflect.ValueOf(s)
  7. e := sv.Index(1)
  8. fmt.Println(e.CanAddr()) // true
  9. u := &User{Name: "dj", Age: 18}
  10. uv := reflect.ValueOf(u)
  11. f := uv.Elem().Field(0)
  12. fmt.Println(f.CanAddr()) // true

如果一个reflect.Value可寻址,我们可以调用其Addr()方法返回一个reflect.Value,包含一个指向原始数据的指针。然后在这个reflect.Value上调用Interface{}方法,会返回一个包含这个指针的interface{}值。如果我们知道类型,可以使用类型断言将其转为一个普通指针。通过普通指针来更新值:

  1. func main() {
  2. x := 2
  3. d := reflect.ValueOf(&x).Elem()
  4. px := d.Addr().Interface().(*int)
  5. *px = 3
  6. fmt.Println(x) // 3
  7. }

这样的更新方法有点麻烦,我们可以直接通过可寻址的reflect.Value调用Set()方法来更新,不用通过指针:

  1. d.Set(reflect.ValueOf(4))
  2. fmt.Println(x) // 4

如果传入的类型不匹配,会 panic。reflect.Value为基本类型提供特殊的Set方法:SetIntSetUintSetFloat等:

  1. d.SetInt(5)
  2. fmt.Println(x) // 5

反射可以读取未导出结构字段的值,但是不能更新这些值。一个可寻址的reflect.Value会记录它是否是通过遍历一个未导出字段来获得的,如果是则不允许修改。所以在更新前使用CanAddr()判断并不保险。CanSet()可以正确判断一个值是否可以修改。

CanSet()判断的是可设置性,它是比可寻址性更严格的性质。如果一个reflect.Value是可设置的,它一定是可寻址的。反之则不然:

  1. type User struct {
  2. Name string
  3. age int
  4. }
  5. u := &User{Name: "dj", age: 18}
  6. uv := reflect.ValueOf(u)
  7. name := uv.Elem().Field(0)
  8. fmt.Println(name.CanAddr(), name.CanSet()) // true true
  9. age := uv.Elem().Field(1)
  10. fmt.Println(age.CanAddr(), age.CanSet()) // true false
  11. name.SetString("lidajun")
  12. fmt.Println(u) // &{lidajun 18}
  13. // 报错
  14. // age.SetInt(20)

StructTag

在定义结构体时,可以为每个字段指定一个标签,我们可以使用反射读取这些标签:

  1. type User struct {
  2. Name string `json:"name"`
  3. Age int `json:"age"`
  4. }
  5. func main() {
  6. u := &User{Name: "dj", Age: 18}
  7. t := reflect.TypeOf(u).Elem()
  8. for i := 0; i < t.NumField(); i++ {
  9. f := t.Field(i)
  10. fmt.Println(f.Tag)
  11. }
  12. }

标签就是一个普通的字符串,上面程序输出:

  1. json:"name"
  2. json:"age"

StructTag定义在reflect/type.go文件中:

  1. type StructTag string

一般惯例是将各个键值对,使用空格分开,键值之间使用:。例如:

  1. `json:"name" xml:"age"`

StructTag提供Get()方法获取键对应的值。

总结

本文系统地介绍了 Go 语言中的反射机制,从类型、接口到反射用法。还使用反射实现了一个简单的基于 HTTP 的 RPC 库。反射虽然在平时开发中不建议使用,但是阅读源码,自己编写库的时候需要频繁用到反射知识。熟练掌握反射可以使源码阅读事半功倍。

大家如果发现好玩、好用的 Go 语言库,欢迎到 Go 每日一库 GitHub 上提交 issue😄

参考

  1. Rob Pike, laws of reflection: https://golang.org/doc/articles/laws_of_reflection.html
  2. Go 程序设计语言,第 12 章:反射
  3. reflect 官方文档,https://pkg.go.dev/reflect
  4. Go 每日一库 GitHub:https://github.com/go-quiz/go-daily-lib