Subversion Repositories SmartDukaan

package com.smartdukaan.cron.config;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.springframework.context.annotation.Configuration;

import org.springframework.context.annotation.Profile;

import org.springframework.scheduling.annotation.EnableScheduling;

import org.springframework.scheduling.annotation.SchedulingConfigurer;

import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;

import org.springframework.scheduling.config.ScheduledTaskRegistrar;

@Configuration

@EnableScheduling

@Profile("scheduled")

public class SchedulerConfig implements SchedulingConfigurer {

	/**

	 * The pool size.

	 */

	private final int POOL_SIZE = 10;

	/**"prod"

	 * Configures the scheduler to allow multiple pools.

	 *

	 * @param taskRegistrar The task registrar.

	 */

	private static final Logger LOGGER = LogManager.getLogger(SchedulerConfig.class);

	@Override

	public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {

		ThreadPoolTaskScheduler threadPoolTaskScheduler = new ThreadPoolTaskScheduler();

		threadPoolTaskScheduler.setErrorHandler(t -> {

			LOGGER.info("Exception Thrown - {}", t);

		});

		threadPoolTaskScheduler.setPoolSize(POOL_SIZE);

		threadPoolTaskScheduler.setThreadNamePrefix("scheduled-task-pool-");

		threadPoolTaskScheduler.initialize();

		taskRegistrar.setTaskScheduler(threadPoolTaskScheduler);

	}

}