StreamHandler Class — netty Architecture

Source Code

codec-native-quic/src/test/java/io/netty/handler/codec/quic/QuicStreamHalfClosureTest.java lines 101–155

    private static final class StreamHandler extends ChannelInboundHandlerAdapter {
        private final BlockingQueue<Integer> queue = new LinkedBlockingQueue<>();

        @Override
        public void channelRegistered(ChannelHandlerContext ctx) {
            queue.add(0);
        }

        @Override
        public void channelInactive(ChannelHandlerContext ctx) {
            queue.add(5);
            // Close the QUIC channel as well.
            ctx.channel().parent().close();
        }

        @Override
        public void channelRead(ChannelHandlerContext ctx, Object msg) {
            ReferenceCountUtil.release(msg);
            if (((QuicStreamChannel) ctx.channel()).isInputShutdown()) {
                queue.add(1);
            }
        }

        @Override
        public void userEventTriggered(ChannelHandlerContext ctx, Object evt) {
            if (evt == ChannelInputShutdownEvent.INSTANCE) {
                addIsShutdown(ctx);
                queue.add(3);
            } else if (evt == ChannelInputShutdownReadComplete.INSTANCE) {
                queue.add(4);
                ctx.close();
            }
        }

        private void addIsShutdown(ChannelHandlerContext ctx) {
            if (((QuicStreamChannel) ctx.channel()).isInputShutdown()) {
                queue.add(2);
            }
        }

        void assertSequence() throws Exception {
            assertEquals(0, (int) queue.take());
            int value = queue.take();
            if (value == 1) {
                // If we did see the value of 1 it should be followed by 2 directly.
                assertEquals(2, (int) queue.take());
            } else {
                assertEquals(2, value);
            }
            assertEquals(3, (int) queue.take());
            assertEquals(4, (int) queue.take());
            assertEquals(5, (int) queue.take());
            assertTrue(queue.isEmpty());
        }
    }