[java] Byte Buddy를 사용하여 메소드가 호출될 때 상태를 확인하는 방법은?

20 Nov 2023

java

먼저, Byte Buddy를 프로젝트에 추가해야 합니다. Maven을 사용하는 경우, pom.xml 파일에 다음 의존성을 추가합니다:

<dependency>
    <groupId>net.bytebuddy</groupId>
    <artifactId>byte-buddy</artifactId>
    <version>1.11.22</version>
</dependency>

그런 다음, 다음 예제 코드를 사용하여 메소드 호출 상태를 확인하는 MethodInterceptor를 구현할 수 있습니다.

import net.bytebuddy.ByteBuddy;
import net.bytebuddy.implementation.MethodDelegation;
import net.bytebuddy.matcher.ElementMatchers;

public class MethodLoggingInterceptor implements MethodInterceptor {

    @Override
    public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
        System.out.println("메소드 호출 전: " + method.getName());
        
        Object result = proxy.invokeSuper(obj, args);
        
        System.out.println("메소드 호출 후: " + method.getName());
        
        return result;
    }
    
    public static void main(String[] args) throws IllegalAccessException, InstantiationException {
        Foo foo = new ByteBuddy()
                .subclass(Foo.class)
                .method(ElementMatchers.any())
                .intercept(MethodDelegation
                            .to(new MethodLoggingInterceptor()))
                .make()
                .load(Foo.class.getClassLoader())
                .getLoaded()
                .newInstance();
        
        foo.bar();
    }
}

class Foo {
    public void bar() {
        System.out.println("bar 메소드 실행");
    }
}

위 코드에서 MethodLoggingInterceptor 클래스는 MethodInterceptor 인터페이스를 구현합니다. intercept 메소드에서는 메소드가 호출되기 전과 후에 원하는 동작을 수행할 수 있습니다. 이 예제에서는 호출되는 메소드의 이름을 출력합니다.

main 메소드에서는 ByteBuddy를 사용하여 Foo 클래스를 서브클래스화하고, MethodLoggingInterceptor를 이용하여 모든 메소드 호출을 가로챕니다. 그 후, Foo의 인스턴스를 생성하고 bar 메소드를 호출하게 됩니다.

위 코드를 컴파일하고 실행하면, bar 메소드가 호출되기 전과 후에 “메소드 호출 전: bar”와 “메소드 호출 후: bar”가 출력될 것입니다.

더 자세한 정보는 Byte Buddy 문서를 참조하시기 바랍니다.