不使用premain/Agent如何修改替换第三方类的方法

你不需要依赖premain的agent机制，完全可以按照最初的思路（读取类资源→修改字节码→注入ClassLoader）实现需求，核心是解决修改后字节码的类加载注入问题，具体实现如下：

实现方案 #

这个方案不需要任何agent相关配置，仅需在目标类加载前执行注入逻辑即可：

1. 编写ByteBuddy方法拦截逻辑，在目标shutdown()方法入口插入需要的System.out.println("ho hum");调用
2. 从当前上下文ClassLoader读取目标第三方类的原始class字节流
3. 用ByteBuddy对原始字节码进行重定义，生成修改后的字节数组
4. 通过反射注入的方式，将修改后的字节数组注册到对应ClassLoader中，后续类加载时会直接使用修改后的版本

参考实现代码：

import net.bytebuddy.ByteBuddy;
import net.bytebuddy.dynamic.DynamicType;
import net.bytebuddy.dynamic.loading.ClassInjector;
import net.bytebuddy.implementation.MethodDelegation;
import net.bytebuddy.implementation.bind.annotation.RuntimeType;
import net.bytebuddy.implementation.bind.annotation.SuperCall;
import net.bytebuddy.matcher.ElementMatchers;
import java.io.IOException;
import java.io.InputStream;
import java.util.Collections;
import java.util.concurrent.Callable;

public class ThirdPartyClassPatcher {
    // 方法拦截器，用于插入前置逻辑
    public static class ShutdownMethodInterceptor {
        @RuntimeType
        public static Object run(@SuperCall Callable<?> originalMethod) throws Exception {
            // 自定义前置逻辑
            System.out.println("ho hum");
            // 执行原始方法逻辑
            return originalMethod.call();
        }
    }

    /**
     * 注入修改后的类逻辑，必须在目标类被加载前调用
     * @param targetClassName 第三方类的全限定名，例：com.example.ThirdPartyService
     */
    public static void patchShutdownMethod(String targetClassName) throws IOException {
        ClassLoader targetClassLoader = Thread.currentThread().getContextClassLoader();
        String classPath = targetClassName.replace('.', '/') + ".class";

        // 读取原始类字节码
        byte[] originalClassBytes;
        try (InputStream classStream = targetClassLoader.getResourceAsStream(classPath)) {
            if (classStream == null) {
                throw new IOException("未找到目标类资源：" + targetClassName);
            }
            originalClassBytes = classStream.readAllBytes();
        }

        // 重定义字节码，拦截无参shutdown方法
        DynamicType.Unloaded<?> modifiedType = new ByteBuddy()
                .redefine(
                        Object.class,
                        net.bytebuddy.dynamic.ClassFileLocator.Simple.of(targetClassName, originalClassBytes)
                )
                .name(targetClassName)
                .method(ElementMatchers.named("shutdown")
                        .and(ElementMatchers.isPublic())
                        .and(ElementMatchers.takesNoArguments())
                        .and(ElementMatchers.returns(void.class)))
                .intercept(MethodDelegation.to(ShutdownMethodInterceptor.class))
                .make();

        // 用ByteBuddy封装的反射注入工具，把修改后的字节码注入到目标ClassLoader
        ClassInjector.UsingReflection.of(targetClassLoader)
                .inject(Collections.singletonMap(modifiedType.getTypeDescription(), modifiedType.getBytes()));
    }

    public static void main(String[] args) throws IOException {
        // 注意：必须在目标类被任何代码引用、加载前执行patch逻辑
        patchShutdownMethod("替换成你要修改的第三方类全限定名");
        // 后续正常使用第三方类即可，调用shutdown方法时会自动先执行打印逻辑
    }
}

关键注意事项 #

• 严格保证patch逻辑在目标类第一次被加载前执行：不要在patch方法调用前出现任何目标类的静态引用（比如TargetClass.class、new TargetClass()），否则类会被提前加载，后续注入会抛出LinkageError，完全匹配你提到的「目标类尚未被加载」的场景
• 方法匹配规则可按需调整：如果存在同名重载方法，可以额外增加匹配规则（比如匹配抛出IOException的方法），避免误拦截其他方法
• 多ClassLoader环境适配：如果是Web容器、OSGi这类多ClassLoader场景，要确保注入逻辑使用的ClassLoader和后续加载第三方类的ClassLoader一致，否则会加载到未修改的原始类
• JDK高版本兼容：JDK16+默认开启强模块封装，需要添加启动参数--add-opens java.base/java.lang=ALL-UNNAMED，允许反射调用类加载相关方法，这是JDK模块系统的通用限制，不是ByteBuddy的特有要求

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