com.payneteasy.mysql.scheduler.Scheduler Maven / Gradle / Ivy

Go to download
package com.payneteasy.mysql.scheduler;

import com.google.inject.Inject;
import com.payneteasy.mysql.scheduler.dao.model.TTask;
import com.payneteasy.mysql.scheduler.util.ExecutorServiceUtils;
import com.payneteasy.mysql.scheduler.util.TaskExecutor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.List;

public class Scheduler {

    private static final Logger LOG = LoggerFactory.getLogger(Scheduler.class);
    private static final int MAX_THREADS = SchedulerConfig.getIntConfig(SchedulerConfig.Config.MAX_THREADS) ;
    private static final int SLEEP_MS = SchedulerConfig.getIntConfig(SchedulerConfig.Config.SLEEP_MS) ;

    public void start() {
        theIsStarted = true;
         while ( theIsStarted ) {

             try {
                 runTasks();
             } catch (Exception e) {
                 LOG.error("Can't execute tasks", e);
             } finally {
                 if (theIsStarted) {
                     try {
                         Thread.sleep(SLEEP_MS);
                     } catch (InterruptedException e) {
                         LOG.warn("Sleep interrupted: "+SLEEP_MS+" ms");
                     }
                 }
             }

//             theIsStarted = false;
         }
    }

    public void stop() {
        theIsStarted = false;
        try {
            ExecutorServiceUtils.shutdownAndAwaitTermination(theTaskExecutor, "scheduler");
        } catch (InterruptedException e) {
            LOG.error("Can't stop executor", e);
        }
    }

    private void runTasks() {


        List tasks = theSchedulerService.getTasks(getAvailableThreadsCount());

        for (TTask task : tasks) {
            LOG.debug("Starting {}...", task.getTaskName());
            theSchedulerService.setTaskStarting(task.getTaskId());
            theTaskExecutor.execute(new TaskRunner(theSchedulerService, task));

        }


    }

    private int getAvailableThreadsCount() {
        return theTaskExecutor.getCorePoolSize() - theTaskExecutor.getActiveCount();
    }


    private final TaskExecutor theTaskExecutor = ExecutorServiceUtils.createTaskExecutor(MAX_THREADS);

    @Inject
    private ISchedulerService theSchedulerService;

    private volatile boolean theIsStarted = false;

}