Spring Batch의 트랜잭션 범위는 어디까지일까?

개요

어제 팀원들과 이야기 하던 중, 스프링 배치의 트랜잭션에 대해서 이야기가 나와서 오프라인에서 간략하게 설명을 드렸었는데요.

이야기 하고 보니, 팀 내에 공유해도 좋을 만한 내용이라 생각되어서 작성하게 되었습니다.

본 글은 스프링 배치를 쓰면서 스프링 배치의 트랜잭션이 정확히 어떻게 적용되는지를 이해하는 것을 목표로 합니다.

Chunk와 Task

스프링 배치에서 가장 중요한 개념은 바로 Task 와 Chunk 입니다.

스프링 배치는 기본적으로 Task Based와 Chunk Based Transaction 두 가지 방식을 지원합니다.

Task Based

Task Based Transaction의 경우 단일 작업으로 처리되는 경우 (단순히 동작하는 1회성 배치?)에 적합한 방식입니다.

Chunk Based

Chunk Based는 우리가 익히 사용해오고 있던 예약의 배치 프로세스입니다.

Reader -> Processor -> Writer로 이어지는 3단 콤보죠.

여기서 이름 그대로 저희는 Chunk라고 하는 사이즈를 지정해서 배치 애플리케이션을 작성해왔는데요.

위 스탭만 보면 마치 Reader 스탭도 트랜잭션으로 엮여있는 것 처럼 보여서 많이들 헷갈리셨던 것 같습니다.

Reader가 트랜잭션으로 엮인 것은 사실 개념적으로는 맞지만, 개발 구현 측면에선 아닙니다.

Chunk Based 방식의 철학은 Chunk 단위로 트랜잭션을 걸것이며, 롤백도 Chunk 단위로 하겠다라는 것입니다.

아래에서도 자세히 들어가보겠지만, Chunk Based는 Reader 함수에서 조회된 결과를 Chunk 단위로 자르고, Processor 스텝으로 넘기게 되는데, 이 때 넘어가는 Chunk 데이터가 트랜잭션으로 묶이는 것이지 Reader 스탭 자체가 트랜잭션으로 묶이는 것은 아닙니다.

말이 좀 어려워서 그렇지 막상 코드로 보면 별 것 아닙니다.

Spring Batch Chunk Based 동작 방식 Deep Dive

시작은 TransactionTemplate에서 시작합니다.

본 내용의 핵심인 구현체 ChunkTransactionCallback로 진입합니다.

doInTransaction 메소드 내부에서 tasklet.execute(contribution, chunkContext);가 있는데, 여기서 tasklet의 구현체 중 ChunkOrientedTasklet을 선택합니다. (이유는 예약에서는 Chunk 기반 배치를 사용하기 때문)

이 때 눈 여겨서 봐야할 부분은 chunkProvider.provide(contribution);와 chunkProcessor.process(contribution, inputs); 입니다.

chunkProvider.provide(contribution);의 경우 Reader 스탭에서 반환된 결과 값을 Chunk 단위로 잘라서 저장하는 함수입니다.

잘 보면 리스트에 하나씩 값을 담는 것을 볼 수 있습니다.

위에서 얻어진 Chunk를 process라는 스탭으로 넘기게 되는데요.

이 때 이 process가 우리가 익히 잘 알고 있는 processor 스탭 + writer 스탭 입니다.

이 프로세서에서 transform이라는 메소드가 우리가 작성해둔 processor 로직을 기반으로 객체의 타입을 변경하는 부분입니다.

거기서 얻어진 Chunk를 writer로 보내게 됩니다.

여기 까지 보면, 결국 트랜잭션은 Chunk를 reader가 반환한 시점부터 적용된다고 볼 수 있는데요.

실제로 확인해본 결과 스프링 배치는 reader 함수에서는 별도로 트랜잭션을 걸어주지 않는다는 사실을 확인할 수 있었습니다.

출처

• 스프링 공식 문서
• 나의 뇌와 눈