Programming/Spring

Spring - 스프링 트랜잭션 이해

잇(IT) 2023. 7. 7. 00:47

- 스프링 트랜잭션 추상화

 

- JDBC 트랜잭션 코드 예시

public void accountTransfer(String fromId, String toId, int money) throws
SQLException {
 	Connection con = dataSource.getConnection();
 	try {
 		con.setAutoCommit(false); //트랜잭션 시작
 		//비즈니스 로직
		bizLogic(con, fromId, toId, money);
 		con.commit(); //성공시 커밋
	 } catch (Exception e) {
		 con.rollback(); //실패시 롤백
 		throw new IllegalStateException(e);
 	} finally {
 	release(con);
	}
}

 

- JPA 트랜잭션 코드 예시

public static void main(String[] args) {

	 //엔티티 매니저 팩토리 생성
	 EntityManagerFactory emf = Persistence.createEntityManagerFactory("jpabook");
 	EntityManager em = emf.createEntityManager(); //엔티티 매니저 생성
 	EntityTransaction tx = em.getTransaction(); //트랜잭션 기능 획득
    
 	try {
 		tx.begin(); //트랜잭션 시작
 		logic(em); //비즈니스 로직
 		tx.commit();//트랜잭션 커밋
        
 	} catch (Exception e) {
		 tx.rollback(); //트랜잭션 롤백
 	} finally {
 		em.close(); //엔티티 매니저 종료
 	}
 	emf.close(); //엔티티 매니저 팩토리 종료
}

 

- JDBC 기술을 사용하다 JPA 기술로 변경하면 트랜잭션을 사용하는 코드를 전부 변경해야 한다.

- 이런 문제를 해결하기 위해 스프링은 추상화를 제공한다. PlatformTransactionManger 인터페이스를 사용한다.

 

- PlatformTransactionManager 인터페이스

package org.springframework.transaction;

public interface PlatformTransactionManager extends TransactionManager {

	TransactionStatus getTransaction(@Nullable TransactionDefinition definition)
		throws TransactionException;
        
	void commit(TransactionStatus status) throws TransactionException;
	void rollback(TransactionStatus status) throws TransactionException;
}

 

1. 필요한 구현체를 스프링 빈으로 등록하고 주입 받아서 사용하기만 하면된다.

2. 스프링 부트는 데이터 접근 기술을 사용하는지를 자동으로 인식해서 적절한 트랜잭션 매니저를 선택해서 스프링 빈으로 등록해주기 때문에 등록하는 과정을 생략할 수 있다. JdbcTemplate , MyBatis 를 사용하면 DataSourceTransactionManager(JdbcTransactionManager) 를 스프링 빈으로 등록하고, JPA를 사용하면 JpaTransactionManager 를 스프링 빈으로 등록해준다.


- 프록시 도입 후

- 트랜잭션을 처리하기 위한 프록시를 적용하면 트랜잭션을 처리하는 객체와 비즈니스 로직을 처리하는 서비스 객체를 명확하게 분리할 수 있다.

 

- 트랜잭션 프록시 코드 예시

public class TransactionProxy {

 	private MemberService target;
 
	 public void logic() {
 		//트랜잭션 시작
 		TransactionStatus status = transactionManager.getTransaction(..);
 	try {
 	//실제 대상 호출
 	target.logic();
 	transactionManager.commit(status); //성공시 커밋
 	} catch (Exception e) {
 	transactionManager.rollback(status); //실패시 롤백
 	throw new IllegalStateException(e);
 	}
 	}
}

 

- 트랜잭션 프록시 적용 후 서비스 코드 예시

public class Service {
 	public void logic() {
	 //트랜잭션 관련 코드 제거, 순수 비즈니스 로직만 남음
 	bizLogic(fromId, toId, money);
 }
}

- 프록시 도입 전: 서비스에 비즈니스 로직과 트랜잭션 처리 로직이 함께 섞여있다.

- 프록시 도입 후: 트랜잭션 프록시가 트랜잭션 처리 로직을 모두 가져간다. 그리고 트랜잭션을 시작한 후에 실제 서비스를 대신 호출한다. 트랜잭션 프록시 덕분에 서비스 계층에는 순수한 비즈니즈 로직만 남길 수 있다.


- 프록시 도입 후 전체 과정

1. 트랜잭션은 커넥션에 con.setAutocommit(false) 를 지정하면서 시작한다.

2. 같은 트랜잭션을 유지하려면 같은 데이터베이스 커넥션을 사용해야 한다.

3. 이것을 위해 스프링 내부에서는 트랜잭션 동기화 매니저가 사용된다.

4. JdbcTemplate 을 포함한 대부분의 데이터 접근 기술들은 트랜잭션을 유지하기 위해 내부에서 트랜잭션 동기화 매니저를 통해 리소스(커넥션)를 동기화 한다.


- 트랜잭션 적용 확인

 

- 스프링 컨테이너에 트랜잭션 프록시 등록

1. @Transactional 애노테이션이 특정 클래스나 메서드에 하나라도 있으면 있으면 트랜잭션 AOP는 프록시를 만들어서 스프링 컨테이너에 등록한다. 그리고 실제 basicService 객체 대신에 프록시인 basicService$$CGLIB 를 스프링 빈에 등록한다. 그리고 프록시는 내부에 실제 basicService 를 참조하게 된다. 여기서 핵심은 실제 객체 대신에 프록시가 스프링 컨테이너에 등록되었다는 점이다.

2. 클라이언트인 txBasicTest 는 스프링 컨테이너에 @Autowired BasicService basicService 로 의존관계 주입을 요청한다. 스프링 컨테이너에는 실제 객체 대신에 프록시가 스프링 빈으로 등록되어 있기 때문에 프록시를 주입한다.

3. 프록시는 BasicService 를 상속해서 만들어지기 때문에 다형성을 활용할 수 있다. 따라서 BasicService 대신에 프록시인 BasicService$$CGLIB 를 주입할 수 있다


- 트랜잭션 프록시 동작 방식

- 클라이언트가 주입 받은 basicService$$CGLIB 는 트랜잭션을 적용하는 프록시이다.


- 트랜잭션 적용 위치

 

- 스프링에서 우선순위는 항상 더 구체적이고 자세한 것이 높은 우선순위를 가진다

 

@SpringBootTest
public class TxLevelTest {

    @Autowired LevelService service;

    @Test
    void orderTest() {
        service.write();
        service.read();
    }

    @TestConfiguration
    static class TxLevelTestConfig {
        @Bean
        LevelService levelService() {
            return new LevelService();
        }
    }

    @Slf4j
    @Transactional(readOnly = true)
    static class LevelService {

        @Transactional(readOnly = false)
        public void write() {
            log.info("call write");
            printTxInfo();
        }

        public void read() {
            log.info("call read");
            printTxInfo();
        }

        private void printTxInfo() {
            boolean txActive = TransactionSynchronizationManager.isActualTransactionActive();
            log.info("tx active={}", txActive);
            boolean readOnly = TransactionSynchronizationManager.isCurrentTransactionReadOnly();
            log.info("tx readOnly={}", readOnly);
        }
    }
}

- 스프링의 @Transactional 은 다음 두 가지 규칙이 있다.

1. 우선순위 규칙

2. 클래스에 적용하면 메서드는 자동 적용

 

- 위의 규칙에 의해서 클래스에 붙은 @Transactional보다 메서드에 붙은 @Transactional이 우선시 된다.

- 즉, write() 메서드의 경우 (readOnly = false)가 적용되고 read() 메서드의 경우 클래스에 붙은 (readOnly = true)가 적용된다.


- 트랜잭션 AOP 주의 사항 - 프록시 내부 호출1

 

- @Transactional 을 사용하면 스프링의 트랜잭션 AOP가 적용된다.

- @Transactional 을 적용하면 프록시 객체가 요청을 먼저 받아서 트랜잭션을 처리하고, 실제 객체를 호출해준다.

- 대상 객체의 내부에서 메서드 호출이 발생하면 프록시를 거치지 않고 대상 객체를 직접 호출하는 문제가 발생한다. 이렇게 되면 @Transactional 이 있어도 트랜잭션이 적용되지 않는다.

@Slf4j
@SpringBootTest
public class InternalCallV1Test {

    @Autowired
    CallService callService;

    @Test
    void printProxy() {
        log.info("callService class={}", callService.getClass());
    }

    @Test
    void internalCall() {
        callService.internal();
    }

    @Test
    void externalCall(){
        callService.external();
    }

    @TestConfiguration
    static class InternalCallV1TestConfig {

        @Bean
        CallService callService() {
            return new CallService();
        }
    }


    @Slf4j
    static class CallService {

        public void external() {
            log.info("call external");
            printTxInfo();
            internal();
        }

        @Transactional
        public void internal() {
            log.info("call internal");
            printTxInfo();
        }

        private void printTxInfo() {
            boolean txActive = TransactionSynchronizationManager.isActualTransactionActive();
            log.info("tx active={}", txActive);
            boolean readOnly = TransactionSynchronizationManager.isCurrentTransactionReadOnly();
            log.info("tx readOnly={}", readOnly);
        }
    }
}

1. external() 은 트랜잭션이 없다.

2. internal() 은 @Transactional 을 통해 트랜잭션을 적용한다

 

- @Transactional 이 하나라도 있으면 트랜잭션 프록시 객체가 만들어진다. 그리고 callService 빈을 주입 받으면 트랜잭션 프록시 객체가 대신 주입된다.

 

- printProxy() 메서들 실행

- printProxy의 메서드의 경우 프록시 객체가 생성된 것을 확인 할 수 있다.

 

- internalCall() 메서드 실행

- internalCall()을 호출 시 정상적으로 트랜잭션이 시작된 것을 확인 할 수 있다.

1. 클라이언트인 테스트 코드는 callService.internal() 을 호출한다. 여기서 callService 는 트랜잭션 프록시이다.

2. callService 의 트랜잭션 프록시가 호출된다.

3. internal() 메서드에 @Transactional 이 붙어 있으므로 트랜잭션 프록시는 트랜잭션을 적용한다.

4. 트랜잭션 적용 후 실제 callService 객체 인스턴스의 internal() 을 호출한다. 실제 callService 가 처리를 완료하면 응답이 트랜잭션 프록시로 돌아오고, 트랜잭션 프록시는 트랜잭션을 완료한다.

 

 

- external() 은 @Transactional 애노테이션이 없다. 따라서 트랜잭션 없이 시작한다. 그런데 내부에서 @Transactional 이 있는 internal() 을 호출하는 것을 확인할 수 있다.

- 이 경우 external() 은 트랜잭션이 없지만, internal() 에서는 트랜잭션이 적용되는 것 처럼 보인다.

- 실행 로그를 보면 트랜잭션 관련 코드가 전혀 보이지 않는다. 프록시가 아닌 실제 callService 에서 남긴 로그만 확인된다. 추가로 internal() 내부에서 호출한 tx active=false 로그를 통해 확실히 트랜잭션이 수행되지 않은 것을 확인할 수 있다

1. 클라이언트인 테스트 코드는 callService.external() 을 호출한다. 여기서 callService 는 트랜잭션 프록시이다.

2. callService 의 트랜잭션 프록시가 호출된다.

3. external() 메서드에는 @Transactional 이 없다. 따라서 트랜잭션 프록시는 트랜잭션을 적용하지 않는다.

4. 트랜잭션 적용하지 않고, 실제 callService 객체 인스턴스의 external() 을 호출한다. 5. external() 은 내부에서 internal() 메서드를 호출한다. 그런데 여기서 문제가 발생한다.

 

- 자바 언어에서 메서드 앞에 별도의 참조가 없으면 this 라는 뜻으로 자기 자신의 인스턴스를 가리킨다. 결과적으로 자기 자신의 내부 메서드를 호출하는 this.internal() 이 되는데, 여기서 this 는 자기 자신을 가리키므로, 실제 대상 객체( target )의 인스턴스를 뜻한다. 결과적으로 이러한 내부 호출은 프록시를 거치지 않는다. 따라서 트랜잭션을 적용할 수 없다. 결과적으로 target 에 있는 internal() 을 직접 호출하게 된 것이다.

 

- 가장 단순한 방법은 내부 호출을 피하기 위해 internal() 메서드를 별도의 클래스로 분리하는 것이다.


- 트랜잭션 AOP 주의 사항 - 프록시 내부 호출2

 

- InternalCallV2Test

@Slf4j
@SpringBootTest
public class InternalCallV2Test {

    @Autowired
    CallService callService;

    @Test
    void printProxy() {
        log.info("callService class={}", callService.getClass());
    }

    @Test
    void externalCallV2(){
        callService.external();
    }

    @TestConfiguration
    static class InternalCallV1TestConfig {

        @Bean
        CallService callService() {
            return new CallService(internalService());
        }

        @Bean
        InternalService internalService() {
            return new InternalService();
        }
    }


    @Slf4j
    @RequiredArgsConstructor
    static class CallService {

        private final InternalService internalService;

        public void external() {
            log.info("call external");
            printTxInfo();
            internalService.internal();
        }

        private void printTxInfo() {
            boolean txActive = TransactionSynchronizationManager.isActualTransactionActive();
            log.info("tx active={}", txActive);
            boolean readOnly = TransactionSynchronizationManager.isCurrentTransactionReadOnly();
            log.info("tx readOnly={}", readOnly);
        }
    }

    static class InternalService {

        @Transactional
        public void internal() {
            log.info("call internal");
            printTxInfo();
        }

        private void printTxInfo() {
            boolean txActive = TransactionSynchronizationManager.isActualTransactionActive();
            log.info("tx active={}", txActive);
            boolean readOnly = TransactionSynchronizationManager.isCurrentTransactionReadOnly();
            log.info("tx readOnly={}", readOnly);
        }
    }
}

1. InternalService 클래스를 만들고 internal() 메서드를 여기로 옮겼다.

2. 이렇게 메서드 내부 호출을 외부 호출로 변경했다.

3. CallService 에는 트랜잭션 관련 코드가 전혀 없으므로 트랜잭션 프록시가 적용되지 않는다.

4, InternalService 에는 트랜잭션 관련 코드가 있으므로 트랜잭션 프록시가 적용된다.

 

1. 클라이언트인 테스트 코드는 callService.external() 을 호출한다.

2. callService 는 실제 callService 객체 인스턴스이다.

3. callService 는 주입 받은 internalService.internal() 을 호출한다.

4. internalService 는 트랜잭션 프록시이다. internal() 메서드에 @Transactional 이 붙어 있으므로 트랜잭션 프록시는 트랜잭션을 적용한다.

5. 트랜잭션 적용 후 실제 internalService 객체 인스턴스의 internal() 을 호출한다.


- public 메서드만 트랜잭션 적용


- 트랜잭션 AOP 주의 사항 - 초기화 시점

 

@SpringBootTest
public class InitTxTest {

    @Autowired Hello hello;

    @Test
    void go() {
        //초기화 코드는 스프링이 초기화 시점에 호출한다.
    }

    @TestConfiguration
    static class InitTxTestConfig {

        @Bean
        Hello hello(){
            return new Hello();
        }
    }

    @Slf4j
    static class Hello {

        @PostConstruct
        @Transactional
        public void initV1() {
            boolean isActive = TransactionSynchronizationManager.isActualTransactionActive();
            log.info("Hello init @PostConstruct tx active={}", isActive);
        }

        @EventListener(ApplicationReadyEvent.class) //스프링 컨테이너가 다 뜬 다음 호출해준다.
        @Transactional
        public void initV2() {
            boolean isActive = TransactionSynchronizationManager.isActualTransactionActive();
            log.info("Hello init ApplicationReadyEvent tx active={}", isActive);
        }
    }
}

- 초기화 코드(예: @PostConstruct )와 @Transactional 을 함께 사용하면 트랜잭션이 적용되지 않는다

- 왜냐하면 초기화 코드가 먼저 호출되고, 그 다음에 트랜잭션 AOP가 적용되기 때문이다. 따라서 초기화 시점에는 해당 메서드에서 트랜잭션을 획득할 수 없다.

 

@EventListener(value = ApplicationReadyEvent.class)
@Transactional
public void init2() {
 	log.info("Hello init ApplicationReadyEvent");
}

- 이 이벤트는 트랜잭션 AOP를 포함한 스프링이 컨테이너가 완전히 생성되고 난 다음에 이벤트가 붙은 메서드를 호출해준다. 따라서 init2() 는 트랜잭션이 적용된 것을 확인할 수 있다.


- 예외와 트랜잭션 커밋, 롤백 - 기본

- 예외 발생시 스프링 트랜잭션 AOP는 예외의 종류에 따라 트랜잭션을 커밋하거나 롤백한다.

1. 언체크 예외인 RuntimeException , Error 와 그 하위 예외가 발생하면 트랜잭션을 롤백한다.

2. 체크 예외인 Exception 과 그 하위 예외가 발생하면 트랜잭션을 커밋한다.

3. 물론 정상 응답(리턴)하면 트랜잭션을 커밋한다.


- 예외와 트랜잭션 커밋, 롤백 - 활용

 

- 비지니스 예외 생성

package hello.springtx.order;

public class NotEnoughMoneyException extends Exception{

    public NotEnoughMoneyException(String message) {
        super(message);
    }
}

- Exception 예외를 상속 받아 체크 예외에 해당한다.

 

- Order 엔티티 생성

package hello.springtx.order;

import lombok.Getter;
import lombok.Setter;

import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.Id;
import javax.persistence.Table;

@Entity //JPA 사용하려면 꼭 작성
@Table(name = "orders") //테이블과 매핑해서 사용하기 위해 작성한 것이다.
@Getter
@Setter
public class Order {

    @Id
    @GeneratedValue
    private Long id;

    private String username; //정상, 예외, 잔고부족
    private String payStatus; //대기, 완료
}

- @Table(name = "orders") 라고 했는데, 테이블 이름을 지정하지 않으면 테이블 이름이 클래스 이름인 order 가 된다. order 는 데이터베이스 예약어( order by )여서 사용할 수 없다. 그래서 orders 라는 테이블 이름을 따로 지정해주었다.

 

- OrderRepository

package hello.springtx.order;

import org.springframework.data.jpa.repository.JpaRepository;

public interface OrderRepository extends JpaRepository<Order, Long> {
}

 

- OrderService

package hello.springtx.order;

import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.weaver.ast.Or;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

@Slf4j
@Service
@RequiredArgsConstructor
public class OrderService {

    private final OrderRepository orderRepository;

    //JPA는 트랜잭션 커밋 시점에 Order 데이터를 DB에 반영한다.
    @Transactional
    public void order(Order order) throws NotEnoughMoneyException {
        log.info("order 호출");
        orderRepository.save(order);

        log.info("결제 프로세스 진입");
        if (order.getUsername().equals("예외")) {
            log.info("시스템 예외 발생");
            throw new RuntimeException("시스템 예외");
        } else if (order.getUsername().equals("잔고부족")) {
            log.info("잔고 부족 비즈니스 예외 발생");
            order.setPayStatus("대기");
            throw new NotEnoughMoneyException("잔고가 부족합니다.");
        } else {
            //정상 승인
            log.info("정상 승인");
            order.setPayStatus("완료");
        }
        log.info("결제 프로스세스 완료");
    }
}

- 여러 상황을 만들기 위해서 사용자 이름( username )에 따라서 처리 프로세스를 다르게 했다.

1. 기본 : payStatus 를 완료 상태로 처리하고 정상 처리된다.

2. 예외 : RuntimeException("시스템 예외") 런타임 예외가 발생한다.

3. 잔고부족 :

   3.1. payStatus 를 대기 상태로 처리한다.

   3.2. NotEnoughMoneyException("잔고가 부족합니다") 체크 예외가 발생한다.

   3.3. 잔고 부족은 payStatus 를 대기 상태로 두고, 체크 예외가 발생하지만, order 데이터는 커밋되기를 기대한다.

 

- OrderServiceTest

package hello.springtx.order;

import lombok.extern.slf4j.Slf4j;
import org.assertj.core.api.Assertions;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import java.util.Optional;

import static org.assertj.core.api.Assertions.*;

@Slf4j
@SpringBootTest
class OrderServiceTest {

    @Autowired
    OrderService orderService;
    @Autowired
    OrderRepository orderRepository;

    @Test
    void complete() throws NotEnoughMoneyException {
        //given
        Order order = new Order();
        order.setUsername("정상입니다");

        //when
        orderService.order(order);

        //then
        Order findOrder = orderRepository.findById(order.getId()).get();
        assertThat(findOrder.getPayStatus()).isEqualTo("완료");
    }

    @Test
    void runtimeException() throws NotEnoughMoneyException {
        //given
        Order order = new Order();
        order.setUsername("예외");

        //when
        assertThatThrownBy(() -> orderService.order(order))
                .isInstanceOf(RuntimeException.class);

        //then
        Optional<Order> orderOptional = orderRepository.findById(order.getId());
        assertThat(orderOptional.isEmpty()).isTrue();
    }

    @Test
    void bizException() {
        //given
        Order order = new Order();
        order.setUsername("잔고부족");

        //when
        try {
            orderService.order(order);
            fail("잔고 부족 예외가 발생해야 합니다.");
        } catch (NotEnoughMoneyException e) {
            log.info("고객에게 잔고 부족을 알리고 별도의 계좌로 입금하도록 안내");
        }
        //then
        Order findOrder = orderRepository.findById(order.getId()).get();
        assertThat(findOrder.getPayStatus()).isEqualTo("대기");
    }
}

 

- complete() 메서드 실행

 

- runtimeException() 메서드 호출

 

- bizException() 메서드 실행

 

 

 

 

 

 

 

 

 

 

 

 

출처 : 인프런 - 우아한 형제들 기술이사 김영한의 스프링 완전 정복 (스프링 DB 2편 - 데이터 접근 활용 기술)

728x90