observeOutputIdle() — netty Function Reference

Source Code

handler/src/test/java/io/netty/handler/timeout/IdleStateHandlerTest.java lines 238–364

    private static void observeOutputIdle(boolean writer) throws Exception {

        long writerIdleTime = 0L;
        long allIdleTime = 0L;
        IdleStateEvent expected;

        if (writer) {
            writerIdleTime = 5L;
            expected = IdleStateEvent.FIRST_WRITER_IDLE_STATE_EVENT;
        } else {
            allIdleTime = 5L;
            expected = IdleStateEvent.FIRST_ALL_IDLE_STATE_EVENT;
        }

        IdleStateHandler idleStateHandler = new IdleStateHandler(
                true, 0L, writerIdleTime, allIdleTime, TimeUnit.SECONDS);

        final List<Object> events = new ArrayList<Object>();
        ChannelInboundHandlerAdapter handler = new ChannelInboundHandlerAdapter() {
            @Override
            public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
                events.add(evt);
            }
        };

        ObservableChannel channel = new ObservableChannel(idleStateHandler, handler);
        channel.freezeTime();
        try {
            // We're writing 3 messages that will be consumed at different rates!
            channel.writeAndFlush(Unpooled.wrappedBuffer(new byte[] { 1 }));
            channel.writeAndFlush(Unpooled.wrappedBuffer(new byte[] { 2 }));
            channel.writeAndFlush(Unpooled.wrappedBuffer(new byte[] { 3 }));
            channel.writeAndFlush(Unpooled.wrappedBuffer(new byte[5 * 1024]));

            // Establish a baseline. We're not consuming anything and let it idle once.
            channel.advanceTimeBy(5, TimeUnit.SECONDS);
            channel.runPendingTasks();
            assertEquals(1, events.size());
            assertSame(expected, events.get(0));
            events.clear();

            // Consume one message in 4 seconds, then be idle for 2 seconds,
            // then run the task and we shouldn't get an IdleStateEvent because
            // we haven't been idle for long enough!
            channel.advanceTimeBy(4, TimeUnit.SECONDS);
            channel.runPendingTasks();
            assertNotNullAndRelease(channel.consume());

            channel.advanceTimeBy(2, TimeUnit.SECONDS);
            channel.runPendingTasks();
            assertEquals(0, events.size());

            // Consume one message in 3 seconds, then be idle for 4 seconds,
            // then run the task and we shouldn't get an IdleStateEvent because
            // we haven't been idle for long enough!
            channel.advanceTimeBy(2, TimeUnit.SECONDS);
            channel.runPendingTasks();
            assertNotNullAndRelease(channel.consume());

            channel.advanceTimeBy(4, TimeUnit.SECONDS);
            channel.runPendingTasks();
            assertEquals(0, events.size());

            // Don't consume a message and be idle for 5 seconds.
            // We should get an IdleStateEvent!
            channel.advanceTimeBy(5, TimeUnit.SECONDS);
            channel.runPendingTasks();
            assertEquals(1, events.size());
            events.clear();

            // Consume one message in 2 seconds, then be idle for 1 seconds,
            // then run the task and we shouldn't get an IdleStateEvent because
            // we haven't been idle for long enough!
            channel.advanceTimeBy(2, TimeUnit.SECONDS);
            channel.runPendingTasks();
            assertNotNullAndRelease(channel.consume());

            channel.advanceTimeBy(1, TimeUnit.SECONDS);
            channel.runPendingTasks();
            assertEquals(0, events.size());