fix: Register async executor as a bean using @Bean

[psh10066]

fix: Graceful Shutdown 버그 수정 위한 AsyncExecutor Bean 등록

안녕하세요! 해당 템플릿 기반으로 프로젝트를 진행하다가 비동기 동작에서 Graceful Shutdown이 제대로 동작하지 않는 문제를 발견하여, 버그를 수정하여 PR 요청 드립니다.

최초 발견

Spring Batch App을 구성하고 배치 수행 결과를 Slack으로 발송하는 로직을 비동기로 구현하였는데, 배치가 종료된 이후 Slack 메시지가 발송되지 않고 App이 종료되는 문제를 확인하였습니다.

검증

이 문제를 제대로 확인하기 위해 아래와 같은 테스트를 진행하였습니다.
해당 테스트는 프로젝트의 PR의 수정 사항에 추가하지는 않고, 따로 진행하였습니다.
(필요하시다면 해당 테스트 코드는 테스트 커밋 링크에서 직접 확인하실 수 있습니다.)

1. 임시 비동기 3초 로직 수행 메서드 추가

   @Service
   class AsyncService {
   
       private val logger = LoggerFactory.getLogger(javaClass)
       val isTaskCompleted = AtomicBoolean(false)
   
       @Async
       fun longRunningTask() {
           logger.info("비동기 작업을 시작합니다.")
   
           Thread.sleep(3000)
           isTaskCompleted.set(true)
   
           logger.info("비동기 작업이 성공적으로 완료되었습니다.")
       }
   }

2. 아래 테스트 코드 구성 후 실패 확인

   class AsyncServiceTest {
   
       @Test
       fun gracefulShutdown() {
           val context = SpringApplicationBuilder(
               AsyncConfig::class.java,
               AsyncService::class.java
           )
               .web(WebApplicationType.NONE)
               .run()
           val asyncService = context.getBean(AsyncService::class.java)
   
           asyncService.longRunningTask()
           Thread.sleep(500)
           context.close()
   
           assertThat(asyncService.isTaskCompleted.get()).isTrue()
       }
   }

   

수정

기존 ThreadPoolTaskExecutor가 Bean으로 등록되어 있지 않아 스프링 컨텍스트 종료 시 Graceful Shutdown이 제대로 동작하지 않은 것으로 보고, ThreadPoolTaskExecutor를 Bean으로 관리하도록 설정하였습니다.
또한 Bean으로 등록됨에 따라 initialize가 불필요해지게 되어 해당 부분 삭제하였습니다.
(이에 대한 내용은 @EnableAsync의 주석에서도 확인할 수 있었습니다.)

검증

동일한 테스트 코드 실행 결과 성공하는 것을 확인하였습니다.

참고 사항

잘못된 내용이나 보완해야 할 내용이 있다면 편하게 말씀 부탁드립니다.
감사합니다.

[geminiKim]

흠.. 이 구성에서 배치 어플리케이션을 구성한다면 별개의 처리가 필요할 것 같은데요
그 측면에서 이 설정을 하는게 맞는가에 대한 고민을 조금하느라 의견을 드리는게 늦어지긴했습니다.

일단 제 기준에선 모호함이 있긴 했는데요, 그럼에도 프레임워크 단에서도 가이드로 안내하는 점, 어떤 방식으로 활용해도 graceful shutdown 이 잘 되야하는건 굉장히 중요한점을 기준으로 반영하도록 하겠습니다!

기여 감사드립니다! 👍

[psh10066]

흠.. 이 구성에서 배치 어플리케이션을 구성한다면 별개의 처리가 필요할 것 같은데요 그 측면에서 이 설정을 하는게 맞는가에 대한 고민을 조금하느라 의견을 드리는게 늦어지긴했습니다.

일단 제 기준에선 모호함이 있긴 했는데요, 그럼에도 프레임워크 단에서도 가이드로 안내하는 점, 어떤 방식으로 활용해도 graceful shutdown 이 잘 되야하는건 굉장히 중요한점을 기준으로 반영하도록 하겠습니다!

기여 감사드립니다! 👍

말씀해 주신 모호한 점에 대해서도 추가로 고민해 볼 수 있는 계기가 되었습니다!
감사합니다 :)