UserServiceTx 를 다이내믹 프록시 방식으로 변경해보자.

현재 상태에서 고쳐야 할 점

• 서비스 인터페이스의 모든 메서드를 구현해야 한다.
• 트랜잭션이 필요한 메서드마다 트랜잭션 처리 코드 중복이 나타난다.
• 트랜잭션이 필요한 클래스와 메서드가 증가하면 프록시 클래스를 일일이 구현해야 한다.

다이내믹 프록시와 연동해서 트랜잭션 기능을 부가하도록 InvocationHandler 를 구현하자

트랜잭션 InvocationHandler

다이내믹 프록시를 위한 트랜잭션 부가기능을 갖는 객체를 구현하자.

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.DefaultTransactionDefinition;

public class TransactionHandler implements InvocationHandler {

  private Object target;

  private PlatformTransactionManager transactionManager;
  private String pattern;
  
  @Override
  public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    if(method.getName().startsWith(pattern)){
      return invokeInTransaction(method, args);
    }else return method.invoke(target, args);
  }
  
  private Object invokeInTransaction(Method method, Object[] args) throws Throwable{
    TransactionStatus status = this.transactionManager.getTransaction(new DefaultTransactionDefinition());
    try{
      Object ret = method.invoke(target, args);
      this.transactionManager.commit(status);
      return ret;
    } catch(InvocationTargetException e){
      this.transactionManager.rollback(status);
      throw e.getTargetException();
    }
  }

  public void setTarget(Object target) {
    this.target = target;
  }

  public void setTransactionManager(
      PlatformTransactionManager transactionManager) {
    this.transactionManager = transactionManager;
  }

  public void setPattern(String pattern) {
    this.pattern = pattern;
  }
}

InvocationHandler 의 invoke() 메서드

• 타깃 오브젝트의 메서드에 선별적으로 트랜잭션을 적용한다.
  • 적용 대상 선정 작업 먼저 진행
• DI 받은 이름 패턴으로 시작되는 이름을 가진 메서드인지 확인
  • 패턴과 일치 → 트랜잭션 적용 메서드 호출
  • 패턴과 불일차 → 타깃 오브젝트 메서드 호출하여 결과 리턴

UserServiceTx 와 다이내믹 프록시에서 비교

• 공통점: 트랜잭션을 적용하면서 타깃 오브젝트의 메서드를 호출한다.
• 차이점
  • 롤백을 적용하기 위한 예외는 RuntimeException 이 아니라 InvocationTargetException 을 적용한다.
  • Method.invoke() 사용 시 예외가 포장되어 나온다.

Method.invoke()