From a9d07e0e686f9596501df506d3ad9ba9ef0246da Mon Sep 17 00:00:00 2001
From: Rod Smith <rod.smith@canonical.com>
Date: Fri, 21 Jul 2017 14:23:54 -0400
Subject: Improve network test by enabling parallel iperf/iperf3 runs (-P
 option to iperf) on > 10 Gbps network connections, or by manually specifying
 via --num-threads option to network script

---
 bin/network | 25 ++++++++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/bin/network b/bin/network
index c855f0a..db40ebe 100755
--- a/bin/network
+++ b/bin/network
@@ -28,6 +28,7 @@ import datetime
 import fcntl
 import ipaddress
 import logging
+import math
 import os
 import re
 import shlex
@@ -57,6 +58,7 @@ class IPerfPerformanceTest(object):
             fail_threshold,
             cpu_load_fail_threshold,
             iperf3,
+            num_threads,
             protocol="tcp",
             data_size="1",
             run_time=None,
@@ -68,6 +70,7 @@ class IPerfPerformanceTest(object):
         self.fail_threshold = fail_threshold
         self.cpu_load_fail_threshold = cpu_load_fail_threshold
         self.iperf3 = iperf3
+        self.num_threads = num_threads
         self.data_size = data_size
         self.run_time = run_time
         self.scan_timeout = scan_timeout
@@ -84,10 +87,22 @@ class IPerfPerformanceTest(object):
         else:
             self.executable = "iperf"
 
+        # Determine number of parallel threads
+        if self.num_threads == -1:
+            # Below is a really crude guesstimate based on our
+            # initial testing. It's likely possible to improve
+            # this method of setting the number of threads.
+            threads = math.ceil(self.iface.link_speed / 10000)
+        else:
+            threads = self.num_threads
+
+        if threads != 1:
+            logging.info("Using {} threads.".format(threads))
+
         # If we set run_time, use that instead to build the command.
         if self.run_time is not None:
-            cmd = "{} -c {} -t {} -i 1 -f m".format(
-                self.executable, self.target, self.run_time)
+            cmd = "{} -c {} -t {} -i 1 -f m -P {}".format(
+                self.executable, self.target, self.run_time, threads)
         else:
             # Because we can vary the data size, we need to vary the timeout as
             # well.  It takes an estimated 15 minutes to send 1GB over 10Mb/s.
@@ -389,7 +404,7 @@ def run_test(args, test_target):
         iperf_benchmark = IPerfPerformanceTest(args.interface, test_target,
                                                args.fail_threshold,
                                                args.cpu_load_fail_threshold,
-                                               args.iperf3)
+                                               args.iperf3, args.num_threads)
         if args.datasize:
             iperf_benchmark.data_size = args.datasize
         if args.runtime:
@@ -709,6 +724,10 @@ TEST_TARGET_IPERF = iperf-server.example.com
     test_parser.add_argument(
         '--underspeed-ok', default=False, action="store_true",
         help="Run test even if an underspeed 1ink is detected")
+    test_parser.add_argument(
+        '--num-threads', type=int, default=-1,
+        help=("Number of threads to use in the test. "
+              "(Default is computed based on network speed.)"))
 
     # Sub info options
     info_parser.add_argument(
-- 
cgit v1.2.3