From 5a7c8753e150dfbed6d36e48535e91df8f767e35 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Adrien=20B=C3=A9raud?= <adrien.beraud@savoirfairelinux.com>
Date: Mon, 9 Jan 2017 14:40:06 -0500
Subject: [PATCH] python/tools: use asyncio in pingpong test
---
python/tools/pingpong.py | 24 +++++++++++++++++-------
1 file changed, 17 insertions(+), 7 deletions(-)
diff --git a/python/tools/pingpong.py b/python/tools/pingpong.py
index a7945820..d2138624 100755
--- a/python/tools/pingpong.py
+++ b/python/tools/pingpong.py
@@ -17,6 +17,7 @@
import opendht as dht
import time
+import asyncio
ping_node = dht.DhtRunner()
ping_node.run(port = 5632)
@@ -35,19 +36,28 @@ loc_pong = dht.InfoHash.get(str(loc_ping))
i = 0
MAX = 8192
-def pong(node, h):
+loop = asyncio.get_event_loop()
+
+def done(h, ok):
+ print(h, "over", ok)
+
+def ping(node, h):
global i
- print("got ping", h, i)
- time.sleep(0.0075)
+ time.sleep(0.0075)
i += 1
if i < MAX:
- node.put(h, dht.Value(b"hey"), lambda ok, nodes: print("over", ok))
+ node.put(h, dht.Value(b"hey"), lambda ok, nodes: done(node.getNodeId().decode(), ok))
+ else:
+ loop.stop()
+
+def pong(node, h):
+ print(node.getNodeId().decode(), "got ping", h, i)
+ loop.call_soon_threadsafe(ping, node, h);
return True
ping_node.listen(loc_ping, lambda v: pong(pong_node, loc_pong))
pong_node.listen(loc_pong, lambda v: pong(ping_node, loc_ping))
-pong_node.put(loc_pong, dht.Value(b"hey"), lambda ok, nodes: print("i over", ok))
+ping(pong_node, loc_ping)
-while (i < MAX):
- time.sleep(1)
+loop.run_forever()
--
GitLab