UdpServerPipelineFactory.java

package com.outbrain.gruffalo.netty;

import org.springframework.util.Assert;

import io.netty.channel.Channel;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.handler.timeout.IdleStateHandler;
import io.netty.util.concurrent.EventExecutorGroup;

class UdpServerPipelineFactory extends ChannelInitializer<Channel> {

  private final int readerIdleTimeSeconds;
  private final MetricBatcherFactory metricBatcherFactory;
  private final DatagramPacketToStringDecoder datagramPacketDecoder;
  private final MetricPublishHandler publishHandler;
  private final EventExecutorGroup publishExecutor; // TODO remove - unused

  public UdpServerPipelineFactory(final int readerIdleTimeSeconds, final DatagramPacketToStringDecoder datagramPacketDecoder,
      final MetricBatcherFactory metricBatcherFactory, final MetricPublishHandler publishHandler, final EventExecutorGroup publishExecutor) {
    Assert.notNull(datagramPacketDecoder, "datagramPacketDecoder may not be null");
    Assert.notNull(metricBatcherFactory, "metricBatcherFactory may not be null");
    Assert.notNull(publishHandler, "publishHandler may not be null");
    Assert.notNull(publishExecutor, "publishExecutor may not be null");

    this.readerIdleTimeSeconds = readerIdleTimeSeconds;
    this.metricBatcherFactory = metricBatcherFactory;
    this.datagramPacketDecoder = datagramPacketDecoder;
    this.publishHandler = publishHandler;
    this.publishExecutor = publishExecutor;
  }

  @Override
  protected void initChannel(final Channel channel) throws Exception {
    final ChannelPipeline pipeline = channel.pipeline();
    pipeline.addLast("idleStateHandler", new IdleStateHandler(readerIdleTimeSeconds, 0, 0));
    pipeline.addLast("decoder", datagramPacketDecoder);
    pipeline.addLast("batchHandler", metricBatcherFactory.getMetricBatcher());
    pipeline.addLast("publishHandler", publishHandler);
  }

}