com.codebullets.external.party.simulator.connections.websocket.inbound.WebSocketServerInitializer.java Source code

Introduction

Here is the source code for com.codebullets.external.party.simulator.connections.websocket.inbound.WebSocketServerInitializer.java

Source

/*
 * Copyright 2012 The Netty Project
 *
 * The Netty Project licenses this file to you under the Apache License,
 * version 2.0 (the "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at:
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */
package com.codebullets.external.party.simulator.connections.websocket.inbound;

import com.codebullets.external.party.simulator.connections.ConnectionConfig;
import com.codebullets.external.party.simulator.connections.ConnectionMonitor;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.timeout.ReadTimeoutHandler;

import java.net.URI;
import java.util.concurrent.TimeUnit;

/**
 */
public class WebSocketServerInitializer extends ChannelInitializer<SocketChannel> {
    private static final long DEFAULT_TIMEOUT = TimeUnit.SECONDS.toMillis(30);

    private final URI endpoint;
    private final ConnectionMonitor connectionMonitor;
    private final ConnectionConfig connectionConfig;
    private final ChannelGroup connectedChannels;

    /**
     * Generates a new instance of WebSocketServerInitializer.
     */
    public WebSocketServerInitializer(final URI endpoint, final ConnectionMonitor connectionMonitor,
            final ConnectionConfig connectionConfig, final ChannelGroup connectedChannels) {
        this.endpoint = endpoint;
        this.connectionMonitor = connectionMonitor;
        this.connectionConfig = connectionConfig;
        this.connectedChannels = connectedChannels;
    }

    @Override
    public void initChannel(final SocketChannel ch) throws Exception {
        long timeoutVal = connectionConfig.getTimeout() > 0 ? connectionConfig.getTimeout() : DEFAULT_TIMEOUT;

        ChannelPipeline pipeline = ch.pipeline();
        pipeline.addLast("codec-http", new HttpServerCodec());
        pipeline.addLast("readTimeoutHandler", new ReadTimeoutHandler(timeoutVal, TimeUnit.MILLISECONDS));
        pipeline.addLast("aggregator", new HttpObjectAggregator(Integer.MAX_VALUE));
        pipeline.addLast("handler", new NettyWebSocketServerHandler(endpoint, connectionMonitor,
                connectionConfig.getName(), connectedChannels));
    }
}