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