--- /dev/null
+/*
+ * Copyright (c) 2016 Pantheon Technologies s.r.o. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.openflowjava.protocol.impl.clients;
+
+import static com.google.common.base.Preconditions.checkArgument;
+
+import javax.annotation.Nullable;
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.Callable;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.TimeUnit;
+
+import com.google.common.util.concurrent.FutureCallback;
+import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.common.util.concurrent.ListeningExecutorService;
+import com.google.common.util.concurrent.MoreExecutors;
+import io.netty.bootstrap.Bootstrap;
+import io.netty.channel.EventLoopGroup;
+import io.netty.channel.nio.NioEventLoopGroup;
+import net.sourceforge.argparse4j.ArgumentParsers;
+import net.sourceforge.argparse4j.annotation.Arg;
+import net.sourceforge.argparse4j.inf.ArgumentParser;
+import net.sourceforge.argparse4j.inf.ArgumentParserException;
+import org.slf4j.LoggerFactory;
+
+/**
+ * ControllerConnectionTestTool class, utilities for testing device's connect
+ * @author Jozef Bacigal
+ * Date: 4.3.2016.
+ */
+public class ControllerConnectionTestTool {
+
+ private static final org.slf4j.Logger LOG = LoggerFactory.getLogger(ControllerConnectionTestTool.class);
+
+ public static class Params {
+
+ @Arg(dest = "controller-ip")
+ public String controllerIP;
+
+ @Arg(dest = "devices-count")
+ public int deviceCount;
+
+ @Arg(dest = "ssl")
+ public boolean ssl;
+
+ @Arg(dest = "threads")
+ public int threads;
+
+ @Arg(dest = "port")
+ public int port;
+
+ @Arg(dest = "timeout")
+ public int timeout;
+
+ @Arg(dest = "freeze")
+ public int freeze;
+
+ @Arg(dest = "sleep")
+ public long sleep;
+
+ static ArgumentParser getParser() throws UnknownHostException {
+ final ArgumentParser parser = ArgumentParsers.newArgumentParser("openflowjava test-tool");
+
+ parser.description("Openflowjava switch -> controller connector simulator");
+
+ parser.addArgument("--device-count")
+ .type(Integer.class)
+ .setDefault(1)
+ .help("Number of simulated switches. Has to be more than 0")
+ .dest("devices-count");
+
+ parser.addArgument("--controller-ip")
+ .type(String.class)
+ .setDefault("127.0.0.1")
+ .help("ODL controller ip address")
+ .dest("controller-ip");
+
+ parser.addArgument("--ssl")
+ .type(Boolean.class)
+ .setDefault(false)
+ .help("Use secured connection")
+ .dest("ssl");
+
+ parser.addArgument("--threads")
+ .type(Integer.class)
+ .setDefault(1)
+ .help("Number of threads: MAX 1024")
+ .dest("threads");
+
+ parser.addArgument("--port")
+ .type(Integer.class)
+ .setDefault(6653)
+ .help("Connection port")
+ .dest("port");
+
+ parser.addArgument("--timeout")
+ .type(Integer.class)
+ .setDefault(60)
+ .help("Timeout in seconds")
+ .dest("timeout");
+
+ parser.addArgument("--scenarioTries")
+ .type(Integer.class)
+ .setDefault(3)
+ .help("Number of tries in scenario, while waiting for response")
+ .dest("freeze");
+
+ parser.addArgument("--timeBetweenScenario")
+ .type(Long.class)
+ .setDefault(100)
+ .help("Waiting time in milliseconds between tries.")
+ .dest("sleep");
+
+ return parser;
+ }
+
+ void validate() {
+ checkArgument(deviceCount > 0, "Switch count has to be > 0");
+ checkArgument(threads > 0 && threads < 1024, "Switch count has to be > 0 and < 1024");
+ }
+ }
+
+ public static void main(final String[] args) {
+
+ List<Callable<Boolean>> callableList = new ArrayList<>();
+ final EventLoopGroup workerGroup = new NioEventLoopGroup();
+
+ try {
+ final Params params = parseArgs(args, Params.getParser());
+ params.validate();
+
+ for(int loop=0;loop < params.deviceCount; loop++){
+
+ CallableClient cc = new CallableClient(
+ params.port,
+ params.ssl,
+ InetAddress.getByName(params.controllerIP),
+ "Switch no." + String.valueOf(loop),
+ new ScenarioHandler(ScenarioFactory.createHandshakeScenarioWithBarrier(), params.freeze, params.sleep),
+ new Bootstrap(),
+ workerGroup);
+
+ callableList.add(cc);
+
+ }
+
+ ExecutorService executorService = Executors.newFixedThreadPool(params.threads);
+ final ListeningExecutorService listeningExecutorService = MoreExecutors.listeningDecorator(executorService);
+
+ final List<ListenableFuture<Boolean>> listenableFutures = new ArrayList<>();
+ for (Callable<Boolean> booleanCallable : callableList) {
+ listenableFutures.add(listeningExecutorService.submit(booleanCallable));
+ }
+ final ListenableFuture<List<Boolean>> summaryFuture = Futures.successfulAsList(listenableFutures);
+ List<Boolean> booleanList = summaryFuture.get(params.timeout, TimeUnit.SECONDS);
+ Futures.addCallback(summaryFuture, new FutureCallback<List<Boolean>>() {
+ @Override
+ public void onSuccess(@Nullable final List<Boolean> booleanList) {
+ LOG.info("Tests finished");
+ workerGroup.shutdownGracefully();
+ LOG.info("Summary:");
+ int testsOK = 0;
+ int testFailure = 0;
+ for (Boolean aBoolean : booleanList) {
+ if (aBoolean) {
+ testsOK++;
+ } else {
+ testFailure++;
+ }
+ }
+ LOG.info("Tests OK: {}", testsOK);
+ LOG.info("Tests failure: {}", testFailure);
+ System.exit(0);
+ }
+
+ @Override
+ public void onFailure(final Throwable throwable) {
+ LOG.warn("Tests call failure");
+ workerGroup.shutdownGracefully();
+ System.exit(1);
+ }
+ });
+ } catch (Exception e) {
+ LOG.warn("Exception has been thrown: {}", e);
+ System.exit(1);
+ }
+ }
+
+ private static Params parseArgs(final String[] args, final ArgumentParser parser) throws ArgumentParserException {
+ final Params opt = new Params();
+ parser.parseArgs(args, opt);
+ return opt;
+ }
+
+
+}
Code Review / openflowjava.git / blobdiff