时间戳

首页 > 解决方案 > 无法使用 ByteBuddy 代理转换类

java - 无法使用 ByteBuddy 代理转换类

问题描述

我实际上正在使用 ByteBuddy API 编写一个 Java 代理，我需要在其中监视一些方法。比如说我需要记录一个方法的执行时间。

这是我的代码：

    public class PerfAgents {

     public static void premain(String agentArgs, Instrumentation inst){
        LOGGER.info("[Agent] Loading classes ...");
        Class classToMonitor = getClassFromArgs(agentArgs);
        String methodToMonitor = getMethodFromArgs(agentArgs);
        installAgent(inst, classToMonitor, methodToMonitor);
    }

     private static void installAgent(Instrumentation instrumentation, Class<?> classToMonitor, String methodToMonitor) {
      new AgentBuilder.Default()
           .type(is(classToMonitor))
           .transform((builder, typeDescription, classLoader, module) ->
                {
                    LOGGER.info("Transforming {} for {}", method, classToMonitor.getSimpleName());
                    return builder.method(named(methodToMonitor))
                           .intercept(MethodDelegation.to(TimerInterceptor.class));
           }).installOn(instrumentation);
  }

}

这TimerInterceptor类似于LoggerInterceptor在 ByteBuddy 教程中找到的，我在其中使用了@SuperCall注释。

问题不在于我不确定 ByteBuddy 是否将转换应用于提供的类和方法。我可以看到代理正在我的应用程序中加载，但是当我执行我的方法进行监视时，什么也没有发生。

这是我的 TimerInterceptor 类：

static class TimerInterceptor {

    private static Logger LOGGER = LoggerFactory.getLogger(LogInterceptor.class);


    public static Object log(@SuperCall Callable<Object> callable) throws Exception {
        LocalTime start = LocalTime.now();
        Object called = callable.call();
        LocalTime end = LocalTime.now();
        Duration between = Duration.between(start, end);

        LOGGER.info("Execution time : {} ms", between.toMillis());
        return called;
    }
}

任何帮助，将不胜感激。

标签： javabyte-buddyjavaagents

解决方案

您没有将 Byte Buddy 配置为重新转换已加载的类。您可以通过.with(RetransformationStrategy.RETRANSFORM)在代理生成器 DSL 中进行设置来实现。

如果您可以避免重新转换，即如果您仅检测执行代理时未加载的应用程序类，则可以跳过此步骤。相反，使用基于字符串的匹配器并且不加载类。如果需要更丰富的描述，也可以使用 aTypePool.Default让 Byte Buddy 解析类文件而不加载类。

要查看 Byte Buddy 正在做什么，您可以注册一个Listener.StreamWriting.toSystemOut()所有发现的类都打印到控制台的位置，包括任何潜在的错误。

推荐阅读