[Spring DB] 트랜잭션 템플릿

이 글은 인프런 김영한님의 Spring 강의를 바탕으로 개인적인 정리를 위해 작성한 글입니다.

---

트랜잭션 템플릿(Transaction Template)은 스프링 프레임워크에서 제공하는 프로그래밍 방식(명시적)의 트랜잭션 관리 방법 중 하나이다. TransactionTemplate 클래스를 사용하면 선언적 트랜잭션 관리(@Transactional)와 달리, 코드 내에서 직접 트랜잭션의 시작, 커밋, 롤백을 제어할 수 있다. 이 방식은 트랜잭션 관리를 더 세밀하게 제어할 필요가 있는 복잡한 비즈니스 로직에서 유용하게 사용된다.

TransactionTemplate

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

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

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

 

템플릿 콜백 패턴

트랜잭션을 사용하는 로직을 살펴보면 아래와 같은 패턴이 반복된다.

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

이러한 반복을 조금이라도 줄이려면 템플릿 콜백 패턴을 활용하면 된다.

이 글에서 자세한 템플릿 콜백 패턴의 내용은 다루지 않는다.

 

서비스 계층

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 을 생성했다.

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

트랜잭션 템플릿의 기본 동작은 다음과 같다.

• 비즈니스 로직이 정상 수행되면 커밋한다.
• 언체크 예외가 발생하면 롤백한다. 그 외의 경우 커밋한다. (체크 예외의 경우에는 커밋)

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

 

execute()

TransactionTemplate의 execute 메소드를 사용하여 트랜잭션 범위 내에서 실행할 로직을 정의할 수 있다. TransactionCallback 인터페이스를 구현하는 익명 클래스나 람다 표현식을 사용하여 로직을 전달한다.

transactionTemplate.execute(new TransactionCallbackWithoutResult() {
    @Override
    protected void doInTransactionWithoutResult(TransactionStatus status) {
        // 여기에 트랜잭션을 사용하는 비즈니스 로직을 구현
        // 필요한 경우 status.setRollbackOnly();를 호출하여 롤백할 수 있다.
    }
});

 

또는 반환 값이 있는 작업의 경우 TransactionCallback<T> 제네릭 인터페이스를 사용할 수 있다.

String result = transactionTemplate.execute(status -> {
    // 비즈니스 로직 실행
    return "결과";
});

TransactionTemplate는 스프링 애플리케이션에서 복잡한 트랜잭션 관리가 필요한 경우 유용한 도구이다. 그러나 가능한 경우, 간결하고 선언적 트랜잭션 관리 방법인 @Transactional 어노테이션을 사용하는 것이 좋다.