Spring MVC의 비동기처리 (ThreadPoolTaskExecutor 사용)

Spring MVC에서 프로세스는 기본적으로 동기식 처리로 이루어집니다.

코드 한줄한줄 차례대로 실행이 완료된 후 넘어갈 수 있다는 뜻이죠. 보통 이걸 Blocking 이라고 하고 반대가 되는 말은 Non-Blocking 이라고 합니다. 코드가 차례대로 실행되기 때문에 코드 실행에 따른 결과 값을 보는게 명확한 장점이 있습니다.

 

Node.js로 개발을 좀 해오다보니 (Node.js는 반대로 비동기처리가 기본 컨셉입니다.) Async에 대한 용어에 더 눈이가서 Spring 어노테이션중 @Async 어노테이션에 먼저 눈이 갔습니다. 어노테이션 하나만 추가하면 내가 원하는게 된다?

 

@Async

@Async는 Spring 프레임워크에서 비동기적인 메서드 실행을 지원하기 위한 어노테이션입니다. 이 어노테이션을 메서드에 붙이면 해당 메서드는 별도의 쓰레드에서 실행되며, 호출한 쓰레드는 메서드가 완료될 때까지 블로킹되지 않습니다. @Async를 사용하면 동기적인 방식이 아닌 비동기적인 방식으로 메서드를 실행할 수 있습니다.

 

메소드에 그냥 어노테이션만 붙이면 Async 메소드가 되는 것인가? 했지만 @Async를 사용할 때 주의할 점은 이 어노테이션도 Thread pool을 사용하는 것이기 때문에 무분별하게 사용하면 오버헤드가 발생할 수도 있고, 스프링의 프록시 매커니즘을 통해 비동기 처리가 구현되기 때문에 @Async 메서드는 같은 클래스 내에서 호출되면 비동기적으로 실행되지 않을 수 있습니다. 이 경우 AOP 프록시가 적용되지 않기 때문입니다.

 

저같은 경우도 Spring MVC 프로젝트에 기본적으로 Filter를 사용하고있는데 위 내용때문에 제대로 작동하지 않는 부분을 확인하고 바로 다른 방법을 찾아보았습니다.

 

ThreadPoolTaskExecutor

스프링에서 제공해주는 클래스로 쓰레드풀을 이용하여 멀티쓰레드 구현을 쉽게 해주는 클래스입니다.

일단 제가 의도한 부분은 API 호출 시에 내부 처리와 상관없이 빠르게 응답 값을 주는걸 생각했습니다.

그러기위해선 부하가 큰 Service 메소드에 대한 Async 처리가 필요했고, 아래와같이 처리했습니다.

 

AsyncConfig.class

: 쓰레드풀을 사용하기 위한 custom 설정을 하는 클래스입니다.

@Configuration
public class AsyncConfig {

    @Bean
    public ThreadPoolTaskExecutor taskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(10);  // 쓰레드풀의 기본 크기
        executor.setMaxPoolSize(50);   // 쓰레드풀의 최대 크기
        executor.setQueueCapacity(100); // 작업 큐 크기
        executor.setThreadNamePrefix("Custom-Async-"); // 쓰레드 이름 접두사 설정
        executor.initialize();
        return executor;
    }
}

 

Controller코드

: 서비스 메소드를 백그라운드 로직으로 수정하였고, DeferredResult 객체에 비동기로직에 대한 결과 값을 저장할 수 있습니다.

@RequiredArgsConstructor
public class AlarmController {
    private final AlarmService alarmService;
    @Qualifier("taskExecutor")
    private final ThreadPoolTaskExecutor taskExecutor;

   @PostMapping(value = "/{id}/send")
   public Response<Map<String,Object>> immediatelyPush(
           HttpServletRequest req,
           @PathVariable("id") Long id
   ) {
       try {
           // DeferredResult<Response<Map<String, Object>>> deferredResult = new DeferredResult<>();

           taskExecutor.submit(() -> {
               // 비동기로 실행되는 백그라운드 로직
               alarmService.immediatelyPush(id);
               // deferredResult.setResult(new Response<>(ResultCode.REGISTER_SUCCESS));
               // deferredResult.setResult(new Response<>(ResultCode.REGISTER_FAIL));
           });
           return new Response<>(ResultCode.REGISTER_SUCCESS);
       } catch (Exception e) {
           return new Response<>(ResultCode.REGISTER_FAIL);
       }
   }
}

이렇게되면 서비스 메소드에 대한 부분이 실행완료되지않아도 즉시 response를 넘겨주게 됩니다. 물론 response후에도 Spring Application 에서는 해당 로직이 실행중이게 됩니다.