go 切面 AOP 实现

go AOP 实现

使用Go语言的反射机制和函数类型实现AOP,通过在需要切入的函数前后添加额外的逻辑代码实现AOP

go 复制代码
package main

import (
	"errors"
	"fmt"
	"log"
	"reflect"
)

// User 结构体表示一个用户
type User struct {
	ID   int
	Name string
}

type Aspect struct {
	Before []func([]reflect.Value) error
	After  []func([]reflect.Value) error
}

// Apply 利用反射执行切面的前置和后置处理函数
func (a *Aspect) Apply(targetFunc interface{}, args []reflect.Value) ([]reflect.Value, error) {
	for _, beforeFunc := range a.Before {
		if err := beforeFunc(args); err != nil {
			return nil, err
		}
	}

	result := reflect.ValueOf(targetFunc).Call(args)
	for _, v := range result {
		if v.Type().AssignableTo(reflect.TypeOf((*error)(nil)).Elem()) {	
			if v.IsNil() {
				continue
			}
			if err, ok := v.Interface().(error); ok {
				if err != nil {
					return nil, err
				}
			} else {  
				log.Panicln("Failed to convert return value to error")  
			}  
		}
	}

	for _, afterFunc := range a.After {
		go afterFunc(args)
	}

	return result, nil
}

func BeforeAspect1(args []reflect.Value) error {
	id := args[0].Interface().(int)
	log.Println("BeforeAspect1 get id:", id)
	return nil
}

func BeforeAspect2(args []reflect.Value) error {
	name := args[1].Interface().(string)
	log.Println("BeforeAspect2 get name:", name)
	return nil
}

func BeforeAspect3(args []reflect.Value) error {
	user := args[2].Interface().(*User)
	log.Println("BeforeAspect3 get user:", user.ID, user.Name)
	return nil
	return errors.New("BeforeAspect3 error")
}

func AfterAspect1(args []reflect.Value) error {
	log.Println("AfterAspect1")
	return nil
}

func TargetFunction(id int, name string, user *User) (*User, error) {
	user.ID = 13
	user.Name = "John Doe"
	fmt.Printf("Executing target function with ID: %d, Name: %s, User: %+v\n", id, name, user)
	return user, nil
}

func main() {
	myAspect := &Aspect{
		Before: []func([]reflect.Value) error{
			BeforeAspect1, 
			BeforeAspect2,
			BeforeAspect3,
		},
		After: []func([]reflect.Value) error{
			AfterAspect1,
		},
	}

	// 使用切面执行目标函数
	targetFuncValue := reflect.ValueOf(TargetFunction)

	// 构造目标函数的参数值
	id := 1
	name := "John"
	user := &User{ID: 100, Name: "Alice"}
	args := []reflect.Value{
		reflect.ValueOf(id),
		reflect.ValueOf(name),
		reflect.ValueOf(user),
	}

	result, err := myAspect.Apply(targetFuncValue.Interface(), args)
	if err != nil {
		fmt.Println("have err:", err)
		return
	}

	fmt.Println("Target function result:", result)
}
相关推荐
投笔丶从戎23 分钟前
Kotlin Multiplatform--01:项目结构基础
android·开发语言·kotlin
杜小暑1 小时前
动态内存管理
c语言·开发语言·动态内存管理
想不明白的过度思考者1 小时前
Java从入门到“放弃”(精通)之旅——JavaSE终篇(异常)
java·开发语言
我真的不会C1 小时前
QT窗口相关控件及其属性
开发语言·qt
CodeCraft Studio1 小时前
Excel处理控件Aspose.Cells教程:使用 Python 在 Excel 中进行数据验
开发语言·python·excel
火柴盒zhang2 小时前
websheet之 编辑器
开发语言·前端·javascript·编辑器·spreadsheet·websheet
景天科技苑2 小时前
【Rust】Rust中的枚举与模式匹配,原理解析与应用实战
开发语言·后端·rust·match·enum·枚举与模式匹配·rust枚举与模式匹配
阿让啊2 小时前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
椰羊~王小美2 小时前
LeetCode -- Flora -- edit 2025-04-25
java·开发语言
孞㐑¥2 小时前
C++11介绍
开发语言·c++·经验分享·笔记