Golang 反射如何检测函数是否存在_Golang 动态方法反查与类型映射

package main

import (
    "fmt"
    "reflect"
)

type User struct{}

func (u *User) GetName() string {
    return "Alice"
}

func (u *User) SetName(name string) {
    // 设置名字
}

func main() {
    user := &User{}
    v := reflect.ValueOf(user)

    // 查找方法
    method := v.MethodByName("GetName")
    if method.IsValid() {
        fmt.Println("方法 GetName 存在")
        // 调用方法
        result := method.Call(nil)
        fmt.Println("调用结果:", result[0].String())
    } else {
        fmt.Println("方法 GetName 不存在")
    }

    // 检查不存在的方法
    method = v.MethodByName("InvalidMethod")
    if !method.IsValid() {
        fmt.Println("方法 InvalidMethod 不存在")
    }
}

func hasMethod(obj interface{}, methodName string, argTypes []reflect.Type, returnTypes []reflect.Type) bool {
    t := reflect.TypeOf(obj)
    
    // 如果是指针，取其指向的类型
    if t.Kind() == reflect.Ptr {
        t = t.Elem()
    }

    method, ok := t.MethodByName(methodName)
    if !ok {
        return false
    }

    mt := method.Type
    
    // 参数数量匹配（包含 receiver）
    if mt.NumIn() != len(argTypes)+1 { // +1 是 receiver
        return false
    }
    for i, argType := range argTypes {
        if mt.In(i+1) != argType { // In(0) 是 receiver
            return false
        }
    }

    // 返回值数量匹配
    if mt.NumOut() != len(returnTypes) {
        return false
    }
    for i, retType := range returnTypes {
        if mt.Out(i) != retType {
            return false
        }
    }

    return true
}

func main() {
    user := &User{}
    
    args := []reflect.Type{reflect.TypeOf("")} // 参数是 string
    returns := []reflect.Type{nil}            // 没有返回值

    exists := hasMethod(user, "SetName", args, returns)
    if exists {
        fmt.Println("SetName(string) 方法存在且签名匹配")
    } else {
        fmt.Println("方法签名不匹配或不存在")
    }
}

func buildMethodMap(obj interface{}) map[string]reflect.Value {
    m := make(map[string]reflect.Value)
    v := reflect.ValueOf(obj)

    t := v.Type()
    if t.Kind() == reflect.Ptr {
        t = t.Elem()
        v = v.Elem()
    }

    for i := 0; i < t.NumMethod(); i++ {
        method := t.Method(i)
        m[method.Name] = v.MethodByName(method.Name)
    }
    return m
}

methods := buildMethodMap(&User{})
if fn, ok := methods["GetName"]; ok {
    result := fn.Call(nil)
    fmt.Println(result[0])
}