Byte Buddy能否通过Advice方法监控Java比较运算符？求参考示例代码

完全可以通过Byte Buddy实现这类比较运算符的监控，核心思路是利用Byte Buddy的Advice机制拦截Java字节码中对应比较操作的指令——Java的比较运算符会被编译为特定的字节码指令（比如引用相等==对应if_acmpeq、引用不等!=对应if_acmpne，数值比较</>对应if_icmplt/if_icmpgt等），通过Advice可以在这些指令执行前后插入监控逻辑。

以下是可直接参考的示例代码：

1. 定义监控逻辑的Advice类 #

这个Advice会拦截方法内的引用比较（==/!=）和数值比较操作，捕获操作数与比较结果：

import net.bytebuddy.asm.Advice;

public class ComparisonMonitorAdvice {

    // 监控引用类型的==/!=操作
    @Advice.OnMethodEnter
    public static void monitorReferenceComparison(
            @Advice.Argument(0) Object left,
            @Advice.Argument(1) Object right,
            @Advice.Origin String methodName,
            @Advice.Local("comparisonType") String comparisonType) {
        comparisonType = "reference_comparison";
        System.out.printf("[引用比较] 方法:%s, 左操作数:%s, 右操作数:%s%n", methodName, left, right);
    }

    // 监控数值类型的<>/<=/>=/==/!=操作
    @Advice.OnMethodEnter
    public static void monitorNumericComparison(
            @Advice.Argument(0) int left,
            @Advice.Argument(1) int right,
            @Advice.Origin String methodName,
            @Advice.Local("comparisonType") String comparisonType) {
        comparisonType = "numeric_comparison";
        System.out.printf("[数值比较] 方法:%s, 左操作数:%d, 右操作数:%d%n", methodName, left, right);
    }

    // 捕获比较结果（可选）
    @Advice.OnMethodExit
    public static void captureComparisonResult(
            @Advice.Return(readOnly = false) boolean result,
            @Advice.Local("comparisonType") String comparisonType) {
        System.out.printf("[比较结果] 类型:%s, 结果:%b%n", comparisonType, result);
    }
}

2. 构建Byte Buddy Agent实现拦截 #

通过AgentBuilder指定要拦截的方法和对应的字节码指令，将Advice插入到目标方法中：

import net.bytebuddy.agent.ByteBuddyAgent;
import net.bytebuddy.agent.builder.AgentBuilder;
import net.bytebuddy.asm.Advice;
import net.bytebuddy.matcher.ElementMatchers;

import java.lang.instrument.Instrumentation;

public class ComparisonMonitorAgent {

    public static void premain(String args, Instrumentation inst) {
        new AgentBuilder.Default()
                // 匹配所有需要监控的类（可自定义匹配规则，比如指定包名）
                .type(ElementMatchers.nameStartsWith("com.yourpackage"))
                // 匹配包含比较指令的方法并插入Advice
                .transform((builder, typeDescription, classLoader, module) ->
                        builder.visit(Advice.to(ComparisonMonitorAdvice.class)
                                // 拦截引用比较指令：if_acmpeq（==）、if_acmpne（!=）
                                .on(ElementMatchers.isInstruction(net.bytebuddy.implementation.bytecode.Opcode.IF_ACMPEQ)
                                        .or(ElementMatchers.isInstruction(net.bytebuddy.implementation.bytecode.Opcode.IF_ACMPNE)))
                                // 拦截数值比较指令：<、>、<=、>=、==、!=对应的字节码指令
                                .or(ElementMatchers.isInstruction(net.bytebuddy.implementation.bytecode.Opcode.IF_ICMPLT)
                                        .or(ElementMatchers.isInstruction(net.bytebuddy.implementation.bytecode.Opcode.IF_ICMPGT))
                                        .or(ElementMatchers.isInstruction(net.bytebuddy.implementation.bytecode.Opcode.IF_ICMPLE))
                                        .or(ElementMatchers.isInstruction(net.bytebuddy.implementation.bytecode.Opcode.IF_ICMPGE))
                                        .or(ElementMatchers.isInstruction(net.bytebuddy.implementation.bytecode.Opcode.IF_ICMPEQ))
                                        .or(ElementMatchers.isInstruction(net.bytebuddy.implementation.bytecode.Opcode.IF_ICMPNE)))
                        )
                ).installOn(inst);
    }

    // 测试用：直接启动Agent并执行测试代码
    public static void main(String[] args) {
        Instrumentation inst = ByteBuddyAgent.install();
        premain(null, inst);

        testComparisons();
    }

    private static void testComparisons() {
        String param1 = "test";
        String param2 = null;
        int a = 1, b = 2;

        System.out.println("执行测试比较操作...");
        boolean refEq = param1 == param2;
        boolean refNeq = param1 != null;
        boolean numLt = a < b;
        boolean numGt = b > a;

        System.out.println("测试结果：");
        System.out.println("param1 == param2: " + refEq);
        System.out.println("param1 != null: " + refNeq);
        System.out.println("a < b: " + numLt);
        System.out.println("b > a: " + numGt);
    }
}

关键说明 #

• 字节码指令匹配：通过ElementMatchers.isInstruction()指定要拦截的比较指令，不同比较运算符对应不同的字节码指令，需根据需求精准匹配。
• 操作数捕获：Advice的@Advice.Argument可捕获比较操作的两个操作数，注意引用类型和数值类型需要分开处理（字节码中两者的操作数栈处理逻辑不同）。
• 结果捕获：通过@Advice.OnMethodExit的@Advice.Return可获取比较的布尔结果，结合@Advice.Local传递比较类型信息。

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