Java tutorial
/* Copyright 1995-2015 Esri 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. For additional information, contact: Environmental Systems Research Institute, Inc. Attn: Contracts Dept 380 New York Street Redlands, California, USA 92373 email: contracts@esri.com */ package com.esri.geoevent.test.performance; import java.util.Comparator; import org.apache.commons.cli.BasicParser; import org.apache.commons.cli.CommandLine; import org.apache.commons.cli.CommandLineParser; import org.apache.commons.cli.HelpFormatter; import org.apache.commons.cli.Option; import org.apache.commons.cli.OptionBuilder; import org.apache.commons.cli.Options; import org.apache.commons.cli.ParseException; import org.apache.commons.lang3.math.NumberUtils; import com.esri.geoevent.test.performance.activemq.ActiveMQEventConsumer; import com.esri.geoevent.test.performance.azure.AzureIoTHubConsumer; import com.esri.geoevent.test.performance.bds.BdsEventConsumer; import com.esri.geoevent.test.performance.kafka.KafkaEventConsumer; import com.esri.geoevent.test.performance.rabbitmq.RabbitMQEventConsumer; import com.esri.geoevent.test.performance.streamservice.StreamServiceEventConsumer; import com.esri.geoevent.test.performance.tcp.TcpServerEventConsumer; import com.esri.geoevent.test.performance.tcp.TcpEventConsumer; import com.esri.geoevent.test.performance.ui.ConsumerUI; import com.esri.geoevent.test.performance.websocket.WebsocketEventConsumer; import com.esri.geoevent.test.performance.websocket.WebsocketServerEventConsumer; /* This App listens for commands on specified port from the Ochestrator and receives messages to the target server. */ public class ConsumerMain { public static boolean DEBUG = false; //------------------------------------------------------- // Constructor // ------------------------------------------------------ public ConsumerMain() { } /** * Main method - this is used to when running from command line * * @param args Command Line Parameters */ @SuppressWarnings("static-access") public static void main(String[] args) { // performer options Options performerOptions = new Options(); performerOptions .addOption(OptionBuilder .withLongOpt("type").withDescription("One of the following values: [" + Protocol.getAllowableValues() + "]. (Default value is tcp).") .hasArg().create("t")); performerOptions.addOption(OptionBuilder.withLongOpt("commandListenerPort").withDescription( "The TCP Port where consumer will listen for commands from the orchestrator. (Default value is 5010).") .hasArg().create("p")); performerOptions.addOption(OptionBuilder.withLongOpt("serverPort") .withDescription("The TCP Port where the server will listen for events. (Default value is 5675)") .hasArg().create("sp")); performerOptions.addOption("h", "help", false, "print the help message"); // parse the command line CommandLineParser parser = new BasicParser(); CommandLine cmd; // parse try { cmd = parser.parse(performerOptions, args, false); } catch (ParseException error) { printHelp(performerOptions); return; } // User requested Help if (cmd.hasOption("h")) { printHelp(performerOptions); return; } if (cmd.getOptions().length == 0) { ConsumerUI ui = new ConsumerUI(); ui.run(); } else { // parse out the performer options String protocolValue = cmd.getOptionValue("t"); String commandListenerPortValue = cmd.getOptionValue("p"); if (protocolValue == null) { protocolValue = "tcp"; } if (commandListenerPortValue == null) { commandListenerPortValue = "5020"; } // validate if (!validateTestHarnessOptions(protocolValue, commandListenerPortValue)) { printHelp(performerOptions); return; } // parse the values Protocol protocol = Protocol.fromValue(protocolValue); boolean isLocal = "local".equalsIgnoreCase(commandListenerPortValue); int commandListenerPort = -1; if (!isLocal) { commandListenerPort = Integer.parseInt(commandListenerPortValue); } int serverPort = NumberUtils.toInt(cmd.getOptionValue("sp"), 5775); PerformanceCollector consumer; switch (protocol) { case TCP: consumer = new TcpEventConsumer(); break; case TCP_SERVER: consumer = new TcpServerEventConsumer(serverPort); break; case WEBSOCKETS: consumer = new WebsocketEventConsumer(); break; case WEBSOCKET_SERVER: consumer = new WebsocketServerEventConsumer(serverPort); break; case ACTIVE_MQ: consumer = new ActiveMQEventConsumer(); break; case RABBIT_MQ: consumer = new RabbitMQEventConsumer(); break; case STREAM_SERVICE: consumer = new StreamServiceEventConsumer(); break; case KAFKA: consumer = new KafkaEventConsumer(); break; case BDS: consumer = new BdsEventConsumer(); break; case AZURE: consumer = new AzureIoTHubConsumer(); break; default: return; } consumer.listenOnCommandPort((isLocal ? 5020 : commandListenerPort), true); } } // ------------------------------------------------------------------------------- // Main Helper Methods // ------------------------------------------------------------------------------- private static void printHelp(Options performerOptions) { HelpFormatter formatter = new HelpFormatter(); formatter.setLongOptPrefix("-"); formatter.setArgName("value"); formatter.setWidth(100); // do not sort the options in any order formatter.setOptionComparator(new Comparator<Option>() { @Override public int compare(Option o1, Option o2) { return 0; } }); formatter.printHelp(ImplMessages.getMessage("CONSUMER_EXECUTOR_HELP_TITLE_MSG"), performerOptions, true); System.out.println(""); } private static boolean validateTestHarnessOptions(String protocolStr, String commandListenerPort) { Protocol protocol = Protocol.fromValue(protocolStr); if (protocol == Protocol.UNKNOWN) { System.err.println(ImplMessages.getMessage("TEST_HARNESS_EXECUTOR_PROTOCOL_VALIDATION", protocolStr, Protocol.getAllowableValues())); return false; } // validate the port - it could be set to local if (!"local".equalsIgnoreCase(commandListenerPort)) { try { Integer.parseInt(commandListenerPort); } catch (NumberFormatException error) { System.err.println(ImplMessages.getMessage("TEST_HARNESS_EXECUTOR_COMMAND_PORT_VALIDATION", String.valueOf(commandListenerPort))); return false; } } return true; } }