netconf/tools/netconf-testtool/src/main/java/org/opendaylight/netconf/test/tool/client/stress/Parameters.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. 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
   9 package org.opendaylight.netconf.test.tool.client.stress;
  10
  11 import com.google.common.base.Preconditions;
  12 import java.io.File;
  13 import java.net.InetAddress;
  14 import java.net.InetSocketAddress;
  15 import java.net.UnknownHostException;
  16 import net.sourceforge.argparse4j.ArgumentParsers;
  17 import net.sourceforge.argparse4j.annotation.Arg;
  18 import net.sourceforge.argparse4j.inf.ArgumentParser;
  19
  20 public class Parameters {
  21
  22     @Arg(dest = "ip")
  23     public String ip;
  24
  25     @Arg(dest = "port")
  26     public int port;
  27
  28     @Arg(dest = "edit-count")
  29     public int editCount;
  30
  31     @Arg(dest = "edit-content")
  32     public File editContent;
  33
  34     @Arg(dest = "edit-batch-size")
  35     public int editBatchSize;
  36
  37     @Arg(dest = "candidate-datastore")
  38     public boolean candidateDatastore;
  39
  40     @Arg(dest = "debug")
  41     public boolean debug;
  42
  43     @Arg(dest = "legacy-framing")
  44     public boolean legacyFraming;
  45
  46     @Arg(dest = "exi")
  47     public boolean exi;
  48
  49     @Arg(dest = "async")
  50     public boolean async;
  51
  52     @Arg(dest = "ssh")
  53     public boolean ssh;
  54
  55     @Arg(dest = "username")
  56     public String username;
  57
  58     @Arg(dest = "password")
  59     public String password;
  60
  61     @Arg(dest = "msg-timeout")
  62     public long msgTimeout;
  63
  64     @Arg(dest = "tcp-header")
  65     public String tcpHeader;
  66
  67     @Arg(dest = "thread-amount")
  68     public int threadAmount;
  69
  70     static ArgumentParser getParser() {
  71         final ArgumentParser parser = ArgumentParsers.newArgumentParser("netconf stress client");
  72
  73         parser.description("Netconf stress client");
  74
  75         parser.addArgument("--ip")
  76                 .type(String.class)
  77                 .setDefault("127.0.0.1")
  78                 .type(String.class)
  79                 .help("Netconf server IP")
  80                 .dest("ip");
  81
  82         parser.addArgument("--port")
  83                 .type(Integer.class)
  84                 .setDefault(2830)
  85                 .type(Integer.class)
  86                 .help("Netconf server port")
  87                 .dest("port");
  88
  89         parser.addArgument("--edits")
  90                 .type(Integer.class)
  91                 .setDefault(50000)
  92                 .type(Integer.class)
  93                 .help("Netconf edit rpcs to be sent")
  94                 .dest("edit-count");
  95
  96         parser.addArgument("--edit-content")
  97                 .type(File.class)
  98                 .setDefault(new File("edit.txt"))
  99                 .type(File.class)
 100                 .dest("edit-content");
 101
 102         parser.addArgument("--edit-batch-size")
 103                 .type(Integer.class)
 104                 .required(false)
 105                 .setDefault(-1)
 106                 .dest("edit-batch-size");
 107
 108         parser.addArgument("--candidate-datastore")
 109                 .type(Boolean.class)
 110                 .required(false)
 111                 .setDefault(true)
 112                 .help("Edit candidate or running datastore. Defaults to candidate.")
 113                 .dest("candidate-datastore");
 114
 115         parser.addArgument("--debug")
 116                 .type(Boolean.class)
 117                 .setDefault(false)
 118                 .help("Whether to use debug log level instead of INFO")
 119                 .dest("debug");
 120
 121         parser.addArgument("--legacy-framing")
 122                 .type(Boolean.class)
 123                 .setDefault(false)
 124                 .dest("legacy-framing");
 125
 126         parser.addArgument("--exi")
 127                 .type(Boolean.class)
 128                 .setDefault(false)
 129                 .dest("exi");
 130
 131         parser.addArgument("--async-requests")
 132                 .type(Boolean.class)
 133                 .setDefault(true)
 134                 .dest("async");
 135
 136         parser.addArgument("--msg-timeout")
 137                 .type(Integer.class)
 138                 .setDefault(60)
 139                 .dest("msg-timeout");
 140
 141         parser.addArgument("--ssh")
 142                 .type(Boolean.class)
 143                 .setDefault(false)
 144                 .dest("ssh");
 145
 146         parser.addArgument("--username")
 147                 .type(String.class)
 148                 .setDefault("admin")
 149                 .dest("username");
 150
 151         parser.addArgument("--password")
 152                 .type(String.class)
 153                 .setDefault("admin")
 154                 .dest("password");
 155
 156         parser.addArgument("--tcp-header")
 157                 .type(String.class)
 158                 .required(false)
 159                 .dest("tcp-header");
 160
 161         parser.addArgument("--thread-amount")
 162                 .type(Integer.class)
 163                 .setDefault(1)
 164                 .dest("thread-amount");
 165
 166         return parser;
 167     }
 168
 169     void validate() {
 170         Preconditions.checkArgument(port > 0, "Port =< 0");
 171         Preconditions.checkArgument(editCount > 0, "Edit count =< 0");
 172         if (editBatchSize == -1) {
 173             editBatchSize = editCount;
 174         } else {
 175             Preconditions.checkArgument(editBatchSize <= editCount, "Edit count =< 0");
 176         }
 177
 178         Preconditions.checkArgument(editContent.exists(), "Edit content file missing");
 179         Preconditions.checkArgument(editContent.isDirectory() == false, "Edit content file is a dir");
 180         Preconditions.checkArgument(editContent.canRead(), "Edit content file is unreadable");
 181         Preconditions.checkArgument(threadAmount > 0, "Parameter thread-amount must be greater than 0");
 182         Preconditions.checkArgument(msgTimeout >= 0, "Parameter msg-timeout must be greater than 0");
 183     }
 184
 185     public InetSocketAddress getInetAddress() {
 186         try {
 187             return new InetSocketAddress(InetAddress.getByName(ip), port);
 188         } catch (final UnknownHostException e) {
 189             throw new IllegalArgumentException("Unknown ip", e);
 190         }
 191     }
 192 }