@Scheduled 사용할 때 스레드 설정

개요

스프링 프레임워크에서 제공하는 @Scheduled 어노테이션은 메서드에 스케줄링 기능을 부여하는 데 사용됩니다. 기본적으로 @Scheduled 어노테이션만을 사용하면 스프링은 단일 스레드에서 스케줄링 작업들을 순차적으로 처리하는데, 이는 하나의 스케줄링 작업이 완료되어야만 다음 스케줄링 작업이 실행될 수 있다는 것을 의미합니다.

따라서 부가적인 설정 없이 여러 개의 스케줄러를 작성하면, 일부 작업이 예상치 못한 시간에 동작할 수 있습니다.

예시 상황을 한번 보겠습니다.

@Configuration
@EnableScheduling
public class ScheduledTasks {
    private final Logger log = LoggerFactory.getLogger(ScheduledTasks.class);

    @Scheduled(fixedRate = 1000) // 1초마다 taskA를 실행합니다.
    public void taskA() throws InterruptedException {
        Thread.sleep(10000); // 10초 동안 일시 중단합니다.
        log.info("taskA - {} - {}", LocalDateTime.now(), Thread.currentThread().getName());
    }

    @Scheduled(fixedRate = 1000) // 1초마다 taskB를 실행합니다.
    public void taskB() {
        // taskB의 로직을 실행합니다.
        log.info("taskB - {} - {}", LocalDateTime.now(), Thread.currentThread().getName());
    }

실행 결과

위의 예시에서 taskA와 taskB는 각각 1초마다 실행되도록 설정되어 있습니다. 그러나 예상과 달리 실제로 로그를 확인해 보면, taskB가 10초에 한 번만 실행되는 것을 확인할 수 있습니다.

이는 @Scheduled 어노테이션의 기본 설정이 단일 스레드에서 작업을 수행하기 때문입니다. 이에 따라, taskA와 taskB는 동일한 스레드에서 순차적으로 실행되며, taskA는 10초 동안 일시 중단되므로, taskB는 taskA의 수행이 끝날 때까지 기다려야 합니다. 실제로 로그에서 'scheduling-1'이라는 동일한 스레드에서 두 작업이 실행되는 것을 확인할 수 있습니다.

이 문제를 해결하기 위해서는 부가적인 설정이 필요한데, 크게 다음과 같이 세 가지 방법이 존재합니다.

1. Async 처리
2. SchedulingConfigurer 구현
3. TaskScheduler 빈 등록

 
 

1. Async 처리

스프링의 @Async 어노테이션을 사용하면 메서드가 별도의 스레드에서 비동기적으로 실행됩니다. 이를 통해 여러 작업들을 동시에 처리할 수 있게 됩니다.

@Configuration
@EnableAsync
@EnableScheduling
public class ScheduledTasks {
    private final Logger log = LoggerFactory.getLogger(ScheduledTasks.class);

    @Async
    @Scheduled(fixedRate = 1000)
    public void taskA() throws InterruptedException {
        Thread.sleep(10000);
        log.info("taskA - {} - {}", LocalDateTime.now(), Thread.currentThread().getName());
    }

    @Async
    @Scheduled(fixedRate = 1000)
    public void taskB() {
        log.info("taskB - {} - {}", LocalDateTime.now(), Thread.currentThread().getName());
    }
}

실행 결과

위의 코드에서, taskA와 taskB는 각각 1초마다 별도의 스레드에서 비동기로 실행되도록 설정되어 있습니다. 이렇게 설정하면 taskA의 실행이 taskB에 영향을 주지 않게 됩니다. 실행 로그를 확인해 보면, 실제로 taskA와 taskB가 각각 1초마다 독립적으로 동작하는 것을 확인할 수 있습니다.

이는 @Async 어노테이션의 기본 설정이 SimpleAsyncTaskExecutor를 사용하기 때문입니다. 특히 taskA의 경우 이전 설정에서는 10초 동안 일시 중단되었지만, 이번 설정에서는 1초마다 새로운 스레드에서 독립적으로 수행되는 것을 확인할 수 있습니다. 이렇게 비동기로 수행하면 별도의 스레드에서 독립적으로 작업을 수행하므로, 한 작업의 지연이 다른 작업에 영향을 미치는 것을 방지할 수 있습니다.
 
 

2. SchedulingConfigurer 구현

SchedulingConfigurer 인터페이스를 구현하면, 스케줄러의 동작을 더욱 세부적으로 제어할 수 있습니다. 이를 통해 스케줄러가 사용하는 스레드 풀의 크기를 조정하거나 스레드의 이름을 변경하는 등의 작업을 수행할 수 있습니다.

@Configuration
@EnableScheduling
public class SchedulerConfig implements SchedulingConfigurer {
    @Override
    public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {
        ThreadPoolTaskScheduler threadPoolTaskScheduler = new ThreadPoolTaskScheduler();

        threadPoolTaskScheduler.setPoolSize(10);
        threadPoolTaskScheduler.setThreadNamePrefix("my-scheduler-");
        threadPoolTaskScheduler.initialize();

        taskRegistrar.setTaskScheduler(threadPoolTaskScheduler);
    }
}

위 예시에서는 SchedulingConfigurer 인터페이스를 구현하는 SchedulerConfig 클래스가 스레드 풀의 크기를 10으로 설정하고, 스레드 이름의 접두사를 "my-scheduler-"로 설정하고 있습니다. 이렇게 하면, 스케줄링 작업은 "my-scheduler-" 접두사를 가진 10개의 스레드에서 동시에 처리될 수 있게 됩니다.

실행 결과

실행 결과를 보면 taskA와 taskB가 각각 별도의 스레드에서 수행되는 것을 알 수 있습니다. 또한 비동기 처리를 사용했을 때와는 달리 taskA의 한 번의 실행이 끝나야 그다음 taskA의 실행이 시작되는 것을 확인할 수 있습니다.

이를 통해, 동일한 작업에 대해서는 순차적인 실행을 보장하면서도, 다른 작업들에 대해서는 병렬적인 실행을 가능하게 하는 스케줄링 설정을 구성할 수 있습니다.
 

 

3. TaskScheduler 빈 등록

TaskScheduler를 스프링 빈으로 등록하는 방법은 설정 방식이 간단하다는 장점이 있습니다.

이 방법도 SchedulingConfigurer을 구현하는 것과 마찬가지로 스케줄링 기능을 활성화하고, 스레드 풀의 크기와 스레드 이름의 접두사를 설정하는 것은 동일합니다. 하지만 스케줄링 설정을 스프링 컨테이너에 위임하여 자동으로 처리할 수 있습니다.

@Configuration
@EnableScheduling
public class SchedulerConfig {
    @Bean
    public TaskScheduler taskScheduler() {
        ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
        
        scheduler.setPoolSize(10);
        scheduler.setThreadNamePrefix("my-scheduler-");
        scheduler.initialize();
        
        return scheduler;
    }
}

위 코드에서는 ThreadPoolTaskScheduler를 생성하고 스레드 풀의 크기를 10으로 설정하며, 스레드 이름의 접두사를 "my-scheduler-"로 설정한 후 스프링 빈으로 등록하고 있습니다. 이 설정은 SchedulingConfigurer 인터페이스를 구현하는 방법과 동일한 결과를 제공하지만, 스프링 컨테이너가 스케줄링 설정을 자동으로 처리하므로 별도의 설정 없이도 여러 스케줄링 작업을 병렬로 처리할 수 있습니다.

여기서 "별도의 설정"이란, SchedulingConfigurer를 구현하는 방식에서 필요한 configureTasks 메서드를 직접 구현하는 등의 작업을 의미합니다. TaskScheduler 빈 등록 방식에서는 이런 추가적인 작업 없이도 스프링이 스케줄러를 자동으로 관리하게 됩니다.

실행 결과

실행 결과를 확인하면, SchedulingConfigurer를 구현했을 때와 동일한 결과를 나타내는 것을 확인할 수 있습니다.

 

 

정리

스프링의 @Scheduled 어노테이션을 사용하면 메서드에 스케줄링 기능을 부여할 수 있지만 기본 설정으로는 모든 작업이 한 스레드에서 순차적으로 처리됩니다. 이를 해결하기 위해 Async 처리, SchedulingConfigurer 구현, TaskScheduler 빈 등록 등의 방법을 사용하여 스케줄링 동작을 세부적으로 제어할 수 있습니다. 이를 통해 여러 작업들을 동시에 처리하거나, 스케줄링 설정을 자동으로 처리할 수 있습니다.