TaskManager.java

package com.osroyale.game.task;
 
import com.google.common.base.Preconditions;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
 
import java.util.ArrayList;
import java.util.List;
import java.util.ListIterator;
import java.util.Queue;
import java.util.concurrent.ConcurrentLinkedQueue;

 
* The class that handles scheduling tasks, and processing them.
 *
 * @author nshusa
 */
public final class TaskManager {

    private static final Logger logger = LogManager.getLogger(TaskManager.class);

    private final Queue<Task> pending = new ConcurrentLinkedQueue<>();

    private final List<Task> active = new ArrayList<>();

    public synchronized void schedule(Task task) {
        /*StackTraceElement[] stackTrace = Thread.currentThread().getStackTrace();
        System.out.println("Displaying Stack trace using StackTraceElement in Java");
        String myString = "";
        for(StackTraceElement st : stackTrace){
            myString = myString + st + '\n';
        }
        LogManager.getLogger(TaskManager.class).info(myString);*/
        Preconditions.checkNotNull(task);
 
        try {
            if (!task.canSchedule()) {
                return;
            }
 
            task.setRunning(true);
            task.onSchedule();
 
            if (task.isInstant()) {
                try {
                    task.baseExecute();
                } catch (Exception ex) {
                    logger.warn(String.format("error executing task: %s", task.getClass().getSimpleName()), ex);
                    return;
                }
            }
            pending.add(task);
        } catch (Exception ex) {
            logger.error(String.format("error scheduling task: %s", task.getClass().getSimpleName()), ex);
        }
    }

    public synchronized void processTasks() {
        Task t;
        while((t = pending.poll()) != null) {
            active.add(t);
        }
 
        for (final ListIterator<Task> itr = active.listIterator(); itr.hasNext();) {
            final Task task = itr.next();
 
            try {
                //Start execution is called before process so that the task can get the elapsed time from the start of the task
                task.setExecutionTime();
                task.process();
            } catch (Exception ex) {
                logger.warn(String.format("Error executing task: %s [pendingQueue: %d executionQueue: %d], ", task.getClass().getSimpleName(), pending.size(), active.size()), ex);
                task.cancel();
            }
 
            if (!task.isRunning()) {
                itr.remove();
            }
        }
    }

    public void cancel(Object attachment) {
        cancel(attachment, false);
    }

    public void cancel(Object attachment, boolean logout) {
        for (Task task : pending) {
            if (attachment.equals(task.getAttachment().orElse(null))) {
                task.cancel(logout);
            }
        }
    }
 
    public List<Task> getTasks() {
        return active;
    }
}