com.caricah.iotracah.server.netty.ServerInitializer.java Source code

Introduction

Here is the source code for com.caricah.iotracah.server.netty.ServerInitializer.java

Source

/*
 *
 * Copyright (c) 2015 Caricah <info@caricah.com>.
 *
 * Caricah 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.caricah.iotracah.server.netty;

import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.ssl.SslContext;
import io.netty.handler.ssl.SslHandler;
import io.netty.util.concurrent.DefaultEventExecutorGroup;
import io.netty.util.concurrent.EventExecutorGroup;

/**
 * @author <a href="mailto:bwire@caricah.com"> Peter Bwire </a>
 * @version 1.0 5/27/15
 */
public abstract class ServerInitializer<T> extends ChannelInitializer<SocketChannel> {

    private final int connectionTimeout;
    private final SslContext sslContext;
    private final ServerImpl serverImpl;
    private EventExecutorGroup iotEventExecutorGroup;

    public ServerInitializer(ServerImpl serverImpl, int connectionTimeout, SSLHandler sslHandler) {
        this.serverImpl = serverImpl;
        this.sslContext = sslHandler.getSslContext();
        this.connectionTimeout = connectionTimeout;

        int countOfAvailableProcessors = Runtime.getRuntime().availableProcessors() + 1;

        this.iotEventExecutorGroup = new DefaultEventExecutorGroup(countOfAvailableProcessors,
                getServerImpl().getExecutorService());

    }

    public ServerInitializer(ServerImpl serverImpl, int connectionTimeout) {
        this.serverImpl = serverImpl;
        this.sslContext = null;
        this.connectionTimeout = connectionTimeout;
    }

    public ServerImpl<T> getServerImpl() {
        return serverImpl;
    }

    public EventExecutorGroup getIotEventExecutorGroup() {
        return iotEventExecutorGroup;
    }

    public SslContext getSslContext() {
        return sslContext;
    }

    public int getConnectionTimeout() {
        return connectionTimeout;
    }

    /**
     * This method will be called once the {@link SocketChannel} was registered. After the method returns this instance
     * will be removed from the {@link ChannelPipeline} of the {@link SocketChannel}.
     *
     * @param ch the {@link SocketChannel} which was registered.
     * @throws Exception is thrown if an error occurs. In that case the {@link SocketChannel} will be closed.
     */
    @Override
    protected void initChannel(SocketChannel ch) throws Exception {

        ChannelPipeline pipeline = ch.pipeline();

        if (null != getSslContext()) {
            // Add SSL handler first to encrypt and decrypt everything.
            // In this application ssl is only used for transport encryption
            // Identification is not yet part of the deal.

            pipeline.addLast("ssl", getSslContext().newHandler(ch.alloc()));
        }

        customizePipeline(getIotEventExecutorGroup(), pipeline);

    }

    protected abstract void customizePipeline(EventExecutorGroup eventExecutorGroup, ChannelPipeline pipeline);

}