ByteBuddy插装代码中如何记录抑制多数异常仅重抛指定异常？

你遇到的核心问题是：自定义Advice.ExceptionHandler的resolve方法属于字节码生成阶段，无法获取运行时抛出的异常实例，所以直接在该方法里判断异常类型的思路走不通。下面提供三种可行的解决方案，不用给每个插装代码手动包裹try-catch：

方案一：利用@Advice.OnMethodExit+@Advice.Thrown直接处理 #

这是最直观的方式，在方法退出阶段直接拿到运行时的异常实例，判断类型后决定是否重抛：

public class SelectiveSuppressionAdvice {
    // 定义允许重抛的异常类型集合
    private static final Set<Class<? extends Throwable>> ALLOWED_EXCEPTIONS = 
        Collections.unmodifiableSet(new HashSet<>(Arrays.asList(
            CriticalBusinessException.class,
            RequiredNotificationException.class
        )));

    @Advice.OnMethodEnter
    public static void enter() {
        // 你的插装逻辑，可能抛出异常
    }

    // 指定onThrowable = Throwable.class，捕获所有异常
    @Advice.OnMethodExit(onThrowable = Throwable.class)
    public static void exit(@Advice.Thrown Throwable throwable) {
        if (throwable != null) {
            boolean isAllowed = ALLOWED_EXCEPTIONS.stream()
                .anyMatch(clazz -> clazz.isInstance(throwable));
            if (!isAllowed) {
                // 记录异常
                LoggerFactory.getLogger(SelectiveSuppressionAdvice.class)
                    .error("插装代码抛出异常已抑制", throwable);
                // 将异常置为null，ByteBuddy会自动停止异常传播
                throwable = null;
            }
            // 允许重抛的异常什么都不用做，ByteBuddy会自动抛出
        }
    }
}

使用时无需额外配置ExceptionHandler，ByteBuddy会自动处理异常的传播逻辑。

方案二：抽取通用异常处理工具类 #

如果多个Advice都需要相同的异常处理逻辑，可以把判断和日志逻辑抽成静态工具类，减少重复代码：

public class InstrumentationExceptionUtil {
    private static final Set<Class<? extends Throwable>> ALLOWED_EXCEPTIONS = 
        Collections.unmodifiableSet(new HashSet<>(Arrays.asList(
            CriticalBusinessException.class,
            RequiredNotificationException.class
        )));

    public static void handleThrowable(@Advice.Thrown Throwable throwable) {
        if (throwable != null && !ALLOWED_EXCEPTIONS.stream().anyMatch(clazz -> clazz.isInstance(throwable))) {
            LoggerFactory.getLogger(InstrumentationExceptionUtil.class)
                .error("插装异常已抑制", throwable);
            throwable = null;
        }
    }
}

然后在Advice中直接调用：

@Advice.OnMethodExit(onThrowable = Throwable.class)
public static void exit(@Advice.Thrown Throwable throwable) {
    InstrumentationExceptionUtil.handleThrowable(throwable);
}

方案三：全局自定义ExceptionHandler（进阶） #

如果需要给所有插装的方法统一设置异常处理规则，可以通过ByteBuddy的StackManipulation API直接生成字节码逻辑，在字节码层面实现类型检查和日志记录：

public class GlobalSelectiveExceptionHandler implements Advice.ExceptionHandler {
    private final Set<TypeDescription> allowedExceptionTypes;

    public GlobalSelectiveExceptionHandler(Set<Class<? extends Throwable>> allowedExceptions) {
        this.allowedExceptionTypes = allowedExceptions.stream()
            .map(TypeDescription.ForLoadedType::new)
            .collect(Collectors.toSet());
    }

    @Override
    public StackManipulation resolve(MethodDescription instrumentedMethod, TypeDescription instrumentedType) {
        LocalVariable exceptionVar = new LocalVariable(TypeDescription.OBJECT);
        // 构建字节码逻辑：保存异常→检查是否允许重抛→允许则抛出，否则记录日志并移除异常
        return new Compound(
            exceptionVar.store(),
            new If(
                // 逻辑或：检查异常是否属于任意允许的类型
                allowedExceptionTypes.stream()
                    .map(InstanceOf::new)
                    .reduce(StackManipulation.FALSE, StackManipulation::or),
                Throw.INSTANCE // 允许重抛，直接抛出
            ),
            // 记录日志
            MethodInvocation.invoke(LoggerFactory.class.getMethod("getLogger", Class.class))
                .with(instrumentedType),
            MethodInvocation.invoke(Logger.class.getMethod("error", String.class, Throwable.class))
                .with("全局抑制插装异常", exceptionVar.load()),
            // 移除异常，停止传播
            Removal.SINGLE
        );
    }
}

使用时在ByteBuddy配置中全局设置：

Set<Class<? extends Throwable>> allowedExceptions = new HashSet<>(Arrays.asList(
    CriticalBusinessException.class,
    RequiredNotificationException.class
));

new ByteBuddy()
    .redefine(TargetClass.class)
    .visit(Advice.to(YourInstrumentationAdvice.class)
        .on(ElementMatchers.any())
        .withExceptionHandler(new GlobalSelectiveExceptionHandler(allowedExceptions)))
    .make();

方案选择建议 #

• 若只有少数Advice需要处理异常：优先用方案一/二，代码直观易调试
• 若需要全局统一处理所有插装方法的异常：用方案三，需要熟悉ByteBuddy字节码操作API

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