Examples with TaskWrapper com.serotonin.timer.TaskWrapper used on opensource projects

Example 1 with TaskWrapper

use of com.serotonin.timer.TaskWrapper in project ma-core-public by infiniteautomation.

the class TaskRejectionHandler method rejectedExecution.

/*
	 * This will be called by the Executor when Runnable's are not executable due to pool constraints
	 * (non-Javadoc)
	 * @see java.util.concurrent.RejectedExecutionHandler#rejectedExecution(java.lang.Runnable, java.util.concurrent.ThreadPoolExecutor)
	 */
@Override
public void rejectedExecution(Runnable r, ThreadPoolExecutor e) {
    if (r instanceof TaskWrapper) {
        TaskWrapper wrapper = (TaskWrapper) r;
        RejectedTaskReason reason = new RejectedTaskReason(RejectedTaskReason.POOL_FULL, wrapper.getExecutionTime(), wrapper.getTask(), e);
        wrapper.getTask().rejected(reason);
        this.rejectedTask(reason);
    } else if (r instanceof OrderedTaskCollection) {
        // Pool must be full since the entire collection was rejected
        TaskWrapper wrapper = ((OrderedTaskCollection) r).getWrapper();
        RejectedTaskReason reason = new RejectedTaskReason(RejectedTaskReason.POOL_FULL, wrapper.getExecutionTime(), wrapper.getTask(), e);
        wrapper.getTask().rejected(reason);
        this.rejectedTask(reason);
    } else {
        log.fatal("SHOULD NOT HAPPEN: " + r.toString());
    }
}

Example 2 with TaskWrapper

use of com.serotonin.timer.TaskWrapper in project ma-core-public by infiniteautomation.

the class OrderedThreadPoolExecutorTest method testFailedExecutions.

@Test
public void testFailedExecutions() throws InterruptedException {
    boolean flushOnReject = false;
    OrderedThreadPoolExecutor exe = new OrderedThreadPoolExecutor(0, 3, 60L, TimeUnit.SECONDS, new LinkedBlockingQueue<Runnable>(), new MangoThreadFactory("medium", Thread.MAX_PRIORITY - 2), new RejectedExecutionHandler() {

        @Override
        public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
            System.out.println("Rejected.");
        }
    }, flushOnReject, new SystemTimeSource());
    // Starup a new thread that inserts failing tasks
    new Thread() {

        public void run() {
            long time = 10000;
            for (int i = 0; i < 10; i++) {
                Task task = new Task("Failure", "TSK_FAIL", -1) {

                    @Override
                    public void run(long runtime) {
                        try {
                            Thread.sleep(20);
                        } catch (InterruptedException e) {
                        }
                        throw new RuntimeException("oops");
                    }

                    @Override
                    public void rejected(RejectedTaskReason reason) {
                        System.out.println("Task Rejected");
                    }
                };
                exe.execute(new TaskWrapper(task, time));
                time += 100;
            }
        }
    }.start();
    Thread.sleep(1000);
    if (exe.queueExists("TSK_FAIL"))
        fail("non empty queue");
}