/* * 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; } }