samples/simple-client/src/main/java/org/opendaylight/openflowjava/protocol/impl/clients/CallableClient.java

   1 /*
   2  * Copyright (c) 2016 Pantheon Technologies s.r.o. and others. All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.openflowjava.protocol.impl.clients;
   9
  10 import java.net.InetAddress;
  11 import java.util.concurrent.Callable;
  12
  13 import com.google.common.base.Preconditions;
  14 import com.google.common.util.concurrent.SettableFuture;
  15 import io.netty.bootstrap.Bootstrap;
  16 import io.netty.channel.EventLoopGroup;
  17 import io.netty.channel.socket.nio.NioSocketChannel;
  18 import org.slf4j.LoggerFactory;
  19
  20
  21 /**
  22  * Callable client class, inspired by SimpleClient class
  23  * Simulating device/switch connected to controller
  24  * @author Jozef Bacigal
  25  * Date: 4.3.2016.
  26  */
  27 public class CallableClient implements Callable<Boolean>, OFClient {
  28
  29     private static final org.slf4j.Logger LOG = LoggerFactory.getLogger(CallableClient.class);
  30
  31     private int port = 6653;
  32     private boolean securedClient = false;
  33     private InetAddress ipAddress = null;
  34     private String name = "Empty name";
  35
  36     private final EventLoopGroup workerGroup;
  37     private SettableFuture<Boolean> isOnlineFuture;
  38     private SettableFuture<Boolean> scenarioDone;
  39     private ScenarioHandler scenarioHandler = null;
  40     private Bootstrap bootstrap = null;
  41
  42     public CallableClient(
  43             final int port,
  44             final boolean securedClient,
  45             final InetAddress ipAddress,
  46             final String name,
  47             final ScenarioHandler scenarioHandler,
  48             final Bootstrap bootstrap,
  49             final EventLoopGroup eventExecutors) {
  50
  51         Preconditions.checkNotNull(ipAddress, "IP address cannot be null");
  52         Preconditions.checkNotNull(scenarioHandler, "Scenario handler cannot be null");
  53         this.port = port;
  54         this.securedClient = securedClient;
  55         this.ipAddress = ipAddress;
  56         this.workerGroup = eventExecutors;
  57         this.bootstrap = bootstrap;
  58         this.name = name;
  59         this.scenarioHandler = scenarioHandler;
  60     }
  61
  62     @Override
  63     public SettableFuture<Boolean> getIsOnlineFuture() {
  64         return isOnlineFuture;
  65     }
  66
  67     @Override
  68     public SettableFuture<Boolean> getScenarioDone() {
  69         return scenarioDone;
  70     }
  71
  72     @Override
  73     public void setScenarioHandler(final ScenarioHandler scenario) {
  74         this.scenarioHandler = scenario;
  75     }
  76
  77     @Override
  78     public void setSecuredClient(final boolean securedClient) {
  79         this.securedClient = securedClient;
  80     }
  81
  82
  83     @Override
  84     public Boolean call() throws Exception {
  85         Preconditions.checkNotNull(bootstrap);
  86         Preconditions.checkNotNull(workerGroup);
  87         LOG.info("Switch {} trying connect to controller", this.name);
  88         SimpleClientInitializer clientInitializer = new SimpleClientInitializer(isOnlineFuture, securedClient);
  89         clientInitializer.setScenario(scenarioHandler);
  90         try {
  91             bootstrap.group(workerGroup)
  92                     .channel(NioSocketChannel.class)
  93                     .handler(clientInitializer);
  94
  95             bootstrap.connect(ipAddress, port).sync();
  96             synchronized (scenarioHandler) {
  97                 LOG.debug("WAITING FOR SCENARIO");
  98                 while (!scenarioHandler.isScenarioFinished()) {
  99                     scenarioHandler.wait();
 100                 }
 101             }
 102         } catch (Exception ex) {
 103             LOG.error(ex.getMessage(), ex);
 104             return false;
 105         }
 106         if (scenarioHandler.isFinishedOK()) {
 107             LOG.info("Device {} finished scenario OK", this.name);
 108         } else {
 109             LOG.error("Device {} finished scenario with error", this.name);
 110         }
 111         return scenarioHandler.isFinishedOK();
 112
 113     }
 114
 115     @Override
 116     public void run() {
 117         throw new UnsupportedOperationException();
 118     }
 119 }