方法被截获两次,即使它被调用一次

时间:2015-11-01 17:12:17

标签: java bytecode agent byte-buddy

在以下代码段中,我在doStuff的实例上调用方法Subclass一次。然而它被拦截了两次。

请注意,doStuff已在父类SuperClass中定义。如果在doStuff中定义了SubClass,则拦截逻辑将按预期工作:只有一个拦截。

我是否错误地使用了Byte Buddy?

package com.test;

import static net.bytebuddy.matcher.ElementMatchers.any;
import static net.bytebuddy.matcher.ElementMatchers.nameStartsWith;

import java.util.concurrent.Callable;

import net.bytebuddy.agent.ByteBuddyAgent;
import net.bytebuddy.agent.builder.AgentBuilder;
import net.bytebuddy.description.type.TypeDescription;
import net.bytebuddy.dynamic.DynamicType.Builder;
import net.bytebuddy.implementation.MethodDelegation;
import net.bytebuddy.implementation.bind.annotation.RuntimeType;
import net.bytebuddy.implementation.bind.annotation.SuperCall;

import org.junit.Test;

public class ReproBugTest {

    @Test
    public void reproBug() {

        new AgentBuilder.Default().type(nameStartsWith("com.test"))
                                    .transform(new AgentBuilder.Transformer() {

                                        @Override
                                        public Builder<?> transform(
                                                Builder<?> builder,
                                                TypeDescription td) {

                                            return builder.method(any())
                                                            .intercept(
                                                                    MethodDelegation.to(MethodInterceptor.class));
                                        }
                                    })
                                    .installOn(
                                            ByteBuddyAgent.installOnOpenJDK());

        SubClass subClass = new SubClass();
        subClass.doStuff();
    }
}

class SuperClass {
    public void doStuff() {
        System.out.println("Doing stuff...");
    }
}

class SubClass extends SuperClass {
}

class MethodInterceptor {

    @RuntimeType
    public static Object intercept(@SuperCall Callable<?> zuper)
            throws Exception {

        // Intercepted twice, bug?
        System.out.println("Intercepted");

        Object returnValue = zuper.call();

        return returnValue;
    }
}

1 个答案:

答案 0 :(得分:1)

您正在拦截每种类型的方法调用,即SubclassSuperClass。您需要进一步指定拦截器以拦截哪些方法。在您的情况下,您只想拦截方法,如果它们是由给定类型声明的。

这很容易实现。您应该拦截builder.method(any()),而不是builder.method(isDeclaredBy(td))。这样,只有截取类型声明方法才会截获方法。

最后,我可以从您的源代码中看到您使用的是较旧版本的Byte Buddy。版本 0.7-rc6 运行稳定,具有其他功能并修复了多个错误。 (但是,仍然需要更改某些API。)