基于ByteBuddy的Java Agent在Gradle+JUnit5测试中类未转换问题

根因定位 #

你遇到的问题核心是Gradle测试执行流程的类加载顺序和IDEA不一致：
IDEA运行测试时会将业务代码打包后执行，目标类在Agent初始化后才加载，ByteBuddy可以正常完成类转换；
Gradle运行测试时，目标业务类在测试类初始化、Agent安装前就被预加载，默认ByteBuddy仅会处理加载过程中的类，不会主动重转换已加载的类，最终导致拦截失效。你日志中出现的CallSiteClassLoader属于Groovy运行时生成的内部调用点类加载器，匹配到的$exposedMethod结尾的类是Groovy动态生成的内部类，不属于业务类，IGNORE是正常行为，不是问题根源。

可行解决步骤 #

1. 开启ByteBuddy类重转换能力 #

修改Agent初始化代码，显式配置允许重转换已加载的类，适配类提前加载的场景：

new AgentBuilder.Default()
        // 原有类匹配逻辑不变
        .type(isAnnotatedWith(TimeProfiling.class))
        .transform((builder, typeDescription, classLoader, module) ->
                builder.method(isDeclaredBy(typeDescription) // 匹配当前类所有声明的方法，避免遗漏非public方法
                        .and(not(isAnnotatedWith(Generated))))
                        .intercept(MethodDelegation.to(TimeProfilingInterceptor.class)))
        // 新增重转换配置
        .with(AgentBuilder.RedefinitionStrategy.RETRANSFORM)
        .with(AgentBuilder.TypeStrategy.Default.REDEFINE)
        .with(AgentBuilder.InitializationStrategy.NoOp.INSTANCE)
        // 原有日志配置不变
        .with(stdoutToLoggerWriter.withTransformationsOnly())
        .installOn(instrumentation)

JVM类重转换仅支持修改方法体，不支持增减类的字段、方法，你的打点拦截场景完全符合约束。

2. 调整Agent加载时机（更稳妥的方案） #

直接通过Gradle测试任务的JVM参数提前挂载Agent，确保Agent在所有业务类加载前完成初始化，和生产环境运行逻辑完全一致：
在你的build.gradle中添加如下配置：

test {
    // 替换为你实际的Agent包路径，如果是用ByteBuddy动态挂载也可以直接指定ByteBuddy Agent路径
    jvmArgs "-javaagent:${project(":agent-module").buildDir}/libs/byte-profiling-agent.jar"
}

这种方式不需要修改Agent代码，也能彻底避免类提前加载的问题。

3. 验证拦截逻辑有效性 #

如果配置后还是没有触发拦截，依次排查：

• 检查TimeProfilingInterceptor的方法签名是否匹配拦截要求：是否加了@RuntimeType注解？是否同时支持静态方法、实例方法的拦截？
• 检查注解的保留策略：@TimeProfiling注解是否设置了@Retention(RetentionPolicy.RUNTIME)，如果是CLASS或SOURCE保留级，ByteBuddy运行时无法识别到注解。
• 显式排除Groovy内部类加载器：添加类加载器过滤逻辑，避免内部类干扰匹配：

  .type(isAnnotatedWith(TimeProfiling.class), not(classLoader(is(CallSiteClassLoader.class))))
  

内容的提问来源于stack exchange，提问作者Danila Zharenkov