[java] Javassist를 이용한 메소드 이동 및 수정

16 Nov 2023

java

Javassist는 자바 코드를 동적으로 생성하고 수정하는 데 사용되는 라이브러리입니다. 이 라이브러리를 사용하여 메소드를 이동하고 수정하는 방법에 대해 알아보겠습니다.

Javassist 설치

Javassist는 Maven 등의 의존성 관리 도구를 사용하여 프로젝트에 추가할 수 있습니다. 다음은 Maven을 통해 Javassist를 추가하는 예시입니다.

<dependencies>
    <dependency>
        <groupId>org.javassist</groupId>
        <artifactId>javassist</artifactId>
        <version>3.27.0-GA</version>
    </dependency>
</dependencies>

메소드 이동하기

Javassist를 사용하여 메소드를 이동하려면 다음 단계를 따르면 됩니다.

ClassPool을 초기화합니다.
원본 클래스(CtClass)를 가져옵니다.
이동할 메소드를 가져옵니다.
메소드를 원하는 위치로 이동시킵니다.
변경된 클래스를 저장합니다.

다음은 Javassist를 사용하여 메소드를 이동하는 예시 코드입니다.

import javassist.*;

public class MethodMover {
    public static void main(String[] args) throws Exception {
        ClassPool classPool = ClassPool.getDefault();
        
        // 원본 클래스 가져오기
        CtClass ctClass = classPool.get("com.example.MyClass");
        
        // 원하는 메소드 가져오기
        CtMethod method = ctClass.getDeclaredMethod("originalMethod");
        
        // 메소드 이동시키기
        ctClass.moveMethod(method, ctClass.getDeclaredMethod("targetMethod"));
        
        // 변경된 클래스 저장하기
        ctClass.writeFile();
    }
}

위 코드에서 com.example.MyClass는 원본 클래스의 패키지와 이름을 나타내며, originalMethod는 이동할 메소드의 이름이고, targetMethod는 메소드를 이동할 위치의 메소드 이름입니다. 변경된 클래스는 ctClass.writeFile()을 통해 저장됩니다.

메소드 수정하기

Javassist를 사용하여 메소드를 수정하려면 다음 단계를 따르면 됩니다.

ClassPool을 초기화합니다.
원본 클래스(CtClass)를 가져옵니다.
수정할 메소드를 가져옵니다.
원하는 수정을 수행합니다.
변경된 클래스를 저장합니다.

다음은 Javassist를 사용하여 메소드를 수정하는 예시 코드입니다.

import javassist.*;

public class MethodModifier {
    public static void main(String[] args) throws Exception {
        ClassPool classPool = ClassPool.getDefault();
        
        // 원본 클래스 가져오기
        CtClass ctClass = classPool.get("com.example.MyClass");
        
        // 수정할 메소드 가져오기
        CtMethod method = ctClass.getDeclaredMethod("originalMethod");
        
        // 메소드 수정하기
        method.insertBefore("System.out.println(\"Before executing the method\");");
        method.insertAfter("System.out.println(\"After executing the method\");");
        
        // 변경된 클래스 저장하기
        ctClass.writeFile();
    }
}

위 코드에서도 com.example.MyClass는 원본 클래스의 패키지와 이름을 나타내며, originalMethod는 수정할 메소드의 이름입니다. method.insertBefore() 및 method.insertAfter()를 사용하여 메소드 안에 원하는 코드를 삽입할 수 있습니다. 변경된 클래스는 ctClass.writeFile()을 통해 저장됩니다.

참고 자료

Javassist 공식 사이트: http://www.javassist.org/
Javassist GitHub 저장소: https://github.com/jboss-javassist/javassist

Javassist는 코드를 동적으로 생성하고 수정하는 기능을 제공하여 자바 언어의 유연성을 확장합니다. 메소드를 이동하고 수정하는 기능을 사용하면 다양한 자바 어플리케이션 개발 시나리오에서 유용하게 활용할 수 있습니다.