cloudeventbus.cli.Server.java Source code

Introduction

Here is the source code for cloudeventbus.cli.Server.java

Source

/*
 *   Copyright (c) 2013 Mike Heath.  All rights reserved.
 *
 *   Licensed 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 cloudeventbus.cli;

import cloudeventbus.pki.CertificateChain;
import cloudeventbus.pki.CertificateUtils;
import cloudeventbus.pki.TrustStore;
import cloudeventbus.server.ClusterManager;
import cloudeventbus.server.GlobalHub;
import cloudeventbus.server.ServerChannelInitializer;
import cloudeventbus.server.ServerConfig;
import com.beust.jcommander.JCommander;
import com.beust.jcommander.Parameter;
import com.beust.jcommander.ParameterException;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;

import java.net.InetSocketAddress;
import java.security.PrivateKey;
import java.util.List;

/**
 * @author Mike Heath <elcapo@gmail.com>
 */
public class Server {
    public static void main(String[] args) throws Exception {
        final JCommander commander = new JCommander();

        final Options options = new Options();
        commander.addObject(options);

        commander.setProgramName("eventbus-server");

        try {
            commander.parse(args);

            DefaultOptions.setLogLevel(options);

            TrustStore trustStore = CertificateUtils.loadTrustStore(options.trustStore);
            if (trustStore != null && trustStore.size() > 0) {
                System.out.println("Using trust store at: " + options.trustStore);
            } else {
                trustStore = null;
            }

            final int port = options.port;

            final CertificateChain certificateChain;
            final PrivateKey privateKey;
            if (options.certificate == null) {
                certificateChain = null;
                privateKey = null;
            } else {
                certificateChain = CertificateUtils.loadCertificateChain(options.certificate);
                System.out.println("Using certificate at: " + options.certificate);
                if (options.privateKey == null) {
                    System.err.print("You must specify a private key when using a certificate.");
                    System.exit(1);
                    return;
                } else {
                    privateKey = CertificateUtils.loadPrivateKey(options.privateKey);
                    System.out.println("Using private key at: " + options.privateKey);
                }
            }

            final NioEventLoopGroup parentGroup = new NioEventLoopGroup();
            final GlobalHub globalHub = new GlobalHub();
            final ServerConfig serverConfig = new ServerConfig(port, "cloudeventbus-simple-server", trustStore,
                    certificateChain, privateKey);
            final ClusterManager clusterManager = new ClusterManager(serverConfig, globalHub, parentGroup);
            if (options.peers != null) {
                for (String peer : options.peers) {
                    final String[] parts = peer.split(":");
                    if (parts.length != 2) {
                        throw new IllegalArgumentException(
                                "Invalid peer address " + peer + " should be HOST:PORT (e.g. 127.0.0.1:4223)");
                    }
                    clusterManager.registerPeer(new InetSocketAddress(parts[0], Integer.valueOf(parts[1])));
                }
            }
            new ServerBootstrap().group(parentGroup, new NioEventLoopGroup()).channel(NioServerSocketChannel.class)
                    .localAddress(new InetSocketAddress(port))
                    .childHandler(new ServerChannelInitializer(serverConfig, clusterManager, globalHub)).bind()
                    .awaitUninterruptibly();
            System.out.println("Server listening on port " + port);
        } catch (ParameterException e) {
            System.err.println(e.getMessage());
            commander.usage();
            System.exit(1);
        }
    }

    private static class Options extends DefaultOptions {

        @Parameter(names = "-port", description = "The port the server will listen on.")
        int port = 4223;

        @Parameter(names = "-certificate", description = "The file containing the certificate used to identify the server.")
        String certificate;

        @Parameter(names = "-privateKey", description = "The file containing the private key for the server's certificate.")
        String privateKey;

        @Parameter(names = "-peer", description = "A peer server to cluster with (e.g. -peer 10.1.2.3:4223)")
        List<String> peers;
    }
}