Golang反射动态代理实现 AOP编程方案

package main

import (
    "fmt"
    "reflect"
)

// 定义切面接口
type Aspect interface {
    Before(args []reflect.Value)
    After(result []reflect.Value)
    OnPanic(panicInfo interface{})
}

// 代理结构体
type Proxy struct {
    target   reflect.Value
    aspect   Aspect
}

// 创建代理实例
func NewProxy(target interface{}, aspect Aspect) interface{} {
    return &Proxy{target: reflect.ValueOf(target), aspect: aspect}
}

// 动态调用方法
func (p *Proxy) Call(methodName string, args ...interface{}) []reflect.Value {
    method := p.target.MethodByName(methodName)
    if !method.IsValid() {
        panic("method not found: " + methodName)
    }

    in := make([]reflect.Value, len(args))
    for i, arg := range args {
        in[i] = reflect.ValueOf(arg)
    }

    var out []reflect.Value
    var panicInfo interface{}

    // 执行前置通知
    p.aspect.Before(in)

    defer func() {
        if panicInfo != nil {
            p.aspect.OnPanic(panicInfo)
        } else {
            p.aspect.After(out)
        }
    }()

    // 调用原始方法
    func() {
        defer func() {
            if err := recover(); err != nil {
                panicInfo = err
            }
        }()
        out = method.Call(in)
    }()

    if panicInfo != nil {
        panic(panicInfo) // 重新抛出
    }

    return out
}

type UserService struct{}

func (u *UserService) GetUser(id int) string {
    fmt.Printf("Loading user %d...\n", id)
    return fmt.Sprintf("User-%d", id)
}

// 日志与耗时监控切面
type LoggingAspect struct{}

func (l LoggingAspect) Before(args []reflect.Value) {
    fmt.Printf("[AOP] 开始调用，参数: %v\n", args)
}

func (l LoggingAspect) After(result []reflect.Value) {
    fmt.Printf("[AOP] 调用完成，返回: %v\n", result)
}

func (l LoggingAspect) OnPanic(panicInfo interface{}) {
    fmt.Printf("[AOP] 方法执行异常: %v\n", panicInfo)
}

func main() {
    userSvc := &UserService{}
    proxy := NewProxy(userSvc, LoggingAspect{}).(*Proxy)

    // 通过代理调用
    result := proxy.Call("GetUser", 1001)
    fmt.Println("最终结果:", result[0].String())
}

[AOP] 开始调用，参数: [0x...]
Loading user 1001...
[AOP] 调用完成，返回: [User-1001]
最终结果: User-1001