[Spring DB] 트랜잭션 템플릿

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

트랜잭션 사용 코드

// 트랜잭션 시작
val status: TransactionStatus = transactionManager.getTransaction(DefaultTransactionDefinition())

try {
	bizLogic(fromId, toId, money)
    transactionManager.commit(status) // 성공 시 커밋
} catch (e: Exception){
      transactionManager.rollback(status) // 실패 시 롤백
      throw IllegalStateException(e)
    }

• 트랜잭션을 시작하려면 try , catch , finally 를 포함한 성공시 커밋, 실패시 롤백 형태로 구성이 되는데, 이런 형태는 각각의 서비스에서 반복되고 비즈니스 로직 부분만 달라진다.
• 이럴 때 템플릿 콜백 패턴을 활용하면 이런 반복 문제를 깔끔하게 해결할 수 있다.

트랜잭션 템플릿

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

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

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

트랜잭션 템플릿을 사용해서 반복하는 부분을 제거해보자

MemberServiceV3_2

package hello.jbdc.service

import hello.jbdc.Log
import hello.jbdc.domain.Member
import hello.jbdc.repository.MemberRepositoryV3
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.Connection
import java.sql.SQLException
import kotlin.jvm.Throws

/*
  트랜잭션 - 트랜잭션 템플릿
 */

class MemberServiceV3_2(
  private val transactionManager: PlatformTransactionManager,
  private val memberRepository: MemberRepositoryV3

) :Log{
  val txTemplate: TransactionTemplate

  init{
    txTemplate = TransactionTemplate(transactionManager)
  }

  @Throws(SQLException::class)
  fun accountTransfer(fromId: String, toId: String, money: Int){
    // 트랜잭션 시작
    txTemplate.executeWithoutResult {
      try{
        bizLogic(fromId, toId, money)
      } catch (e: SQLException){
        throw IllegalStateException(e) 
      }
    }
  }

  @Throws(SQLException::class)
  private fun bizLogic(fromId: String, toId: String, money: Int){
    val fromMember = memberRepository.findById(fromId)
    val toMember = memberRepository.findById(toId)

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

  private fun validation(toMember: Member){
    if (toMember.memberId.equals("ex")){
      throw IllegalStateException("이체 중 예외 발생")
    }
  }
}

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

트랜잭션 템플릿 사용 로직

 txTemplate.executeWithoutResult {
      try{
        bizLogic(fromId, toId, money)
      } catch (e: SQLException){
        throw IllegalStateException(e) // 언체크 예외
      }
    }

• 트랜잭션 템플릿 덕분에 트랜잭션을 시작하고, 커밋하거나 롤백하는 코드가 모두 제거되었다.
• 트랜잭션 템플릿의 기본 동작은 아래와 같다.
  • 비즈니스 로직이 정상 수행되면 커밋
  • 언체크 예외가 발생하면 롤백한다. 그 외의 경우 커밋한다.
• 코드에서 예외를 처리하기 위해 try~catch 가 들어갔는데, bizLogic() 메서드를 호출하면 SQLException 체크 예외를 넘겨준다. 해당 람다에서 체크 예외를 밖으로 던질 수 없기 때문에 언체크 예외로 바꾸어 던지도록 예외를 전환했다.

MemberServiceV3_2Test

package hello.jbdc.service

import hello.jbdc.connection.ConnectionConst
import hello.jbdc.domain.Member
import hello.jbdc.repository.MemberRepositoryV3
import org.assertj.core.api.Assertions.assertThat
import org.assertj.core.api.Assertions.assertThatThrownBy
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


internal class MemberServiceV3_2Test{

  val MEMBER_A = "memberA"
  val MEMBER_B = "memberB"
  val MEMBER_EX = "ex"

  lateinit var memberRepository: MemberRepositoryV3
  lateinit var memberService: MemberServiceV3_2

  @BeforeEach
  fun before(){
    val dataSource = DriverManagerDataSource(ConnectionConst.URL,
      ConnectionConst.USERNAME, ConnectionConst.PASSWORD)

    val transactionManager: PlatformTransactionManager = DataSourceTransactionManager(dataSource)
    memberRepository = MemberRepositoryV3(dataSource)
    memberService = MemberServiceV3_2(transactionManager, memberRepository)
  }

  @AfterEach
  @Throws(SQLException::class)
  fun after(){
    memberRepository.delete(MEMBER_A);
    memberRepository.delete(MEMBER_B);
    memberRepository.delete(MEMBER_EX);
  }

  @Test
  @DisplayName("정상 이체")
  @Throws(SQLException::class)
  fun accountTransfer() {
    //given
    val memberA = Member(MEMBER_A, 10000)
    val memberB = Member(MEMBER_B, 10000)
    memberRepository.save(memberA)
    memberRepository.save(memberB)
    //when
    memberService.accountTransfer(memberA.memberId,
      memberB.memberId, 2000)
//    //then
    val findMemberA: Member = memberRepository.findById(memberA.memberId)
    val findMemberB: Member = memberRepository.findById(memberB.memberId)
    assertThat(findMemberA.money).isEqualTo(8000)
    assertThat(findMemberB.money).isEqualTo(12000)
  }

  @Test
  @DisplayName("이체 중 예외 발생")
  fun accountTransferEx(){
    //given
    val memberA = Member(MEMBER_A, 10000)
    val memberEx = Member(MEMBER_EX, 10000)
    memberRepository.save(memberA)
    memberRepository.save(memberEx)
    // when

    assertThatThrownBy {
      memberService.accountTransfer(memberA.memberId, memberEx.memberId,
        2000)
    }
      .isInstanceOf(IllegalStateException::class.java)

    // then
    val findMemberA = memberRepository.findById(memberA.memberId)
    val findMemberEx = memberRepository.findById(memberEx.memberId)

    assertThat(findMemberA.money).isEqualTo(10000)
    assertThat(findMemberEx.money).isEqualTo(10000)

  }
}

정리

• 트랜잭션 템플릿 덕분에, 트랜잭션을 사용할 때 반복하는 코드를 제거할 수 있었다.
• 하지만 이곳은 서비스 로직인데 비즈니스 로직 뿐만 아니라 트랜잭션을 처리하는 기술 로직이 함께 포함되어 있다.
• 애플리케이션을 구성하는 로직을 핵심 기능과 부가 기능으로 구분하자면 서비스 입장에서 비즈니스 로직은 핵심 기능이고, 트랜잭션은 부가 기능이다.
• 이렇게 비즈니스 로직과 트랜잭션을 처리하는 기술 로직이 한 곳에 있으면 두 관심사를 하나의 클래스에서 처리하게 된다. 결과적으로 코드를 유지보수하기 어려워진다.
• 서비스 로직은 가급적 핵심 비즈니스 로직만 있어야 한다. 하지만 트랜잭션 기술을 사용하려면 어쩔 수 없이 트랜잭션 코드가 나와야 한다. 어떻게 하면 이 문제를 해결할 수 있을까?