ReRegisterHandler Class — netty Architecture

Source Code

transport/src/test/java/io/netty/channel/AbstractEventLoopTest.java lines 164–213

    @ChannelHandler.Sharable
    static class ReRegisterHandler extends ChannelInboundHandlerAdapter {

        final EventLoopGroup eventLoopGroup;
        private final DeregisterMethod method;
        private final AtomicReference<Throwable> throwable;

        ReRegisterHandler(EventLoopGroup eventLoopGroup,
                                 DeregisterMethod method,
                                 AtomicReference<Throwable> throwable) {
            this.eventLoopGroup = eventLoopGroup;
            this.method = method;
            this.throwable = throwable;
        }

        @Override
        public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
            super.exceptionCaught(ctx, cause);
            throwable.set(cause);
            ctx.close();
        }

        @Override
        public void channelRead(ChannelHandlerContext outCtx, Object msg) throws Exception {
            final EventLoop newLoop = anyNotEqual(outCtx.channel().eventLoop());
            ChannelFutureListener pipelieModifyingListener = register -> {
                outCtx.pipeline().addLast(new ChannelInboundHandlerAdapter() {
                    @Override
                    public void channelRead(ChannelHandlerContext inCtx, Object inMsg) {
                        outCtx.writeAndFlush(inMsg);
                    }
                });
                outCtx.fireChannelRead(msg);
            };
            ChannelFutureListener reregisteringListener = deregister -> {
                deregister.channel().pipeline().fireUserEventTriggered("Unregistered user event");
                newLoop.register(deregister.channel()).addListener(pipelieModifyingListener);
            };
            method.apply(outCtx).addListener(reregisteringListener);
        }

        private EventLoop anyNotEqual(EventLoop eventLoop) {
            EventLoop next;
            do {
                next = eventLoopGroup.next();
            } while (eventLoop == next);

            return next;
        }
    }