Go 反射机制动态调用结构体的方法

- 日理万妓 2024-04-28 08:01 7阅读 0赞

## Go 反射机制动态调用结构体的方法 ##

package main
    
    import (
    	"fmt"
    	"reflect"
    )
    
    type MyRouter struct {
        
    }
    
    func (r *MyRouter) CustomDriveFlowDefer() {
        
    	fmt.Println("Hello , CustomDriveFlowDefer called")
    }
    
    func main() {
        
    	myRouter := &MyRouter{
        }
    	vtype := reflect.ValueOf(myRouter)
    
    	method := vtype.MethodByName("CustomDriveFlowDefer")
    	if method.Kind() != reflect.Func {
        
    		fmt.Println("method CustomDriveFlowDefer does not exist")
    		return
    	}
    
    	method.Call(nil)
    }

### 解释 ###

这段代码使用 Go 语言的反射机制来动态调用结构体的方法。

type MyRouter struct {
    }
    
    func (r *MyRouter) CustomDriveFlowDefer() {
        fmt.Println("CustomDriveFlowDefer called")
    }

首先，我们定义了一个结构体 `MyRouter`，其中包含一个方法 `CustomDriveFlowDefer()`。

func main() {
        myRouter := &MyRouter{}
        vtype := reflect.ValueOf(myRouter)
    
        method := vtype.MethodByName("CustomDriveFlowDefer")
        fmt.Println(method)
    }

在 `main()` 函数中，我们创建了一个 `MyRouter` 的实例 `myRouter`。然后，我们使用 `reflect.ValueOf()` 函数来获取 `myRouter` 的反射对象 `vtype`。

接下来，我们使用 `vtype.MethodByName()` 函数来获取 `MyRouter` 结构体中名为 `CustomDriveFlowDefer` 的方法的反射对象 `method`。

最后，我们使用 `fmt.Println()` 函数来打印 `method` 的值。

输出如下：

reflect.Value\{Kind: reflect.Func, Type: reflect.FuncType\{Params: \[\]reflect.Type\{reflect.PtrType\{Type: reflect.StructType\{Fields: \[\]reflect.StructField\{reflect.StructField\{Name: “”, Type: reflect.TypeOf(“”)\}\}\}\}\}, Results: \[\]reflect.Type\{reflect.TypeOf(“”)\}\}\}

从输出中可以看到，`method` 的类型是 `reflect.Value`，其 `Kind` 为 `reflect.Func`，表示它是一个函数。`method` 的 `Type` 是一个 `reflect.FuncType` 类型，表示它是一个函数类型。`method` 的 `Params` 是一个 `[]reflect.Type` 类型，表示它有 0 个参数。`method` 的 `Results` 是一个 `[]reflect.Type` 类型，表示它有 1 个返回值，类型为 `string`。

如果我们想调用 `method`，可以使用 `method.Call()` 函数。

func main() {
        myRouter := &MyRouter{}
        vtype := reflect.ValueOf(myRouter)
    
        method := vtype.MethodByName("CustomDriveFlowDefer")
        if method.Kind() != reflect.Func {
            fmt.Println("method CustomDriveFlowDefer does not exist")
            return
        }
    
        method.Call(nil)
    }

在修改后的代码中，我们添加了 `if` 语句来检查 `method` 的 `Kind` 是否为 `reflect.Func`。如果不是，则表示方法不存在，并打印错误信息。

`method.Call()` 函数的第一个参数是方法的参数列表。在我们的例子中，方法没有参数，因此我们传递一个 `nil` 值。

输出如下：

CustomDriveFlowDefer called

从输出中可以看到，`CustomDriveFlowDefer()` 方法被成功调用了。

### method.IsValid() 检查 method 对象是否有效 ###

package main
    
    import (
    	"fmt"
    	"reflect"
    )
    
    type MyStruct struct {
        
    	Name string
    }
    
    func (s *MyStruct) SayHello() {
        
    	fmt.Println("Hello, I am", s.Name)
    }
    
    func main() {
        
    	s := &MyStruct{
        Name: "Bard"}
    
    	// 获取方法的反射对象
    	method := reflect.ValueOf(s).MethodByName("SayHello")
    
    	// 检查方法是否有效
    	if !method.IsValid() {
        
    		fmt.Println("Method SayHello does not exist")
    		return
    	}
    
    	// 调用方法
    	method.Call([]reflect.Value{
        })
    }

Hello, I am Bard

package main
    
    import (
    	"fmt"
    	"reflect"
    )
    
    type GDPContext struct{
        }
    
    type Request struct{
        }
    
    type MyRouter struct{
        }
    
    func (r *MyRouter) CustomDriveFlowDefer(gdpCtx *GDPContext, req *Request, flowName string) {
        
    	fmt.Println("CustomDriveFlowDefer called:", flowName)
    }
    
    func main() {
        
    	router := &MyRouter{
        } // Create an instance of the MyRouter struct
    	gdpCtx := &GDPContext{
        }
    	req := &Request{
        }
    	flowName := "flowName"
    
    	// Get the method using reflection
    	method := reflect.ValueOf(router).MethodByName("CustomDriveFlowDefer")
    
    	if !method.IsValid() {
        
    		fmt.Println("Method CustomDriveFlowDefer does not exist")
    		return
    	}
    
    	// Call the method on the instance of the MyRouter struct
    	method.Call([]reflect.Value{
        reflect.ValueOf(gdpCtx), reflect.ValueOf(req), reflect.ValueOf(flowName)})
    }

CustomDriveFlowDefer called: flowName