트랜잭션 문제 해결 - 트랜잭션 매니저2 & 트랜잭션 탬플릿

트랜잭션 매니저의 전체 동작 흐름을 자세히 보자.

 

트랜잭션 매니저 1 - 트랜잭션 시작

• 클라이언트의 요청으로 서비스 로직을 실행

 

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

 

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

 

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

 

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

 

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

트랜잭션 매니저2 - 로직 실행

• 서비스는 비즈니스 로직을 실행하면서 리포지토리의 메서드들 호출. 이 때 커넥션을 파라미터로 전달하지 않음
• 리포지토리 메서드들은 트랜잭션이 시작된 커넥션이 필요하다. 리포지토리는
  DataSourceUtils.getConnection() 을 사용해서 트랜잭션 동기화 매니저에 보관된 커넥션을 꺼내서
  사용한다. 이 과정을 통해서 자연스럽게 같은 커넥션을 사용하고, 트랜잭션도 유지된다.

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

 

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

 

 

 

 

 

 

 

 

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

 

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

 

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

 

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

 

• 트랜잭션 동기화 매니저를 정리한다. 쓰레드 로컬은 사용후 꼭 정리해야 한다.

 

• con.setAutoCommit(true) 로 되돌린다. 커넥션 풀을 고려해야 한다.

 

• con.close() 를 호출해셔 커넥션을 종료한다. 커넥션 풀을 사용하는 경우 con.close() 를
  호출하면 커넥션 풀에 반환된다.

 

# 정리

• 트랜잭션 추상화 덕분에 서비스 코드는 이제 JDBC 기술에 의존하지 않는다.

 

• 이후 JDBC에서 JPA로 변경해도 서비스 코드를 그대로 유지할 수 있다.

 

• 기술 변경시 의존관계 주입만 DataSourceTransactionManager 에서 JpaTransactionManager
  로 변경해주면 된다.

 

• java.sql.SQLException 이 아직 남아있지만 이 부분은 뒤에 예외 문제에서 해결하자.
  트랜잭션 동기화 매니저 덕분에 커넥션을 파라미터로 넘기지 않아도 된다.

DataSourceTransactionManager의 동작 방식 위주로 설명 => 다른 트랜잭션 매니저는 해당 기술에 맞도록 변형되어서 동작.

 

 

 

 

트랜잭션 템플릿

 

MemberServiceV3_1에서 성공 실패할 때 진행되는 구문 try catch문이 반복된다. 해결 방법을 알아보자.

 

 

MemberServiceV3_2를 만들어 주고

private final PlatformTransactionManager transactionManager;
// MemberRepositoryV3를 꼭 사용해주어야 한다.
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);
    }
}

 

 

 

 

 

아래 패턴으로 TransactionManager을 주입받고 내부에서는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 PlatformTransactionManager transactionManager;
    private final TransactionTemplate txTemplate;
    private final MemberRepositoryV3 memberRepository;

    // 주입받기 PlatformTransactionManager
    public MemberServiceV3_2(PlatformTransactionManager transactionManager, MemberRepositoryV3 memberRepository) {
        this.txTemplate = new TransactionTemplate(transactionManager);
        this.memberRepository = memberRepository;
    }

 

• TransactionTemplate를 사용하려면 transactionManager가 필요하고, 생성자에서 transactionManager를 주입 받으면서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 PlatformTransactionManager transactionManager;
    private final TransactionTemplate txTemplate;
    private final MemberRepositoryV3 memberRepository;

    // 주입받기 PlatformTransactionManager
    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 {
        // lambda로 로직 넣기
        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);
        validation(toMember);
        memberRepository.update(toId, toMember.getMoney() + money);
    }
    private void validation(Member toMember) {
        if (toMember.getMemberId().equals("ex")) {
            throw new IllegalStateException("이체중 예외 발생");
        }
    }
}

 

 

 

 

트랜잭션 템플릿 사용 로직

public void accountTransfer(String fromId, String toId, int money) throws SQLException {
    // lambda로 로직 넣기
    txTemplate.executeWithoutResult((status) -> {
        try {
            bizLogic(fromId, toId, money);
        } catch (SQLException e) {
            throw new IllegalStateException(e);
        }
    });
}

• 트랜잭션 템플릿 덗분에 트랜잭션을 시작, 커밋하거나 롤백하는 코드가 모두 제거

 

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

 

• 비즈니스 로직이 정상 수행되면 => 커밋

 

• 언체크 예외가 발생하면 롤백한다. 그 외의 경우 커밋한다. (체크 예외의 경우에는 커밋하는데, 이
  부분은 뒤에서 설명한다.)

 

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

 

 

 

 

 

트랜잭션 템플릿이 돌아가도록 테스트 코드를 작성하면

package hello.jdbc.service;

import hello.jdbc.domain.Member;
import hello.jdbc.repository.MemberRepositoryV3;
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.*;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;

/**
 * 기본 동작, 트랜잭션 템플릿
 */

class MemberServiceV3_2Test {

    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_2 memberService;

    @BeforeEach
    void before() {
        DriverManagerDataSource dataSource = new DriverManagerDataSource(URL, USERNAME, PASSWORD);
        memberRepository = new MemberRepositoryV3(dataSource);
        // 파라미터로 dataSource를 넘겨주어야 한다.
        PlatformTransactionManager transactionManager = new DataSourceTransactionManager(dataSource);
        memberService = new MemberServiceV3_2(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 이래 된다는 거 검증해라.
        Member findMemberA = memberRepository.findById(memberA.getMemberId());
        Member findMemberB = memberRepository.findById(memberB.getMemberId());
        assertThat(findMemberA.getMoney()).isEqualTo(8000);
        assertThat(findMemberB.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 이거 수행하면

        // 예외가 터져줘야 한다.
        assertThatThrownBy(() -> memberService.accountTransfer(memberA.getMemberId(), memberEx.getMemberId(), 2000))
                .isInstanceOf(IllegalStateException.class);

        // then 이래 된다는 거 검증해라.
        Member findMemberA = memberRepository.findById(memberA.getMemberId());
        Member findMemberB = memberRepository.findById(memberEx.getMemberId());
        assertThat(findMemberA.getMoney()).isEqualTo(10000);
        assertThat(findMemberB.getMoney()).isEqualTo(10000);
    }
}

 

MemberServiceV3_1에서는 아래와 같은 로직

private final PlatformTransactionManager transactionManager;
// MemberRepositoryV3를 꼭 사용해주어야 한다.
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);
    }
}

 

 

MemberServiceV3_2에서는 더 간단한 로직으로 작성이 되었다.

public void accountTransfer(String fromId, String toId, int money) throws SQLException {
    // lambda로 로직 넣기
    txTemplate.executeWithoutResult((status) -> {
        try {
            bizLogic(fromId, toId, money);
        } catch (SQLException e) {
            throw new IllegalStateException(e);
        }
    });
}

• 트랜잭션 템플릿 덕분에, 트랜잭션을 사용할 때 반복하는 코드를 제거할 수 있었다.

 

• 하지만 이곳은 서비스 로직인데 비즈니스 로직 뿐만 아니라 트랜잭션을 처리하는 기술 로직이 함께
  포함되어 있다.

 

• 애플리케이션을 구성하는 로직을 핵심 기능과 부가 기능으로 구분하자면 서비스 입장에서 비즈니스 로직은
  핵심 기능이고, 트랜잭션은 부가 기능이다.

 

• 이렇게 비즈니스 로직과 트랜잭션을 처리하는 기술 로직이 한 곳에 있으면 두 관심사를 하나의 클래스에서
  처리하게 된다. 결과적으로 코드를 유지보수하기 어려워진다.

 

• 서비스 로직은 가급적 핵심 비즈니스 로직만 있어야 한다. 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

<출처 김영한: 스프링 DB 1편 - 데이터 접근 핵심 원리>

https://www.inflearn.com/course/%EC%8A%A4%ED%94%84%EB%A7%81-db-1/dashboard

스프링 DB 1편 - 데이터 접근 핵심 원리 - 인프런 | 강의