com.tacitknowledge.slowlight.proxyserver.handler.EchoChannelHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of slowlight-proxy Show documentation

Standalone proxy tool for creating issues with synchronous RPC calls

The newest version!

package com.tacitknowledge.slowlight.proxyserver.handler;

import com.tacitknowledge.slowlight.proxyserver.config.HandlerConfig;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelHandlerContext;

/**
 * Use this EchoChannelHandler whenever you what request data to be echoed back to the client.
 *
 * @author Alexandr Donciu ([email protected])
 */
public class EchoChannelHandler extends AbstractChannelHandler
{
    public EchoChannelHandler(final HandlerConfig handlerConfig)
    {
        super(handlerConfig);
    }

    @Override
    public void channelActive(final ChannelHandlerContext ctx) throws Exception
    {
        ctx.read();
    }

    @Override
    public void channelRead(final ChannelHandlerContext ctx, final Object msg) throws Exception
    {
        ctx.channel().writeAndFlush(msg).addListener(new EchoMessageListener(ctx));
    }

    /**
     * Listener class to request next channel read.
     */
    protected class EchoMessageListener implements ChannelFutureListener
    {
        private final ChannelHandlerContext ctx;

        protected EchoMessageListener(final ChannelHandlerContext ctx)
        {
            this.ctx = ctx;
        }

        @Override
        public void operationComplete(final ChannelFuture future) throws Exception
        {
            if (future.isSuccess())
            {
                ctx.channel().read();
            }
            else
            {
                future.channel().close();
            }
        }
    }
}