스프링 트랜잭션

트랜잭션 도입 시 문제점

애플리케이션 구조에는 여러 가지가 있지만 가장 단순하고 많이 사용되는 방법은 아래와 같이 3개의 계층으로 나누는 것이다.

애플리케이션 3개의 계층

• 프레젠테이션 계층 : UI 관련 처리 담당, UI 관련 기술에 종속적
• 서비스 계층 : 비즈니스 로직 담당, 특정 기술에 종속적이지 않고 가급적 자바 순수 코드로 작성
• 데이터 접근 계층 : DB 접근 담당, DB 기술에 종속적

이렇게 3개의 계층으로 나누는 이유는 애플리케이션에서 가장 중요한, 핵심 비즈니스 로직을 담고 있는 서비스 계층이 특정 기술에 종속되지 않도록, 순수 자바 코드로 구성된 비즈니스 로직만 들어가도록 하기 위함이다. 이럼으로써 시간이 지나 기술이 변해도 서비스 계층은 변경 없이 유지될 수 있으며 순수 비즈니스 로직만 있기에 유지보수도 쉬워지고 테스트도 쉬워진다. 예를 들어 UI 관련 기술(HTTP, 서블릿, 웹 ··)이 바뀌면 프레젠테이션 계층을 수정하면 되며 데이터 접근 기술(JDBC, JPA, Mybatis ··)이 바뀌면 데이터 접근 계층을 수정하면 된다. 핵심 계층인 서비스 계층은 변경 없이 유지할 수 있다.

 

아래 서비스 계층의 코드를 보자. 트랜잭션을 작성했더니 JDBC 코드가 서비스 계층에 누수된다.

package hello.jdbc.service;

import hello.jdbc.domain.Member;
import hello.jdbc.repository.MemberRepositoryV2;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;

/**
 * 트랜잭션 - 파라미터 연동, 풀을 고려한 종료
 */
@Slf4j
@RequiredArgsConstructor
public class MemberServiceV2 {
    private final DataSource dataSource;
    private final MemberRepositoryV2 memberRepository;

    public void accountTransfer(String fromId, String toId, int money) throws SQLException {
        Connection conn = dataSource.getConnection();
        try {
            //트랜잭션 시작
            conn.setAutoCommit(false);

            //비즈니스 로직 시작
            bizLogic(conn, fromId, toId, money);

            //커밋 (트랜잭션 종료)
            conn.commit();
        } catch (Exception e) {
            //롤백 (트랜잭션 종료)
            conn.rollback();
            throw new IllegalStateException(e);
        } finally {
            //자원 해제
            if(conn != null) {
                try {
                    conn.setAutoCommit(true);
                    conn.close();
                } catch (Exception e) {
                    log.info("error", e);
                }
            }
        }
    }

    private void bizLogic(Connection conn, String fromId, String toId, int money) throws SQLException {
        Member fromMember = memberRepository.findById(conn, fromId);
        Member toMember = memberRepository.findById(conn, toId);
        memberRepository.update(conn, fromId, fromMember.getMoney() - money);
        if (toMember.getMemberId().equals("ex")) throw new IllegalArgumentException("이체중 예외 발생");
        memberRepository.update(toId, toMember.getMoney() + money);
    }
}

위 코드에는 아래의 문제점들이 있다.

 

트랜잭션 관련 문제

• 특정 데이터 접근 기술이 서비스 계층에 누수 → 향후 JDBC를 JPA로 바꾸면 서비스 계층을 수정해야하며 수정하는 과정에서 버그가 일어나기 쉬움, 비즈니스 로직과 JDBC 로직이 같이 있어 유지보수가 힘듬
• 트랜잭션 동기화 문제 → 트랜잭션 내에서 같은 커넥션을 유지하기 위해 트랜잭션의 커넥션과 데이터 접근 계층과의 커넥션 동기화(같은 커넥션을 맟추어 사용) 과정이 까다로움
• 코드의 반복 → 트랜잭션은 try, catch, finally 등 같은 형식의 코드가 반복됨

예외 누수

• JDBC의 예외인 SQLException 예외가 서비스 계층까지 전파, 향후 다른 데이터 접근 기술로 바뀌면 서비스 계층을 수정해야 함

위의 예외 누수를 제외한 트랜잭션 관련 문제를 차근차근 해결해보자.

트랜잭션 추상화

아래처럼 트랜잭션은 특정 데이터 접근 기술에 의존되어 작성되기에 특정 기술이 서비스 계층에 누수될 수밖에 없었다.

특정 데이터 접근 기술에 의존하는 트랜잭션

그래서 스프링은 트랜잭션을 추상화해 어떤 데이터 접근 기술을 사용해도 트랜잭션 코드의 변화가 없도록 만들었다. 서비스 계층은 PlatformTransactionManager를 의존해 트랜잭션 코드를 작성하고 데이터 접근 기술에 맞는 구현체를 주입하면 된다. 이로 인해 트랜잭션 작성 시 더 이상 특정 데이터 접근 기술에 의존하지 않아 서비스 계층에 특정 기술이 누수되는 문제를 해결할 수 있게 되었다.

트랜잭션 추상화

아래는 PlatformTransactionManager 인터페이스 코드이다.

package org.springframework.transaction;

import org.springframework.lang.Nullable;

public interface PlatformTransactionManager extends TransactionManager {
	TransactionStatus getTransaction(@Nullable TransactionDefinition definition)
			throws TransactionException;

	void commit(TransactionStatus status) throws TransactionException;

	void rollback(TransactionStatus status) throws TransactionException;
}

• getTransaction() : 트랜잭션을 시작한다.
• commit() : 트랜잭션을 커밋한다.
• rollback() : 트랜잭션을 롤백한다.

앞으로 PlatformTransactionManager 인터페이스와 구현체를 포함해서 트랜잭션 매니저로 줄여서 이야기하겠다.

트랜잭션 동기화

스프링이 제공하는 트랜잭션 매니저는 크게 2가지 역할을 한다.

• 트랜잭션 추상화
• 트랜잭션 동기화

트랜잭션 추상화는 위에서 언급했으므로 트랜잭션 동기화를 알아보자. 트랜잭션을 유지하려면 트랜잭션의 시작부터 끝가지 같은 커넥션(JDBC 커넥션)을 유지해야 한다. 이를 위해 트랜잭션의 커넥션과 데이터 접근 계층의 커넥션을 동기화(같은 커넥션을 맞추어 사용)하는 작업이 필요하다. 위 서비스 계층의 트랜잭션에서는 파라미터를 통해 커넥션을 데이터 접근 계층으로 전달하는 방법을 사용했다. 이 방법은 코드가 지저분해지는 것은 물론이고, 커넥션을 넘기는 메서드와 넘기지 않는 메서드를 중복해서 만들어야 하는 등 여러가지 단점들이 많다. 커넥션 동기화를 쉽게 하기 위해 스프링에서는 트랜잭션 동기화 매니저를 제공한다.

트랜잭션 매니저와 트랜잭션 동기화 매니저

트랜잭션 동기화 매니저는 쓰레드 로컬(thread local)을 사용해 트랜잭션의 커넥션과 데이터 접근 계층의 커넥션을 동기화한다. 쓰레드 로컬을 사용하기 때문에 멀티쓰레드 상황에 안전하게 커넥션을 동기화 할 수 있다.

MemberRepositoryV3

아래는 트랜잭션 동기화 매니저로부터 커넥션을 받아 사용하는 Repository 코드이다.

import hello.jdbc.domain.Member;
import lombok.extern.slf4j.Slf4j;
import org.springframework.jdbc.datasource.DataSourceUtils;
import org.springframework.jdbc.support.JdbcUtils;

import javax.sql.DataSource;
import java.sql.*;
import java.util.NoSuchElementException;

/**
 * 트랜잭션 - 트랜잭션 매니저
 * DataSourceUtils.getConnection()
 * DataSourceUtils.releaseConnection()
 */
@Slf4j
public class MemberRepositoryV3 {

    private final DataSource dataSource;

    public MemberRepositoryV3(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    public Member save(Member member) throws SQLException {
        String sql = "insert into member(member_id, money) values (?,?)";
        Connection conn = null;
        PreparedStatement pstmt = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql);
            pstmt.setString(1, member.getMemberId());
            pstmt.setInt(2, member.getMoney());
            pstmt.executeUpdate();
            return member;
        } catch (SQLException e) {
            log.error("db error", e);
            throw e;
        } finally {
            close(conn, pstmt, null);
        }
    }

    public Member findById(String memberId) throws SQLException {
        String sql = "select * from member where member_id = ?";
        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql);
            pstmt.setString(1, memberId);
            rs = pstmt.executeQuery();
            if(rs.next()) {
                Member member = new Member();
                member.setMemberId(rs.getString("member_id"));
                member.setMoney(rs.getInt("money"));
                return member;
            } else {
              throw new NoSuchElementException("member not found memberId=" + memberId);
            }
        } catch (SQLException e) {
            log.error("db error", e);
            throw e;
        } finally {
            close(conn, pstmt, rs);
        }
    }

    public void update(String memberId, int money) throws SQLException {
        String sql = "update member set money = ? where member_id = ?";
        Connection conn = null;
        PreparedStatement pstmt = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql);
            pstmt.setInt(1, money);
            pstmt.setString(2, memberId);
            pstmt.executeUpdate();
        } catch (SQLException e) {
            log.error("db error", e);
            throw e;
        } finally {
            close(conn, pstmt, null);
        }
    }

    public void delete(String memberId) throws SQLException {
        String sql = "delete from member where member_id=?";
        Connection conn = null;
        PreparedStatement pstmt = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql);
            pstmt.setString(1, memberId);
            pstmt.executeUpdate();
        } catch (SQLException e) {
            log.error("db error", e);
            throw e;
        } finally {
            close(conn, pstmt, null);
        }
    }

    private Connection getConnection() throws SQLException {
        //주의! 트랜잭션 동기화를 사용하려면 DataSourceUtils를 사용해야한다.
        Connection conn = DataSourceUtils.getConnection(dataSource);
        log.info("get connection = {}, class = {}", conn, conn.getClass());
        return conn;
    }

    private void close(Connection conn, Statement stmt, ResultSet rs)  {
        //주의! 트랜잭션 동기화를 사용하려면 DataSourceUtils를 사용해야한다.
        JdbcUtils.closeResultSet(rs);
        JdbcUtils.closeStatement(stmt);
        DataSourceUtils.releaseConnection(conn, dataSource);
    }
}

• 커넥션을 파라미터로 받는 부분이 모두 제거 되었다.
• DatasourceUtils.getConnection()는 트랜잭션 동기화 매니저가 관리하는 커넥션이 있으면 해당 커넥션을 반환하고, 없으면 새로운 커넥션을 생성해서 반환한다.
• DatasourceUtils.releaseConnection()는 트랜잭션 동기화 매니저가 관리하는 커넥션이라면 닫지 않고 유지하며, 관리하지 않는 커넥션이라면 닫아준다.

MemberServiceV3_1

아래는 트랜잭션 매니저를 적용한 서비스 코드이다.

package hello.jdbc.service;

import hello.jdbc.domain.Member;
import hello.jdbc.repository.MemberRepositoryV3;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.DefaultTransactionDefinition;

import java.sql.SQLException;

/**
 * 트랜잭션 - 트랜잭션 매니저
 */
@Slf4j
@RequiredArgsConstructor
public class MemberServiceV3_1 {
    private final PlatformTransactionManager transactionManager;
    private final MemberRepositoryV3 memberRepository;

    public void accountTransfer(String fromId, String toId, int money) throws SQLException {
        //트랜잭션 시작
        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 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);
        if (toMember.getMemberId().equals("ex")) throw new IllegalArgumentException("이체중 예외 발생");
        memberRepository.update(toId, toMember.getMoney() + money);
    }
}

• transactionManager.getTransaction() : 트랜잭션을 시작한다.
• new DefaultTransactionDefinition() : 트랜잭션 관련 옵션을 지정할 수 있다.
• transactionManager.commit(status) : 커밋
• transactionManager.rollback(status) : 롤백

아래는 위의 서비스 객체(MemberServiceV3_1)의 테스트 코드이다.

package hello.jdbc.service;

import hello.jdbc.domain.Member;
import hello.jdbc.repository.MemberRepositoryV3;
import org.assertj.core.api.Assertions;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.jdbc.datasource.DriverManagerDataSource;
import org.springframework.transaction.PlatformTransactionManager;

import java.sql.SQLException;

import static hello.jdbc.connection.ConnectionConst.*;

/**
 * 트랜잭션 - 트랜잭션 매니저
 */
class MemberServiceV3_1Test {
    public static final String MEMBER_A = "memberA";
    public static final String MEMBER_B = "memberB";
    public static final String MEMBER_EX = "ex";

    private MemberRepositoryV3 memberRepository;
    private MemberServiceV3_1 memberService;

    @BeforeEach
    void before() {
        DriverManagerDataSource dataSource = new DriverManagerDataSource(URL, USERNAME, PASSWORD);
        PlatformTransactionManager transactionManager = new DataSourceTransactionManager(dataSource);

        memberRepository = new MemberRepositoryV3(dataSource);
        memberService = new MemberServiceV3_1(transactionManager, memberRepository);
    }

    @AfterEach
    void after() throws SQLException {
        memberRepository.delete(MEMBER_A);
        memberRepository.delete(MEMBER_B);
        memberRepository.delete(MEMBER_EX);
    }

    @Test
    @DisplayName("정상 이체")
    void accountTransfer() throws SQLException {
        //given
        Member memberA = new Member(MEMBER_A, 10000);
        Member memberB = new Member(MEMBER_B, 10000);
        memberRepository.save(memberA);
        memberRepository.save(memberB);

        //when
        memberService.accountTransfer(memberA.getMemberId(), memberB.getMemberId(), 2000);

        //then
        memberA = memberRepository.findById(memberA.getMemberId());
        memberB = memberRepository.findById(memberB.getMemberId());
        Assertions.assertThat(memberA.getMoney()).isEqualTo(8000);
        Assertions.assertThat(memberB.getMoney()).isEqualTo(12000);
    }

    @Test
    @DisplayName("이체 중 예외 발생")
    void accountTransferEx() throws SQLException {
        //given
        Member memberA = new Member(MEMBER_A, 10000);
        Member memberEx  = new Member(MEMBER_EX, 10000);
        memberRepository.save(memberA);
        memberRepository.save(memberEx);

        //when
        Member finalMemberA = memberA;
        Member finalMemberEx = memberEx;
        Assertions.assertThatThrownBy(()->
                memberService.accountTransfer(finalMemberA.getMemberId(), finalMemberEx.getMemberId(), 2000))
                .isInstanceOf(IllegalStateException.class);

        //then
        memberA = memberRepository.findById(memberA.getMemberId());
        memberEx = memberRepository.findById(memberEx.getMemberId());
        Assertions.assertThat(memberA.getMoney()).isEqualTo(10000);
        Assertions.assertThat(memberEx.getMoney()).isEqualTo(10000);
    }
}

트랜잭션 매니저 동작 흐름

그림으로 트랜잭션 매니저 동작 흐름을 살펴보자. 여기서 트랜잭션 매니저는 DataSourceTransactionManager라 가정하고 DataSourceTransactionManager의 동작 흐름을 살펴본다. 다른 트랜잭션 매니저들도 해당 기술에 맞도록 변형되어서 동작한다.

1. 서비스 계층에서 transactionManager.getTransaction() 을 호출해서 트랜잭션을 시작한다.

2. 트랜잭션을 시작하려면 먼저 데이터베이스 커넥션이 필요하다. 트랜잭션 매니저는 내부에서 데이터소스를 사용해서 커넥션을 생성한다.

3. 커넥션을 수동 커밋 모드로 변경해서 실제 데이터베이스 트랜잭션을 시작한다.

4. 커넥션을 트랜잭션 동기화 매니저에 보관한다.

5. 트랜잭션 동기화 매니저는 쓰레드 로컬에 커넥션을 보관한다. 따라서 멀티 쓰레드 환경에 안전하게 커넥션을 보관할 수 있다.

6. 서비스는 비즈니스 로직을 실행하면서 리포지토리의 메서드들을 호출한다.

7. 리포지토리 메서드들은 트랜잭션이 시작된 커넥션이 필요하다. 리포지토리는 DataSourceUtils.getConnection() 을 사용해서 트랜잭션 동기화 매니저에 보관된 커넥션을 꺼내서 사용한다. 이 과정을 통해서 자연스럽게 트랜잭션을 시작한 커넥션과 같은 커넥션을 사용하고, 트랜잭션도 유지된다.

8. 획득한 커넥션을 사용해서 SQL을 데이터베이스에 전달해서 실행한다.

9. 비즈니스 로직이 끝나고 트랜잭션을 종료한다. 트랜잭션은 커밋하거나 롤백하면 종료된다.

10. 트랜잭션을 종료하려면 동기화된 커넥션이 필요하다. 트랜잭션 동기화 매니저를 통해 동기화된 커넥션을 획득한다.

11. 획득한 커넥션을 통해 데이터베이스에 트랜잭션을 커밋하거나 롤백한다.

12. 전체 리소스를 정리한다.

• 트랜잭션 동기화 매니저를 정리한다. 쓰레드 로컬은 사용후 꼭 정리해야 한다.
• con.setAutoCommit(true)로 되돌린다. 커넥션 풀을 고려해야 한다.
• con.close()를 호출해셔 커넥션을 종료한다. 커넥션 풀을 사용하는 경우 con.close()를 호출하면 커넥션 풀에 반환된다.

트랜잭션 템플릿

위에서 트랜잭션 관련 문제점인 특정 데이터 접근 기술이 서비스 계층에 누수되는 문제와 트랜잭션 동기화 문제를 해결했다. 이제 트랜잭션 관련 문제점이 하나 남았다. 트랜잭션 코드를 보면 try, catch, finally 등 같은 패턴이 반복됨을 알 수 있다. 트랜잭션이 하나라면 상관없지만 수십에서 수백개의 트랜잭션을 작성하게 되면 같은 패턴의 코드가 반복된다. 이럴때 템플릿 콜백 패턴을 사용하면 반복되는 코드를 제거할 수 있다.

 

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

 

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

MemberServiceV3_1 코드에 TransactionTemplate을 적용해서 반복되는 코드를 제거해보자.

package hello.jdbc.service;

import hello.jdbc.domain.Member;
import hello.jdbc.repository.MemberRepositoryV3;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.DefaultTransactionDefinition;
import org.springframework.transaction.support.TransactionTemplate;

import java.sql.SQLException;

/**
 * 트랜잭션 - 트랜잭션 템플릿
 */
@Slf4j
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);
            }
        });
    }

    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);
        if (toMember.getMemberId().equals("ex")) throw new IllegalStateException("이체중 예외 발생");
        memberRepository.update(toId, toMember.getMoney() + money);
    }
}

템플릿 콜백 패턴을 적용하니 비즈니스 로직만 남고 반복되는 트랜잭션 코드는 TransactionTemplate 객체가 가져갔다. TransactionTemplate 코드를 보면 아래와 같이 동작한다.

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

테스트 코드는 MemberServiceV3_1의 테스트 코드와 같다.

트랜잭션 AOP

트랜잭션 템플릿을 사용해서 반복되는 트랜잭션 코드들을 제거할 수 있었지만 서비스 계층에 순수한 비즈니스 로직만 남긴다는 목표는 달성하지 못했다. 이때 스프링 AOP를 통해 프록시를 도입하면 트랜잭션 코드를 아예 서비스 계층에서 분리시킬 수 있다.

프록시 도입

직접 트랜잭션 AOP 처리를 할 수도 있지만 스프링은 이미 트랜잭션 AOP 처리를 위해 필요한 모든 기능을 제공한다. 개발자는 트랜잭션이 필요한 곳에 @Transactional만 붙여주면 스프링이 알아서 트랜잭션 AOP 처리를 한다. @Transactional로 트랜잭션 AOP를 적용하면 편리하지만 스프링 컨테이너와 스프링 AOP에 의존하기에 테스트 시 관련 빈 객체가 로드된 스프링 컨테이너가 필요하다.

 

참고로 스프링 AOP를 적용하려면 어드바이저, 포인트컷, 어드바이스가 필요하다. 스프링은 트랜잭션 AOP 처리를 위해 다음 클래스를 제공한다. 스프링 부트를 사용하면 해당 빈들은 스프링 컨테이너에 자동으로 등록된다.

• 어드바이저 : BeanFactoryTransactionAttributeSourceAdvisor
• 포인트컷 : TransactionAttributeSourcePointcut
• 어드바이스: TransactionInterceptor

트랜잭션 AOP을 적용하니 서비스 계층에 트랜잭션 관련 코드가 사라졌다.

import hello.jdbc.domain.Member;
import hello.jdbc.repository.MemberRepositoryV3;
import lombok.extern.slf4j.Slf4j;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.support.TransactionTemplate;

import java.sql.SQLException;

/**
 * 트랜잭션 - 트랜잭션 AOP
 */
@Slf4j
public class MemberServiceV3_3 {
    private final MemberRepositoryV3 memberRepository;

    public MemberServiceV3_3(MemberRepositoryV3 memberRepository) {
        this.memberRepository = 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);
        if (toMember.getMemberId().equals("ex")) throw new IllegalStateException("이체중 예외 발생");
        memberRepository.update(toId, toMember.getMoney() + money);
    }
}

아래는 트랜잭션 AOP가 사용된 전체 흐름이다.

스프링 부트 자동 리소스 등록

스프링 부트는 DataSouce를 자동으로 빈으로 등록한다. 스프링 부트는 application.properties 속성을 사용해 DataSource를 생성한 뒤 빈으로 등록한다. 기본으로 생성하는 DataSource는 HikariDataSource이며 spring.datasource.url가 없으면 내장 DB(메모리 DB)를 생성하려고 시도한다.

application.properties에 Datasource 구성 정보 입력

또한 스프링 부트는 적절한 트랜잭션 매니저(PlatformTransactionManager)을 자동으로 빈으로 등록한다. 어떤 트랜잭션 매니저를 생성해 빈으로 등록할지는 현재 등록된 라이브러리를 보고 판단한다. JDBC를 기술을 사용하면 DataSourceTransactionManager를 빈으로 등록하고, JPA를 사용하면 JpaTransactionManager를 빈으로 등록한다. 둘다 사용하는 경우 JpaTransactionManager를 등록한다.

 

아래는 위 @Transactional이 적용된 MemberServiceV3_3의 테스트 코드이다. 스프링 부트가 데이터 소스와 트랜잭션 매니저를 자동으로 빈으로 등록하기에 수동으로 빈으로 등록하지 않았다.

package hello.jdbc.service;

import hello.jdbc.domain.Member;
import hello.jdbc.repository.MemberRepositoryV3;
import lombok.extern.slf4j.Slf4j;
import org.assertj.core.api.Assertions;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.springframework.aop.support.AopUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.context.TestConfiguration;
import org.springframework.context.annotation.Bean;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.jdbc.datasource.DriverManagerDataSource;
import org.springframework.transaction.PlatformTransactionManager;

import javax.sql.DataSource;
import java.sql.SQLException;

import static hello.jdbc.connection.ConnectionConst.*;

/**
 * 트랜잭션 - 트랜잭션 Datasource, transactionManager 자동등록
 */

@Slf4j
@SpringBootTest
class MemberServiceV3_4Test {
    public static final String MEMBER_A = "memberA";
    public static final String MEMBER_B = "memberB";
    public static final String MEMBER_EX = "ex";

    @Autowired
    private MemberRepositoryV3 memberRepository;
    @Autowired
    private MemberServiceV3_3 memberService;

    @TestConfiguration
    static class TestConfig {

        private final DataSource dataSource;

        public TestConfig(DataSource dataSource) {
            this.dataSource = dataSource;
        }

        @Bean
        MemberRepositoryV3 memberRepositoryV3() { return new MemberRepositoryV3(dataSource); }

        @Bean
        MemberServiceV3_3 memberServiceV3_3() {
            return new MemberServiceV3_3(memberRepositoryV3());
        }
    }

    @AfterEach
    void after() throws SQLException {
        memberRepository.delete(MEMBER_A);
        memberRepository.delete(MEMBER_B);
        memberRepository.delete(MEMBER_EX);
    }

    @Test
    void AopCheck() {
        log.info("memberService class={}", memberService.getClass());
        log.info("memberRepository class={}", memberRepository.getClass());
        Assertions.assertThat(AopUtils.isAopProxy(memberService)).isTrue();
        Assertions.assertThat(AopUtils.isAopProxy(memberRepository)).isFalse();
    }

    @Test
    @DisplayName("정상 이체")
    void accountTransfer() throws SQLException {
        //given
        Member memberA = new Member(MEMBER_A, 10000);
        Member memberB = new Member(MEMBER_B, 10000);
        memberRepository.save(memberA);
        memberRepository.save(memberB);

        //when
        memberService.accountTransfer(memberA.getMemberId(), memberB.getMemberId(), 2000);

        //then
        memberA = memberRepository.findById(memberA.getMemberId());
        memberB = memberRepository.findById(memberB.getMemberId());
        Assertions.assertThat(memberA.getMoney()).isEqualTo(8000);
        Assertions.assertThat(memberB.getMoney()).isEqualTo(12000);
    }

    @Test
    @DisplayName("이체 중 예외 발생")
    void accountTransferEx() throws SQLException {
        //given
        Member memberA = new Member(MEMBER_A, 10000);
        Member memberEx  = new Member(MEMBER_EX, 10000);
        memberRepository.save(memberA);
        memberRepository.save(memberEx);

        //when
        Member finalMemberA = memberA;
        Member finalMemberEx = memberEx;
        Assertions.assertThatThrownBy(()->
                memberService.accountTransfer(finalMemberA.getMemberId(), finalMemberEx.getMemberId(), 2000))
                .isInstanceOf(IllegalStateException.class);

        //then
        memberA = memberRepository.findById(memberA.getMemberId());
        memberEx = memberRepository.findById(memberEx.getMemberId());
        Assertions.assertThat(memberA.getMoney()).isEqualTo(10000);
        Assertions.assertThat(memberEx.getMoney()).isEqualTo(10000);
    }
}