From 0f5086fef5d2e63298ad68bab40f749205fb63fa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Simon=20D=C3=A9saulniers?= <sim.desaulniers@gmail.com>
Date: Tue, 5 Apr 2016 16:29:37 -0400
Subject: [PATCH] benchmark: toggling bootstrap dht log USR1 signal
---
python/tools/benchmark.py | 20 +++++++++++++++-----
1 file changed, 15 insertions(+), 5 deletions(-)
diff --git a/python/tools/benchmark.py b/python/tools/benchmark.py
index 257eddba..ff98fbac 100755
--- a/python/tools/benchmark.py
+++ b/python/tools/benchmark.py
@@ -132,7 +132,6 @@ class WorkBench():
if __name__ == '__main__':
-
parser = argparse.ArgumentParser(description='Run, test and benchmark a '\
'DHT network on a local virtual network with simulated packet '\
'loss and latency.')
@@ -165,19 +164,30 @@ if __name__ == '__main__':
'Use "-m" to specify the number of producers on the DHT.'\
'Use "-e" to specify the number of values to put on the DHT.')
-
args = parser.parse_args()
test_opt = { o : True for o in args.opt }
wb = WorkBench(args.ifname, args.virtual_locs, args.node_num, loss=args.loss,
delay=args.delay, disable_ipv4=args.disable_ipv4,
disable_ipv6=args.disable_ipv6)
- wb.create_virtual_net()
-
bootstrap = wb.get_bootstrap()
+
+ bs_dht_log_enabled = False
+ def toggle_bs_dht_log(signum, frame):
+ global bs_dht_log_enabled, bootstrap
+ if bs_dht_log_enabled:
+ bootstrap.front().disableLogging()
+ bs_dht_log_enabled = False
+ else:
+ bootstrap.front().enableLogging()
+ bs_dht_log_enabled = True
+ signal.signal(signal.SIGUSR1, toggle_bs_dht_log)
+
if args.bs_dht_log:
- bootstrap.enableLogging()
+ bs_dht_log_enabled = True
+ bootstrap.front().enableLogging()
+ wb.create_virtual_net()
bootstrap.resize(1)
print("Launching", wb.node_num, "nodes (", wb.clusters, "clusters of", wb.node_per_loc, "nodes)")
--
GitLab