com.tacitknowledge.slowlight.proxyserver.handler.DelayChannelHandler 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.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelPromise;

import java.util.concurrent.TimeUnit;

/**
 * DelayChannelHandler class allows to control the size and delay between messages sent over the channel.

 * 

 * Handler could be configured from two perspectives:

 * 1. maxDataSize - defines the maximum size of data fragment that must be written by this handler

 * 2. delay - defines the delay which must be applied between data fragments

 * 

 * Note: both handler parameters described above are independent and could be set to 0
 * what will cause that particular parameter to be disabled.
 *
 * @author Alexandr Donciu ([email protected])
 */
public class DelayChannelHandler extends AbstractChannelHandler
{
    protected static final String PARAM_MAX_DATA_SIZE = "maxDataSize";
    protected static final String PARAM_DELAY = "delay";

    public DelayChannelHandler(final HandlerConfig handlerConfig)
    {
        super(handlerConfig);
    }

    @Override
    public void write(final ChannelHandlerContext ctx, final Object msg, final ChannelPromise promise) throws Exception
    {
        final int maxDataSize = handlerParams.getInt(PARAM_MAX_DATA_SIZE);
        final long delay = handlerParams.getLong(PARAM_DELAY);

        ctx.executor().schedule(new WriteDataFragmentsTask(ctx, (ByteBuf) msg, promise, maxDataSize, delay), delay, TimeUnit.MILLISECONDS);
    }

    @Override
    protected void populateHandlerParams()
    {
        handlerParams.setProperty(PARAM_MAX_DATA_SIZE, handlerConfig.getParam(PARAM_MAX_DATA_SIZE));
        handlerParams.setProperty(PARAM_DELAY, handlerConfig.getParam(PARAM_DELAY));
    }
}