[java] Byte Buddy를 사용하여 로드 타임 위빙을 적용하는 방법은?

20 Nov 2023

java

Byte Buddy 라이브러리를 의존성으로 추가합니다. 프로젝트의 pom.xml 파일에 다음 의존성을 추가합니다.

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

로드 타임 위빙을 적용할 클래스를 작성합니다. 예를 들어, MyClass라는 클래스를 가정해봅시다.

public class MyClass {
    public void myMethod() {
        System.out.println("Hello, World!");
    }
}

로드 타임 위빙 에이전트 클래스를 작성합니다. 로드 타임 위빙 에이전트 클래스는 Instrumentation 인스턴스를 사용하여 바이트 코드를 변경합니다. MyClass의 메서드를 변경하여 로그를 출력하도록 해보겠습니다.

import net.bytebuddy.agent.ByteBuddyAgent;
import net.bytebuddy.agent.builder.AgentBuilder;
import net.bytebuddy.agent.builder.AgentBuilder.RedefinitionStrategy;
import net.bytebuddy.asm.Advice;
import net.bytebuddy.matcher.ElementMatchers;
import java.lang.instrument.Instrumentation;

public class MyAgent {
    public static void premain(String agentArgs, Instrumentation inst) {
        new AgentBuilder.Default()
            .type(ElementMatchers.nameContains("MyClass")) // 대상 클래스 선택
            .transform((builder, type, classLoader, module) ->
                builder.method(ElementMatchers.named("myMethod")) // 대상 메서드 선택
                       .intercept(Advice.to(MyInterceptor.class)) // 어드바이스 적용
            )
            .with(RedefinitionStrategy.RETRANSFORMATION)
            .installOn(inst);
    }

    public static class MyInterceptor {
        @Advice.OnMethodEnter
        public static void enter() {
            System.out.println("Before method execution");
        }

        @Advice.OnMethodExit
        public static void exit() {
            System.out.println("After method execution");
        }
    }
}

애플리케이션을 실행할 때 로드 타임 위빙 에이전트를 사용하도록 아래와 같이 JVM 옵션을 추가합니다.

-javaagent:/path/to/MyAgent.jar

애플리케이션을 실행하면 MyClass의 myMethod가 호출될 때마다 어드바이스에서 정의한 로그가 출력됩니다.

Byte Buddy를 사용하여 로드 타임 위빙을 적용하는 방법에 대해 알아보았습니다. Byte Buddy는 다양한 바이트 코드 조작 기능을 지원하므로 유연하고 강력한 도구이며, 다양한 케이스에 적용할 수 있습니다.

참고 자료:

Byte Buddy 공식 문서