비동기 멀티 스레드를 위한 ThreadPoolTaskExecutor

🔗1. 비동기 적용하기

🔗 2. 비동기 커스텀 

---

1. 비동기 적용하기

SpringBoot 프로젝트에서는 @EnableAsync 어노테이션을 이용해서 간단하게 비동기를 구현할 수 있다. 

비동기를 적용하는 방법에는 2가지가 있다.

1. SpringBootApplication 에 @EnableAsync 어노테이션 사용하기
2. AsyncConfig 작성 (+@EnableAsync)

추가로 원하는 메서드에 @Async 어노테이션을 달아주면 해당 메소드가 비동기로 작동하게 된다.

1) SpringBootApplication에 @EnableAsync 

@EnableAsync
@SpringBootApplication
public class MyApplication {

어플리케이션에 @EnableAsync 를 사용하는 것 만으로 간단하게 비동기를 구현할 수 있다.
하지만 어플리케이션에 직접 어노테이션을 사용할 경우, EnableAsync의 기본 설정으로 동작을 하게 된다.

ＥｎａｂｌｅＡｓｙｎｃ．ｊａｖａ

taskExecutor나 Executor bean을 찾는데 없다면 SimpleAsyncTaskExecutor로 동작하기 때문이다. 
그래서 대개는 TaskExecutor 를 따로 구현해서 사용하게 된다.

2) AsyncConfig 파일 작성

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

import java.util.concurrent.Executor;

@EnableAsync
@Configuration
public class AsyncConfig {
    private final int CORE_POOL_SIZE=3;
    private final int MAX_POOL_SIZE = 10;
    private int QUEUE_CAPACITY = 100_000;

    @Bean(name = "task-executor")
    public Executor threadPoolExecutor(){

        ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();

        taskExecutor.setCorePoolSize(CORE_POOL_SIZE);   // core thread pool size
        taskExecutor.setMaxPoolSize(MAX_POOL_SIZE);     // 요청 queue size 이상일 시 thread Pool size
        taskExecutor.setQueueCapacity(QUEUE_CAPACITY);  // core full일 시 대기 queue size
        taskExecutor.setThreadNamePrefix("myExecutor");

        return taskExecutor;
    }


}

Configuration 에서 사용할 TaskExecutor의 property를 설정해 줄 수 있다.
제한된 ThreadPool 을 사용할 것이므로 ThreadPoolTaskExecutor를 사용하였다. 

• CorePoolSize : 1. 기본적으로 사용될 스레드 풀 사이즈
• MaxPoolSIze :  3. queue도 전부 사용중일 때 추가요청이 올 시 사용되는 스레드 풀 사이즈
• QueueCapacity : 2. 코어 스레드가 전부 사용중일 때 작업(task)을 대기시켜두는 queue의 사이즈

메서드를 호출해 주면 설정해둔 ThreadNamePrefix의 이름의 스레드가 비동기로 동작하게 된다.

---

 

---

2. 비동기 커스텀

1에서 설정해 준 AsyncConfig 작성하는 방법으로 비동기를 구현해 놓고 테스트를 하던 와중 DB값을 못 받아오는 오류가 발생했다. DB Schema 를 ThreadLocal 에 저장해 두었는데 비동기 스레드로 옮겨서 실행하게 되면서 값을 불러오지 못하게 된 것. ThreadLocal 참조

그래서 TaskDecorator 인터페이스를 TaskDecoratorImpl 클래스로 구현해주어 이전 스레드에 있던 DB정보를 새 작업스레드에 복사할 수 있도록 했다. 

import lombok.extern.slf4j.Slf4j;
import org.springframework.core.task.TaskDecorator;
import org.springframework.stereotype.Component;
@Slf4j
@Component
public class TaskDecoratorImpl implements TaskDecorator {
    @Override
    public Runnable decorate(Runnable runnable) {   //ThreadLocal 정보 비동기스레드에 복사
        String schema = DBConfig.getDBSchema();
        return ()-> {
            DBConfig.addDBSchema(schema);
            log.info("AsyncThread - DBConfig: "+DBConfig.getDBSchema());
            runnable.run();
        };
    }
}

---