[DB 접근 기술] 트랜잭션 (2)

트랜잭션 문제 해결 - 트랜잭션 템플릿

앞서 트랜잭션 매니저를 통해 트랜잭션 추상화, 리소스 동기화를 얻는데에는 성공했다. 
그 덕에 서비스 코드는 이제 JDBC 기술에 의존하지 않으며 트랜잭션 동기화 매니저 덕분에 커넥션을 파라미터로 넘기지 않아도 된다.

하지만 트랜잭션을 사용하는 로직을 살펴보면 다음과 같은 패턴이 반복되는 것을 확인할 수 있다.

트랜잭션 사용 코드
//트랜잭션 시작
TransactionStatus status = transactionManager.getTransaction(new
DefaultTransactionDefinition());
try {
 //비즈니스 로직
 bizLogic(fromId, toId, money);
 transactionManager.commit(status); //성공시 커밋
} catch (Exception e) {
 transactionManager.rollback(status); //실패시 롤백
 throw new IllegalStateException(e);
}

트랜잭션을 시작하고, 비즈니스 로직을 실행하고, 성공하면 커밋하고, 예외가 발생해서 실패하면 롤백한다.
다른 서비스에서 트랜잭션을 시작하려면 try , catch , finally 를 포함한 성공시 커밋, 실패시 롤백 코드가 반복될 것이다.
이런 형태는 각각의 서비스에서 반복된다. 달라지는 부분은 비즈니스 로직 뿐이다.

템플릿 콜백 패턴을 활용하면 이런 반복 문제를 깔끔하게 해결할 수 있다.

본 강의에서 템플릿 콜백 패턴을 자세히 다루지 않는다.
지금은 "스프링이 TransactionTemplate 이라는 편리한 기능을 제공하는 구나" 정도만 이해해도 문제 없다.

트랜잭션 템플릿 템플릿 콜백 패턴을 적용하려면 템플릿을 제공하는 클래스를 작성해야 하는데, 스프링은 TransactionTemplate 라는 템플릿 클래스를 제공한다.

TransactionTemplate
public class TransactionTemplate {
 private PlatformTransactionManager transactionManager;
 public <T> T execute(TransactionCallback<T> action){..}
 void executeWithoutResult(Consumer<TransactionStatus> action){..}
}

execute() : 응답 값이 있을 때 사용한다.
executeWithoutResult() : 응답 값이 없을 때 사용한다.

이제 트랜잭션 템플릿을 적용해보자.

public class MemberServiceV3_2 {
     private final TransactionTemplate txTemplate;
     private final MemberRepositoryV3 memberRepository;
     
     public MemberServiceV3_2(PlatformTransactionManager transactionManager,
    MemberRepositoryV3 memberRepository) {
         this.txTemplate = new TransactionTemplate(transactionManager);
         this.memberRepository = memberRepository;
     }
     
     public void accountTransfer(String fromId, String toId, int money) throws
    SQLException {
         	txTemplate.executeWithoutResult((status) -> {
         try {
         	//비즈니스 로직
         	bizLogic(fromId, toId, money);
         } catch (SQLException e) {
         	throw new IllegalStateException(e);
         }
     	});
 }

TransactionTemplate 을 사용하려면 transactionManager 가 필요하다.
생성자에서 transactionManager 를 주입 받으면서 TransactionTemplate 을 생성했다.

위의 코드에서 템플릿 사용 로직만 떼어냈다.

트랜잭션 템플릿 사용 로직
txTemplate.executeWithoutResult((status) -> {
 try {
 //비즈니스 로직
 bizLogic(fromId, toId, money);
 } catch (SQLException e) {
 throw new IllegalStateException(e);
 }
});

트랜잭션 템플릿 덕분에 트랜잭션을 시작하고, 커밋하거나 롤백하는 코드가 모두 제거되었다.
트랜잭션 템플릿의 기본 동작은 다음과 같다.

비즈니스 로직이 정상 수행되면 커밋한다. 언체크 예외가 발생하면 롤백한다. 그 외의 경우 커밋한다. (체크 예외의 경우에는 커밋하는데, 이 부분은 뒤에서 설명한다.)

코드에서 예외를 처리하기 위해 try~catch 가 들어갔는데, bizLogic() 메서드를 호출하면 SQLException 체크 예외를 넘겨준다. 해당 람다에서 체크 예외를 밖으로 던질 수 없기 때문에 언체크 예외로 바꾸어 던지도록 예외를 전환했다.

---

트랜잭션 템플릿 덕분에, 트랜잭션을 사용할 때 반복하는 코드를 제거할 수 있었다.
하지만 이곳은 서비스 로직인데 비즈니스 로직 뿐만 아니라 트랜잭션을 처리하는 기술 로직이 함께 포함되어 있다.

애플리케이션을 구성하는 로직을 핵심 기능과 부가 기능으로 구분하자면 서비스 입장에서 비즈니스 로직은 핵심 기능이고, 트랜잭션은 부가 기능이다. 이렇게 비즈니스 로직과 트랜잭션을 처리하는 기술 로직이 한 곳에 있으면 두 관심사를 하나의 클래스에서 처리하게 된다. 결과적으로 코드를 유지보수하기 어려워진다.

서비스 로직은 가급적 핵심 비즈니스 로직만 있어야 한다. 하지만 트랜잭션 기술을 사용하려면 어쩔 수 없이 트랜잭션 코드가 나와야 한다. 어떻게 하면 이 문제를 해결할 수 있을까?

바로 스프링 AOP다.

---

트랜잭션 문제 해결 - 트랜잭션 AOP 이해

트랜잭션 템플릿 덕분에 트랜잭션을 처리하는 반복 코드는 해결할 수 있었다. 하지만 서비스 계층에 순수한 비즈니스 로직만 남긴다는 목표는 아직 달성하지 못했다.
이럴 때 스프링 AOP를 통해 프록시를 도입하면 문제를 깔끔하게 해결할 수 있다.

프록시를 도입하기 전에는 기존처럼 서비스의 로직에서 트랜잭션을 직접 시작한다.

프록시를 사용하면 트랜잭션을 처리하는 객체와 비즈니스 로직을 수행하는 서비스 계층을 명확하게 분리할 수 있다.

트랜잭션 프록시 코드 예시
public class TransactionProxy {
     private MemberService target;
     public void logic() {
     //트랜잭션 시작
     TransactionStatus status = transactionManager.getTransaction(..);
     
     try {
         //실제 대상 호출
         target.logic();
         transactionManager.commit(status); //성공시 커밋
     } catch (Exception e) {
         transactionManager.rollback(status); //실패시 롤백
         throw new IllegalStateException(e);
     }
     }
}

트랜잭션 프록시를 적용하면 서비스 계층의 코드는

public class Service {
 public void logic() {
 //트랜잭션 관련 코드 제거, 순수 비즈니스 로직만 남음
 bizLogic(fromId, toId, money);
 }
}

순수 비즈니스 로직만 남게된다.
트랜잭션 프록시가 트랜잭션 처리 로직을 모두 가져간다. 그리고 트랜잭션을 시작한 후에 실제 서비스를 대신 호출한다. 트랜잭션 프록시 덕분에 서비스 계층에는 순수한 비즈니즈 로직만 남길 수 있다.

스프링 부트를 사용하면 트랜잭션 AOP를 처리하기 위해 필요한 스프링 빈들도 자동으로 등록해준다. 개발자는 트랜잭션 처리가 필요한 곳에 @Transactional 애노테이션만 붙여주면 된다. 스프링의 트랜잭션 AOP는 이 애노테이션을 인식해서 트랜잭션 프록시를 적용해준다.

@RequiredArgsConstructor
public class MemberServiceV3_3 {
     private final MemberRepositoryV3 memberRepository;
     
     @Transactional
     public void accountTransfer(String fromId, String toId, int money) throws
    SQLException {
     	bizLogic(fromId, toId, money);
     }
     
     private void bizLogic(String fromId, String toId, int money) throws
    SQLException {
         Member fromMember = memberRepository.findById(fromId);
         Member toMember = memberRepository.findById(toId);
         memberRepository.update(fromId, fromMember.getMoney() - money);
         validation(toMember);
         memberRepository.update(toId, toMember.getMoney() + money);
     }
     
     private void validation(Member toMember) {
         if (toMember.getMemberId().equals("ex")) {
         	throw new IllegalStateException("이체중 예외 발생");
         }
     }
}

순수한 비즈니스 로직만 남기고, 트랜잭션 관련 코드는 모두 제거했다. 스프링이 제공하는 트랜잭션 AOP를 적용하기 위해 @Transactional 애노테이션을 추가했다.
@Transactional 애노테이션은 메서드에 붙여도 되고, 클래스에 붙여도 된다. 클래스에 붙이면 외부에서 호출 가능한 public 메서드가 AOP 적용 대상이 된다.

흐름 순서는 위의 그림과 같다.