您现在的位置：程式碼之家 >> 編程語言 > >> Java 程式設計 >> Java 綜合教學

Byte Buddy 實戰：從零構建高性能 Java Agent 完整指南

編輯：Java 綜合教學

Byte Buddy 實戰：輕鬆構建 Java Agent，實現無侵入式字節碼增強

Java Agent 是一種在目標應用啟動前運行的特殊程序，能通過 Instrumentation API 修改類字節碼，實現 AOP、監控、熱部署等核心功能。傳統字節碼操作（如 ASM）門檻極高，Byte Buddy 以類型安全的 DSL 封裝底層細節，讓普通 Java 開發者也能輕鬆掌握運行時代碼生成。

Java Agent 基礎機制

1. 標準 Agent 入口：premain 方法

public class MyAgent {
    public static void premain(String args, Instrumentation inst) {
        System.setProperty("my-property", "foo"); // 修改環境屬性
        // 註冊 ClassFileTransformer 修改字節碼
        inst.addTransformer(new ClassFileTransformer() {
            @Override
            public byte[] transform(ClassLoader loader, String className,
                                    Class classBeingRedefined,
                                    ProtectionDomain pd, byte[] classFileBuffer) {
                // 返回修改後的字節碼
                return modifiedBytes;
            }
        });
    }
}

2. 打包與啟動

將 Agent 類打包為 JAR，Manifest 中設置 Agent-Class
啟動命令：java -javaagent:myAgent.jar=options -jar app.jar
支持多 Agent 疊加，按命令行順序執行

字節碼操作的痛點與 Byte Buddy 解決方案

痛點	Byte Buddy 方案
需深度理解 JVM 字節碼指令集	類型安全 DSL，以 Java 語法表達字節碼邏輯
手動解析類層級結構（父類、接口）	自動分析類繼承體系，智能匹配方法
反射 API 在類加載前不可用	提供 `@Origin`、`@Super` 等注解注入上下文信息
代碼維護成本高，團隊懼怕修改	代碼可讀性強，降低認知負擔

Byte Buddy 核心 API 速覽

1. 動態生成子類

Class dynamicType = new ByteBuddy()
    .subclass(Object.class)
    .method(ElementMatchers.named("toString"))
    .intercept(FixedValue.value("Hello World!"))
    .make()
    .load(getClass().getClassLoader(), ClassLoadingStrategy.Default.WRAPPER)
    .getLoaded();

2. 方法委托實現邏輯

class ToStringInterceptor {
    static String intercept(@Origin Method method) {
        return "Hello from " + method.getName() + "!";
    }
}

// 使用委托
.intercept(MethodDelegation.to(ToStringInterceptor.class))

支持參數注解：@Origin（方法元信息）、@Super（調用原方法）、@AllArguments（所有參數）、自定義注解。

3. 實戰：註解驅動的方法級安全攔截

@interface Secured { String user(); }

class Service {
    @Secured(user = "ADMIN")
    void doSensitiveAction() { /* 敏感操作 */ }
}

class SecurityInterceptor {
    static String currentUser = "ANONYMOUS";
    static void check(@Origin Method method) {
        if (!method.getAnnotation(Secured.class).user().equals(currentUser))
            throw new IllegalStateException("Unauthorized");
    }
}

// 子類模式（需工廠創建實例）
new ByteBuddy()
    .subclass(Service.class)
    .method(ElementMatchers.isAnnotatedWith(Secured.class))
    .intercept(MethodDelegation.to(SecurityInterceptor.class).andThen(SuperMethodCall.INSTANCE))
    .make().load(...).getLoaded().newInstance().doSensitiveAction();

4. Agent 模式：Rebase 重定義原類（無需工廠）

class SecurityAgent {
    public static void premain(String args, Instrumentation inst) {
        new AgentBuilder.Default()
            .type(ElementMatchers.any())
            .transform((builder, type) -> builder
                .method(ElementMatchers.isAnnotatedWith(Secured.class))
                .intercept(MethodDelegation.to(SecurityInterceptor.class)
                    .andThen(SuperMethodCall.INSTANCE)))
            .installOn(inst);
    }
}

優勢：直接修改原類字節碼，new Service().doSensitiveAction() 即可生效；支持單元測試無感運行；多 Agent 疊加無耦合。