golang 代理实现

golang 代理实现"/>

golang 代理实现

 拦截接口

#invocation_handler.gopackage reflectype InvocationHandler interface {Invoke(proxy *Proxy, method *Method, args []interface{}) ([]interface{}, error)
}

代理

#proxy.gopackage reflecimport ("reflect""fmt""errors"
)type Proxy struct {target  interface{}methods map[string]*Methodhandle  InvocationHandler
}func New(target interface{}, h InvocationHandler) *Proxy {typ := reflect.TypeOf(target)value := reflect.ValueOf(target)methods := make(map[string]*Method, 0)for i := 0; i < value.NumMethod(); i++ {method := value.Method(i)methods[typ.Method(i).Name] = &Method{value:method}}return &Proxy{target:target, methods:methods, handle:h}
}func (p *Proxy)InvokeMethod(name string, args ...interface{}) ([]interface{}, error) {return p.handle.Invoke(p, p.methods[name], args)
}type Method struct {value reflect.Value
}func (m *Method)Invoke(args ...interface{}) (res []interface{}, err error){defer func() {if p := recover(); p != nil {err = errors.New(fmt.Sprintf("%s", p))}}()params := make([]reflect.Value, 0)if args != nil {for i := 0; i < len(args); i++ {params = append(params, reflect.ValueOf(args[i]))}}res = make([]interface{}, 0)r := m.value.Call(params)if r != nil && len(r) > 0{for i := 0; i < len(r); i++ {res = append(res, r[i].Interface())}}return
}

测试 

#main.gopackage mainimport (_ "mmonitor/routers""fmt""mmonitor/reflec"
)func main() {p := &Person{}h := new(PersonProxy)proxy := reflec.New(p,h)r ,_:= proxy.InvokeMethod("ReflectCallFuncNoArgs", 20, "sssssss")fmt.Println(r...)_, err := proxy.InvokeMethod("ReflectCallFuncNoArgs", 20, 323)fmt.Println(err)
}type Person struct {}func (p *Person)ReflectCallFuncNoArgs(num int, num1 string) string {fmt.Println(num)fmt.Println(num1)return fmt.Sprintf("%d,%s", num, num1)
}type PersonProxy struct {}func (p *PersonProxy)Invoke(proxy *reflec.Proxy, method *reflec.Method, args []interface{})([]interface{}, error){fmt.Println("begin")r , err:= method.Invoke(args...)fmt.Println("end")return r, err
}

结果