[Spring] @Async Annotation(비동기 메소드)

3 분 소요

@Async

@Async Annotation은 Spring에서 제공하는 Thread Pool을 활용하는 비동기 메소드 지원 Annotation이다.

기존 Java에서 비동기 방식으로 메소드를 구현할 때는 아래와 같이 구현할 수 있었다.

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class CharlesAsync {

    static ExecutorService executorService = Executors.newFixedThreadPool(5);

    public void asyncMethod(final String message) throws Exception {
        executorService.submit(new Runnable() {
            @Override
            public void run() {
                // do something
            }            
        });
    }
}

java.util.concurrent.ExecutorService을 활용해서 비동기 방식의 method를 정의 할 때마다

위와 같이 Runnable의 run()을 재구현해야 하는 등의 동일한 작업들의 반복이 잦았다.

With @Async

@Async Annotation을 활용하면 손쉽게 비동기 메소드 작성이 가능하다.

만약 Spring Boot에서 간단히 사용하고 싶다면, 단순히 Application Class에 @EnableAsync Annotation을 추가하고,

@EnableAsync
@SpringBootApplication
public class SpringBootApplication {
    ...
}

비동기로 작동하길 원하는 method 위에 @Async Annotation을 붙여주면 사용할 수 있다.

public class CharlesAsync {

    @Async
    public void asyncMethod(final String message) throws Exception {
        ....
    }
}

위와 같은 사용은 간단하지만 @Async의 기본설정인 SimpleAsyncTaskExecutor를 사용한다.

본인의 개발 환경에 맞게 Customize하기에는 직접 AsyncConfigurerSupport를 상속받는 Class를 작성하는 것이 좋다.

AsyncConfigurerSupport

아래와 같은 AsyncConfigurerSupport를 상속 받는 Customize Class를 구현하자.

import java.util.concurrent.Executor;

import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.AsyncConfigurerSupport;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

@Configuration
@EnableAsync
public class AsyncConfig extends AsyncConfigurerSupport {
    
    @Override
    public Executor getAsyncExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(5);
        executor.setMaxPoolSize(30);
        executor.setQueueCapacity(50);
        executor.setThreadNamePrefix("CHARLES-ASYNC-");
        executor.initialize();
        return executor;
    }
}

여기서 설정한 요소들은 아래와 같다.

@Configuration: Spring 설정 관련 Class로 @Component 등록되어 Scanning 될 수 있다.
@EnableAsync: Spring method에서 비동기 기능을 사용하능하게 활성화 한다.
CorePoolSize: 기본 실행 대기하는 Thread의 수
MaxPoolSize: 동시 동작하는 최대 Thread의 수
QueueCapacity: MaxPoolSize 초과 요청에서 Thread 생성 요청시, 해당 요청을 Queue에 저장하는데 이때 최대 수용 가능한 Queue의 수, Queue에 저장되어 있다가 Thread에 자리가 생기면 하니씩 빠져나가 동작
ThreadNamePrefix: 생성되는 Thread 접두사 지정

위와 같이 작성한 후 비동기 방식 사용을 원하는 method에 @Async Annotation을 지정해주면 된다.

주의사항

@Async Annotation을 사용할 때 아래와 같은 세 가지 사항을 주의하자.

private method는 사용 불가, public method만 사용 가능
self-invocation(자가 호출) 불가, 즉 inner method는 사용 불가
QueueCapcity 초과 요청에 대한 비동기 method 호출시 방어 코드 작성

위 주의사항을 아래 사진과 함께 설명을 해보면,

@Async의 동작은 AOP가 적용되어 Spring Context에서 등록된 Bean Object의 method가 호출 될 시에,

Spring이 확인할 수 있고 @Async가 적용된 method의 경우 Spring이 method를 가로채 다른 Thread에서 실행 시켜주는 동작 방식이다.

이때문에 Spring이 해당 @Async method를 가로챈 후, 다른 Class에서 호출이 가능해야 하므로,

private method는 사용할 수 없는 것이다.

또한 Spring Context에 등록된 Bean의 emthod의 호출이어야 Proxy 적용이 가능하므로,

inner method의 호출은 Proxy 영향을 받지 않기에 self-invocation이 불가능하다.

위 주의사항을 아래 예시 코드와 함께 살펴보자

self-invocation(자가 호출) 불가

위에서 작성한 Asyncconfig를 사용하는 Spring Project에서 아래와 같이,

같은 Class에 존재한 method에 @Async Annotation을 작성해 비동기 방식을 사용해보자.

@Controller
public Class TestController {

    @Async
    public void asyncMethod(int i) {
        try {
            Thread.sleep(500);
            log.info("[AsyncMethod]"+"-"+i);
        } catch(InterruptedException e) {
            e.printStackTrace();
        }
    }

    @GetMapping("async")
    public String testAsync() {
        log.info("TEST ASYNC");
        for(int i=0; i<50; i++) {
            asyncMethod(i);
        }
        return "";
    }
}

작동 결과를 보면 비동기 방식으로 호출되지 않았고, 동기적으로 호출 순서대로 동작하는 것을 확인할 수 있다.

자가 호출에서는 @async 사용이 불가하다.

하지만, @Service로 Bean 등록된 Service를 통해 주입하여 위 코드를 다시 작성해보면,

@Service
public class TestService {
    @Async
    public void asyncMethod(int i) {
        try {
            Thread.sleep(500);
            log.info("[AsyncMethod]"+"-"+i);
        } catch(InterruptedException e) {
            e.printStackTrace();
        }
    }
}

@AllArgsConstructor
@Controller
public Class TestController {

    private TestService testService;

    @GetMapping("async")
    public String testAsync() {
        log.info("TEST ASYNC");
        for(int i=0; i<50; i++) {
            testService.asyncMethod(i);
        }
        return "";
    }
    
}

위 사진과 같이 호출 순서에 상관없이 비동기 방식으로 method가 호출 되었고, @AsyncConfig에서 prefix로 작성한 접두사도 정상적으로 붙은 것을 확인할 수 있다.

QueueCapacity 초과 요청 방어 코드 작성

이번엔 AsyncConfig에서 PoolSize와 QueueCapacity를 줄여보고 위 코드를 다시 실행해보자

@Configuration
@EnableAsync
public class AsyncConfig extends AsyncConfigurerSupport {
    
    @Override
    public Executor getAsyncExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(2);
        executor.setMaxPoolSize(10);
        executor.setQueueCapacity(10);
        executor.setThreadNamePrefix("CHARLES-ASYNC-");
        executor.initialize();
        return executor;
    }
}

Exception이 Throw 되었다.

Exception에서 주요한 사항을 살펴보면

Request processing failed; nested exception is org.springframework.core.task.TaskRejectedException: 
Executor [java.util.concurrent.ThreadPoolExecutor@116e53a0
[Running, pool size = 10, active threads = 10, queued tasks = 10, completed tasks = 0]] 
did not accept task: org.springframework.aop.interceptor.AsyncExecutionInterceptor$$
Lambda$2031/170931344@7ef0a05e] with root cause

TaskRejectedException으로 수행된 task는 0으로 설정 thread 수, pool size 는 10, queued 된 tasks = 10개로 최대 수용 가능한 Thread Pool 수와 QueueCapacity 까지 초과된 요청이 들어오자.

Task를 Reject하는 Exception이 발생하였다.

아래와 같이 TaskRejectedException 발생 시 handling 해주는 방어 코드를 작성하자.

@AllArgsConstructor
@Controller
public Class TestController {

    private TestService testService;

    @GetMapping("async")
    public String testAsync() {
        log.info("TEST ASYNC");
        try {
            for(int i=0; i<50; i++) {
                testService.asyncMethod(i);
        } catch (TaskRejectedException e) {
            // ....
        }
        return "";
    }
    
}

참고자료

https://www.hanumoka.net/2020/07/02/springBoot-20200702-sringboot-async-service/
https://jeong-pro.tistory.com/187
https://dzone.com/articles/effective-advice-on-spring-async-part-1
https://velog.io/@gillog/Spring-Async-Annotation%EB%B9%84%EB%8F%99%EA%B8%B0-%EB%A9%94%EC%86%8C%EB%93%9C-%EC%82%AC%EC%9A%A9%ED%95%98%EA%B8%B0

Twitter Facebook LinkedIn

LEE CHANWOO

[Spring] @Async Annotation(비동기 메소드)

@Async

With @Async

AsyncConfigurerSupport

주의사항

self-invocation(자가 호출) 불가

QueueCapacity 초과 요청 방어 코드 작성

공유하기

댓글남기기

참고

[Python] FastAPI Uvicorn + Gunicorn

[Python] FastAPI 동기 vs 비동기

[Linux] Tmux 사용법 및 주요 명령어

[MLOps] 데이터 버전 관리 DVC