[java] Java Byte Buddy를 사용한 AOP 구현

23 Nov 2023

java

Aspect-Oriented Programming (AOP)는 소프트웨어 개발에서 공통적인 관심사를 모듈화할 수 있는 기술입니다. AOP를 사용하면 코드의 가독성과 재사용성을 향상시킬 수 있습니다. Java에서는 자체적인 AOP 지원을 제공하지 않지만, Byte Buddy 라이브러리를 사용하여 AOP를 구현할 수 있습니다.

Byte Buddy는 Java 에이전트를 사용하여 동적으로 클래스를 수정하고 생성하는 데 사용되는 라이브러리입니다. 이 라이브러리를 사용하면 런타임 중에 클래스의 동작을 수정할 수 있습니다. AOP를 구현하기 위해 Byte Buddy가 제공하는 Annotation을 사용할 수 있습니다.

Java Byte Buddy를 사용하여 AOP를 구현하는 단계는 다음과 같습니다:

1. Maven 또는 Gradle을 사용하여 라이브러리 추가

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

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

Gradle을 사용하는 경우 build.gradle 파일에 다음을 추가합니다:

dependencies {
    implementation 'net.bytebuddy:byte-buddy:1.10.10'
}

2. Aspect 클래스 작성

AOP를 적용할 클래스를 생성합니다. 이 클래스는 @Advice 어노테이션을 사용하여 Advice 메서드를 정의합니다. 이 메서드는 타겟 메서드를 감싸는 코드를 포함합니다. 일반적으로 메서드 호출 전/후에 수행할 작업을 정의합니다.

import net.bytebuddy.asm.Advice;

public class MyAspect {
    
    @Advice.OnMethodEnter
    static void enter(@Advice.Origin Method method, @Advice.AllArguments Object[] arguments) {
        System.out.println("Method entered: " + method.getName());
    }
    
    @Advice.OnMethodExit
    static void exit(@Advice.Origin Method method, @Advice.AllArguments Object[] arguments, @Advice.Return Object result) {
        System.out.println("Method exited: " + method.getName());
    }
    
}

위의 코드에서 @Advice.OnMethodEnter 어노테이션은 Advice 메서드가 타겟 메서드 호출 전에 실행된다는 것을 나타냅니다. @Advice.OnMethodExit 어노테이션은 Advice 메서드가 타겟 메서드 호출 후에 실행된다는 것을 나타냅니다.

3. AOP 적용

AOP를 원하는 클래스 또는 메서드에 Annotation을 적용하여 Aspect를 적용할 수 있습니다. 아래 예제에서는 MyClass 클래스의 myMethod 메서드에 AOP를 적용합니다:

import net.bytebuddy.agent.builder.AgentBuilder;

AgentBuilder.Transformer transformer = (builder, typeDescription, classLoader, module) ->
    builder.visit(Advice.to(MyAspect.class).on(named("myMethod")));

new AgentBuilder.Default()
    .type(named("com.example.MyClass"))
    .transform(transformer)
    .installOnByteBuddyAgent();

위의 코드에서 MyClass 클래스의 myMethod 메서드에 MyAspect 클래스의 Advice를 적용합니다. 이를 위해 AgentBuilder와 Transformer를 사용합니다.

4. 실행

Aspect를 적용하여 수정된 클래스를 실행합니다. AOP가 적용된 메서드가 호출될 때, MyAspect 클래스의 Advice 메서드가 실행됩니다.

public class MyClass {
    
    public void myMethod() {
        System.out.println("Executing myMethod");
    }
    
    public static void main(String[] args) {
        MyClass myClass = new MyClass();
        myClass.myMethod();
    }
    
}

실행 시 출력 결과는 다음과 같습니다:

Method entered: myMethod
Executing myMethod
Method exited: myMethod

위의 예제에서는 간단한 AOP 구현을 보여주기 위해 System.out.println을 사용하였지만, 더 복잡한 로직을 포함하는 Advice를 작성할 수도 있습니다.

Byte Buddy를 사용하면 Java에서 AOP를 구현하는 것이 쉬워집니다. Byte Buddy의 많은 기능과 어노테이션을 통해 AOP에 대한 세부 제어도 가능합니다. 공통 관심사를 모듈화하고 코드의 재사용성을 높이기 위해 AOP를 사용해 보세요!

참고 자료

Byte Buddy 공식 문서