[스프링 DB 1편] #3. 트랜잭션 이해 (2)

[YuDeokRin]

1️⃣ 1️⃣트랜잭션 - 적용2

• 이번에는 DB 트랜잭션을 사용해서 앞서 발생한 문제점을 해결해보자.

• 애플리케이션에서 트랜잭션을 어떤 계층에 걸어야 할까?

• 비즈니스 로직과 트랜잭션

  • 트랜잭션은 비즈니스 로직이 있는 서비스 계층에서 시작해야 한다. 비즈니스 로직이 잘못되면 해당 비즈니스 로직으로 인해 문제가 되는 부분을 함께 롤백해야 하기 때문이다.
  • 그런데 트랜잭션을 시작하려면 커넥션이 필요하다. 결국 서비스 계층에서 커넥션을 만들고, 트랜잭션 커밋 이후에 커넥션을 종료해야 한다.
  • 애플리케이션에서 DB 트랜

• MemberRepositoryV2 - findById
  • findById(Connection con, String memberId) → Connection con 추가

public Member findById(Connection con, String memberId) throws SQLException {
        String sql = "select * from member where member_id = ?";

        PreparedStatement pstmt = null;
        ResultSet rs = null;

        try {
            pstmt = con.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 {
            //connection은 여기서 닫지않는다 이유는 같은 세션을 사용하기 때문에
            JdbcUtils.closeResultSet(rs);
            JdbcUtils.closeStatement(pstmt);
        }

    }

• MemberRepositoryV2 - update
  • pdate(Connection con, String memberId, int money) → Connection con 추가

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

        PreparedStatement pstmt = null;

        try {
            pstmt = con.prepareStatement(sql);
            pstmt.setInt(1, money);
            pstmt.setString(2, memberId);
            pstmt.executeUpdate();

        } catch (SQLException e) {
            log.error("db error", e);
            throw e;
        } finally {
            JdbcUtils.closeStatement(pstmt);
        }
 }

• MemberServiceV2
  • Connection con = dateSource.getConnection();
    • 트랜잭션을 시작하려면 커넥션이 필요하다.
  • con.setAutoCommit(false);
    • 트랜잭션 시작
    • 트랜잭션을 시작하려면 자동 커밋 모드를 off해야한다.
  • bizLogic(con, fromId, toId, money);
    • 트랜잭션이 시작된 커넥션을 전달하면서 비즈니스 로직을 수행
    • 분리한 이유 → 트랜잭션 관리 로직, 비즈니스 로직 구분을 위해서
  • release(on);
    • finally{…}를 사용해서 커넥션을 모두 사용하고 나면 완전하게 종료한다.
    • 커넥션 풀을 사용하면 con.close()를 호출 했을 때 커넥션이 종료되는 것이 아니라 풀에 반납한다.

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

    private final DataSource dateSource; 
    private final MemberRepositoryV2 memberRepository;

    public void accountTransfer(String fromId, String toId, int money) throws SQLException {
        Connection con = dateSource.getConnection();

        try {
            con.setAutoCommit(false);
            //비즈니스 로직
            bizLogic(fromId, toId, money, con);
            con.commit();
        } catch (SQLException e) {
            con.rollback();
            throw new IllegalStateException(e);
        } finally {
            release(con);
        }
        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 bizLogic(String fromId, String toId, int money, Connection con) throws SQLException {
        Member fromMember = memberRepository.findById(con, fromId);
        Member toMember = memberRepository.findById(con, toId);

        memberRepository.update(con, fromId, fromMember.getMoney() - money);
        validation(toMember);
        memberRepository.update(con, toId, toMember.getMoney() + money);
    }

    private void validation(Member toMember) {
        if (toMember.getMemberId().equals("ex")) {
            throw new IllegalArgumentException("이체중 예외 발생");
        }
    }
    
    private void release(Connection con) {
        if (con != null) {
            try {
                con.setAutoCommit(true); // 커넥션 풀 고려
                con.close();
            } catch (SQLException e) {
                log.info("error", e);
            }
        }
    }
}

• MemberRepositoryV2Test
  • 정상 이체 테스트
    • 기존 로직과 같이 된다.
  • 이체중 예외 발생 테스트
    • memberA - 10000원
    • memberEx - 10000원
  • 계좌이체 로직을 실행
    • memberService.accountTransfer() 실행
    • 커넥션 생성, 트랜잭션 시작
    • memberA → memberEx로 2000원 계좌이체
      • memberA의 금액이 2000원 감소한다.
      • memberEx회원의 ID는 ex이므로 중간에 예외가 발생
    • 예외가 발생 → 트랜잭션 롤백 수행
  • 계좌이체 실패 → 롤백때문에 memberA의 돈이 다시 10000원으로 복구

/**
 * 기본동작, 트랜잭션이 없어서 문제 발생
 */
@Slf4j
class MemberRepositoryV2Test {

    MemberRepositoryV2 memberRepository;
    MemberServiceV2 memberService;

    @BeforeEach
    void before() throws Exception {
        DriverManagerDataSource dataSource = new DriverManagerDataSource(URL, USERNAME, PASSWORD);
        memberRepository = new MemberRepositoryV2(dataSource);
        memberService = new MemberServiceV2(dataSource, memberRepository);
    }

    @AfterEach
    void after() throws SQLException {
        memberRepository.delete("memberA");
        memberRepository.delete("memberB");
        memberRepository.delete("ex");
    }

    @Test
    @DisplayName("정상 이체")
    void accountTransfer() throws SQLException, InterruptedException {
        //given
        Member memberA = new Member("MemberA", 10000);
        Member memberB = new Member("MemberB", 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());
        Assertions.assertThat(findMemberA).isEqualTo(8000);
        Assertions.assertThat(findMemberB).isEqualTo(10000);

    }

    @Test
    @DisplayName("이체중 예외 발생")
    void accountTransferEx() throws SQLException {
        //given
        Member memberA = new Member("memberA", 10000);
        Member memberEx = new 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 findMemberEx = memberRepository.findById(memberEx.getMemberId());//memberA의 돈이 롤백 되어야함

        //memberA의 돈이 롤백 되어야함
        assertThat(findMemberA.getMoney()).isEqualTo(10000);
        assertThat(findMemberEx.getMoney()).isEqualTo(10000);
    }
}

🔗 출처

• 김영한 - 스프링 DB 1편 - 데이터 접근 핵심 원리