StreamCreationAndTearDownHandler Class — netty Architecture

Source Code

codec-native-quic/src/test/java/io/netty/handler/codec/quic/QuicStreamChannelCloseTest.java lines 228–260

    private static final class StreamCreationAndTearDownHandler extends QuicChannelValidationHandler {
        private final QuicStreamType type;
        private final boolean halfClose;
        private final Promise<Channel> streamPromise;

        StreamCreationAndTearDownHandler(QuicStreamType type, boolean halfClose, Promise<Channel> streamPromise) {
            this.type = type;
            this.halfClose = halfClose;
            this.streamPromise = streamPromise;
        }

        @Override
        public void channelActive(ChannelHandlerContext ctx) {
            super.channelActive(ctx);
            QuicChannel channel = (QuicChannel) ctx.channel();
            channel.createStream(type, new ChannelInboundHandlerAdapter() {
                @Override
                public void channelActive(ChannelHandlerContext ctx)  {
                    final ChannelFuture future;
                    if (halfClose) {
                        future = ((QuicStreamChannel) ctx.channel()).shutdownOutput();
                    } else {
                        future = ctx.channel().close();
                    }
                    future.addListener(f -> {
                        ctx.channel().writeAndFlush("Unsupported message").addListener(wf -> {
                            streamPromise.setFailure(wf.cause());
                        });
                    });
                }
            });
        }
    }