com.ebay.jetstream.messaging.transport.netty.eventproducer.EventConsumerStatisticsHandler.java Source code

Introduction

Here is the source code for com.ebay.jetstream.messaging.transport.netty.eventproducer.EventConsumerStatisticsHandler.java

Source

/*******************************************************************************
 *  Copyright  2012-2015 eBay Software Foundation
 *  This program is dual licensed under the MIT and Apache 2.0 licenses.
 *  Please see LICENSE for more information.
 *******************************************************************************/
package com.ebay.jetstream.messaging.transport.netty.eventproducer;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelOutboundHandlerAdapter;
import io.netty.channel.ChannelPromise;

public class EventConsumerStatisticsHandler extends ChannelOutboundHandlerAdapter {

    @Override
    public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
        ((ExtendedChannelPromise) promise).setWrittenSize(((ByteBuf) msg).readableBytes());
        super.write(ctx, msg, promise);
    }
}