ByteBuddy快速实现AOP

1、定义一个Log注解

@Retention(RetentionPolicy.RUNTIME)
public @interface Log {
}

2、定义一个Service

public class Service {
    @Log
    public int foo(int value) {
        System.out.println("foo: " + value);
        return value;
    }

    public int bar(int value) {
        System.out.println("bar: " + value);
        return value;
    }
}

3、通过byteBuddy对service实现字节码增强

public static void main(String[] args) throws Exception {
        Service service = new ByteBuddy()
        		// 定义父类
                .subclass(Service.class)
                // 定义要拦截的方法
                .method(ElementMatchers.any())
                // 对拦截的方法动态增强,增强逻辑LoggerAdvisor
                .intercept(Advice.to(LoggerAdvisor.class))
                // 编译生成字节码
                .make()
                // 通过类加载器加载(AppClassLoader)
                .load(Service.class.getClassLoader())
                // 获取class对象
                .getLoaded()
                .newInstance();
        service.bar(111);
        service.foo(222);
    }

4、增强逻辑

@RuntimeType 注解:
告诉 Byte Buddy 不要进行严格的参数类型检测,在参数匹配失败时,尝试使用类型转换方式(runtime type casting)进行类型转换,匹配相应方法。

@This 注解:
注入被拦截的目标对象。

@AllArguments 注解:
注入目标方法的全部参数,是不是感觉与 Java 反射的那套 API 有点类似了?

@Origin 注解:
注入目标方法对应的 Method 对象。如果拦截的是字段的话,该注解应该标注到 Field 类型参数。

@Super 注解:
注入目标对象。通过该对象可以调用目标对象的所有方法。

@SuperCall:
这个注解比较特殊,我们要在 intercept() 方法中调用目标方法的话,需要通过这种方式注入,

@SuperCall与 Spring AOP 中的 ProceedingJoinPoint.proceed() 方法有点类似,需要注意的是,这里不能修改调用参数,从上面的示例的调用也能看出来,参数不用单独传递,都包含在其中了。

class LoggerAdvisor {
    @Advice.OnMethodEnter
    public static void onMethodEnter(@Advice.Origin Method method, @Advice.AllArguments Object[] arguments) {
        if (method.getAnnotation(Log.class) != null) {
            System.out.println("Enter " + method.getName() + " with arguments: " + Arrays.toString(arguments));
        }
    }

    @Advice.OnMethodExit
    public static void onMethodExit(@Advice.Origin Method method, @Advice.AllArguments Object[] arguments, @Advice.Return Object ret) {
        if (method.getAnnotation(Log.class) != null) {
            System.out.println("Exit " + method.getName() + " with arguments: " + Arrays.toString(arguments) + " return: " + ret);
        }
    }
}

5、执行结果

bar: 111
Enter foo with arguments: [111]
foo: 222
Exit foo with arguments: [222] return: 222

你可能感兴趣的:(JavaAgent,java,spring,jvm)