com.tacitknowledge.slowlight.proxyserver.server.proxy.ProxyTargetChannelHandler 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.server.proxy;

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

/**
 * * This handler class provides the implementation of proxy channel handler - client side.
 *
 * @author Alexandr Donciu ([email protected])
 */
class ProxyTargetChannelHandler extends AbstractChannelHandler
{
    private Channel sourceChannel;

    public ProxyTargetChannelHandler(final HandlerConfig handlerConfig, final Channel sourceChannel)
    {
        super(handlerConfig);

        this.sourceChannel = sourceChannel;
    }

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

    @Override
    public void channelRead(final ChannelHandlerContext ctx, final Object msg) throws Exception
    {
        if (sourceChannel.isActive())
        {
            final ChannelFutureListener channelFutureListener = getChannelFutureListener(ctx);
            sourceChannel.writeAndFlush(msg).addListener(channelFutureListener);
        }
    }

    @Override
    public void channelInactive(final ChannelHandlerContext ctx) throws Exception
    {
        closeOnFlush(sourceChannel);
    }

    protected Channel getSourceChannel()
    {
        return sourceChannel;
    }

    protected ChannelFutureListener getChannelFutureListener(final ChannelHandlerContext ctx)
    {
        return new ChannelFutureListener()
        {
            @Override
            public void operationComplete(ChannelFuture future) throws Exception
            {
                if (future.isSuccess())
                {
                    ctx.channel().read();
                }
                else
                {
                    future.channel().close();
                }
            }
        };
    }
}