[java] Javassist를 통한 상속과 오버라이딩 처리

16 Nov 2023

java

Javassist는 자바 프로그래밍 언어를 사용하여 바이트 코드를 생성, 수정 및 조작하기 위한 라이브러리입니다. Javassist를 사용하면 동적으로 클래스를 생성하고, 기존 클래스의 메소드를 수정하거나 새로운 메소드를 추가할 수 있습니다.

이번 블로그 포스트에서는 Javassist를 사용하여 상속과 오버라이딩을 처리하는 방법에 대해 알아보겠습니다.

Javassist 설치

Javassist를 사용하기 위해서는 먼저 Javassist 라이브러리를 다운로드하고 설정해야 합니다. Javassist는 Maven, Gradle 등의 의존성 관리 도구를 통해 다운로드할 수 있습니다.

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

상속 처리

Javassist를 사용하여 상속을 처리하려면 원본 클래스와 상속받을 클래스를 생성해야 합니다. 아래의 예제 코드를 통해 이를 확인해보겠습니다.

public class OriginalClass {
    public void originalMethod() {
        System.out.println("Original method");
    }
}

import javassist.*;

public class SubClass extends OriginalClass {
    public void newMethod() {
        System.out.println("New method");
    }
}

위의 예제에서는 OriginalClass라는 클래스를 생성하고, SubClass라는 클래스에서 OriginalClass를 상속받고 있습니다.

오버라이딩 처리

Javassist를 사용하여 오버라이딩을 처리하려면 메소드의 수정사항을 반영해야 합니다. 아래의 예제 코드를 통해 이를 확인해보겠습니다.

public class OriginalClass {
    public void originalMethod() {
        System.out.println("Original method");
    }
}

import javassist.*;

public class SubClass extends OriginalClass {
    @Override
    public void originalMethod() {
        System.out.println("Overridden method");
    }
}

위의 예제에서는 OriginalClass라는 클래스의 originalMethod를 SubClass에서 오버라이딩하여 수정한 것을 확인할 수 있습니다.

Javassist를 통한 클래스 로딩

위에서 작성한 클래스들을 동적으로 로딩하기 위해서는 Javassist의 ClassPool을 사용해야 합니다. 다음의 예제 코드를 통해 이를 확인해보겠습니다.

import javassist.*;

public class DynamicLoader {
    public static void main(String[] args) throws Exception {
        ClassPool classPool = ClassPool.getDefault();

        Loader loader = new Loader(classPool);
        loader.addTranslator(classPool, new ClassTranslator());

        CtClass ctClass = classPool.get("com.example.SubClass");
        Class<?> clazz = ctClass.toClass();
        OriginalClass instance = (OriginalClass) clazz.getDeclaredConstructor().newInstance();
        instance.originalMethod();
    }
}

class ClassTranslator implements Translator {
    @Override
    public void start(ClassPool classPool) throws NotFoundException, CannotCompileException {
    }

    @Override
    public void onLoad(ClassPool classPool, String className) throws NotFoundException, CannotCompileException {
        if (className.equals("com.example.OriginalClass")) {
            CtClass ctClass = classPool.get(className);
            ctClass.defrost();
            CtMethod method = ctClass.getDeclaredMethod("originalMethod");
            method.insertAfter("{ System.out.println(\"Intercepted\"); }");
        }
    }
}

위의 예제에서는 DynamicLoader 클래스를 통해 동적으로 클래스를 로딩하여 사용하는 것을 확인할 수 있습니다. ClassTranslator 클래스는 Javassist의 Translator 인터페이스를 구현하여 클래스를 수정합니다.

결론

Javassist를 사용하면 동적으로 클래스를 생성하고, 상속과 오버라이딩을 처리할 수 있습니다. 이를 통해 프로그램의 유연성과 확장성을 향상시킬 수 있습니다.