net.yudichev.jiotty.common.lang.DelayedExecutors Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jiotty-common Show documentation

There is a newer version: 2.3.0

package net.yudichev.jiotty.common.lang;

import javax.annotation.Nonnull;
import java.util.concurrent.*;

final class DelayedExecutors {
    private DelayedExecutors() {
    }

    static Executor delayedExecutor(long millis) {
        return new DelayedExecutor(millis, TimeUnit.MILLISECONDS, ForkJoinPool.commonPool());
    }

    /**
     * Singleton delay scheduler, used only for starting and
     * cancelling tasks.
     */
    static final class Delayer {
        static final ScheduledThreadPoolExecutor delayer;

        static {
            delayer = new ScheduledThreadPoolExecutor(1, new DaemonThreadFactory());
            delayer.setRemoveOnCancelPolicy(true);
        }

        private Delayer() {
        }

        static void delay(Runnable command, long delay, TimeUnit unit) {
            delayer.schedule(command, delay, unit);
        }

        static final class DaemonThreadFactory implements ThreadFactory {
            @Override
            public Thread newThread(@Nonnull Runnable r) {
                Thread t = new Thread(r);
                t.setDaemon(true);
                t.setName("CompletableFutureDelayScheduler");
                return t;
            }
        }
    }

    private static final class DelayedExecutor implements Executor {
        private final long delay;
        private final TimeUnit unit;
        private final Executor executor;

        DelayedExecutor(long delay, TimeUnit unit, Executor executor) {
            this.delay = delay;
            this.unit = unit;
            this.executor = executor;
        }

        @Override
        public void execute(@Nonnull Runnable command) {
            Delayer.delay(new TaskSubmitter(executor, command), delay, unit);
        }
    }

    /**
     * Action to submit user task
     */
    private static final class TaskSubmitter implements Runnable {
        private final Executor executor;
        private final Runnable action;

        TaskSubmitter(Executor executor, Runnable action) {
            this.executor = executor;
            this.action = action;
        }

        @Override
        public void run() {
            executor.execute(action);
        }
    }
}