25 lines
880 B
Java
25 lines
880 B
Java
package com.nanri.aiimage.config;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.context.annotation.Configuration;
|
|
import org.springframework.scheduling.TaskScheduler;
|
|
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
|
|
|
|
@Configuration
|
|
@Slf4j
|
|
public class SchedulingConfig {
|
|
|
|
@Bean
|
|
public TaskScheduler taskScheduler() {
|
|
ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
|
|
scheduler.setPoolSize(4);
|
|
scheduler.setThreadNamePrefix("aiimage-scheduling-");
|
|
scheduler.setWaitForTasksToCompleteOnShutdown(true);
|
|
scheduler.setAwaitTerminationSeconds(30);
|
|
scheduler.setErrorHandler(ex -> log.warn("[scheduling] task execution failed: {}", ex.getMessage(), ex));
|
|
scheduler.initialize();
|
|
return scheduler;
|
|
}
|
|
}
|