TestableIoEventLoop Class — netty Architecture

Source Code

transport/src/test/java/io/netty/channel/MultiThreadIoEventLoopGroupTest.java lines 90–127

    private static class TestableIoEventLoop extends SingleThreadIoEventLoop {
        private final AtomicBoolean simulateWorkload = new AtomicBoolean(false);

        TestableIoEventLoop(IoEventLoopGroup parent, Executor executor, IoHandlerFactory ioHandlerFactory) {
            super(parent, executor, ioHandlerFactory);
        }

        public void setSimulateWorkload(boolean active) {
            simulateWorkload.set(active);
            if (active) {
                execute(() -> { /* NO-OP */ });
            }
        }

        @Override
        protected void run() {
            do {
                if (simulateWorkload.get()) {
                    try {
                        Thread.sleep(100);
                    } catch (InterruptedException e) {
                        Thread.currentThread().interrupt();
                    }
                    reportActiveIoTime(TimeUnit.MILLISECONDS.toNanos(100));
                } else {
                    Runnable task = takeTask();
                    if (task != null) {
                        safeExecute(task);
                    }
                }
            } while (!confirmShutdown() && !canSuspend());
        }

        @Override
        protected Queue<Runnable> newTaskQueue(int maxPendingTasks) {
            return new LinkedBlockingQueue<>(maxPendingTasks);
        }
    }