[java] Byte Buddy를 사용하여 동적 프록시를 생성하는 방법은?

20 Nov 2023

java

Java에서 동적 프록시를 생성하려면 Byte Buddy 라이브러리를 사용할 수 있습니다. Byte Buddy는 강력한 동적 코드 생성 및 조작 라이브러리로, 자바 바이트 코드를 생성하고 수정하는 데 사용할 수 있습니다.

다음은 Byte Buddy를 사용하여 동적 프록시를 생성하는 단계별 방법입니다.

Byte Buddy 라이브러리를 프로젝트에 추가해야 합니다. Maven을 사용한다면 pom.xml 파일에 다음 의존성을 추가합니다.

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

프록시로 사용할 인터페이스를 작성합니다. 예를 들어, UserService 인터페이스를 사용하는 프록시를 생성하려면 다음과 같이 인터페이스를 정의합니다.

public interface UserService {
    void doSomething();
}

동적 프록시를 생성하는 코드를 작성합니다. 다음 예제는 UserService 인터페이스의 메서드 호출을 가로채어 추가 로직을 수행하는 동적 프록시를 생성하는 방법을 보여줍니다.

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

public class ProxyExample {

    public static void main(String[] args) throws IllegalAccessException, InstantiationException {
        Class<? extends UserService> proxyClass = new ByteBuddy()
                .subclass(UserService.class)
                .method(ElementMatchers.any())
                .intercept(MethodDelegation.to(MyInterceptor.class))
                .make()
                .load(UserService.class.getClassLoader())
                .getLoaded();

        UserService proxy = proxyClass.newInstance();
        proxy.doSomething();
    }
}

public class MyInterceptor {
    public static void intercept(@Origin Method method, @AllArguments Object[] args) {
        System.out.println("Before method: " + method.getName());
        // 추가 로직 수행
    }
}

위의 예제에서는 UserService 인터페이스의 모든 메서드를 가로채어 MyInterceptor 클래스의 intercept 메서드가 호출되도록 설정하였습니다. MyInterceptor 클래스에서는 메서드 호출 이전에 추가 로직을 수행할 수 있습니다.

동적 프록시를 사용하여 작성한 코드를 실행합니다. 위의 예제에서는 UserService의 doSomething 메서드를 호출하고 있습니다. 이 메서드는 프록시를 통해 가로채어진 후 MyInterceptor 클래스의 intercept 메서드가 실행될 것입니다.

Byte Buddy를 사용하여 동적 프록시를 생성하는 방법에 대해 알아보았습니다. Byte Buddy는 매우 강력하고 유연한 라이브러리이며, 동적 코드 생성이나 수정이 필요한 경우에 유용하게 사용할 수 있습니다.

참고 자료: