mirror of
https://github.com/savoirfairelinux/opendht.git
synced 2026-11-08 04:21:51 +00:00
79 lines
2.1 KiB
Python
Executable File
79 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2015-2023 Savoir-faire Linux Inc.
|
|
# Author: Adrien Béraud <adrien.beraud@savoirfairelinux.com>
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import opendht as dht
|
|
import time
|
|
import asyncio
|
|
|
|
config = dht.DhtConfig()
|
|
config.setRateLimit(-1, -1)
|
|
|
|
ping_node = dht.DhtRunner()
|
|
ping_node.run(config=config)
|
|
#ping_node.enableLogging()
|
|
#ping_node.bootstrap("bootstrap.jami.net", "4222")
|
|
pong_node = dht.DhtRunner()
|
|
pong_node.run(config=config)
|
|
#pong_node.enableLogging()
|
|
pong_node.ping(ping_node.getBound())
|
|
|
|
loc_ping = dht.InfoHash.get("toto99")
|
|
loc_pong = dht.InfoHash.get(str(loc_ping))
|
|
|
|
net = []
|
|
for i in range(1,10):
|
|
node = dht.DhtRunner()
|
|
node.run(config=config)
|
|
node.ping(ping_node.getBound())
|
|
net.append(node)
|
|
|
|
i = 0
|
|
MAX = 2048
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
def done(h, ok):
|
|
pass
|
|
#print(h, "over", ok)
|
|
|
|
def ping(node, h):
|
|
global i
|
|
i += 1
|
|
if i < MAX:
|
|
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
|
|
|
|
t1 = time.time()
|
|
|
|
ping_node.listen(loc_ping, lambda v, e: pong(pong_node, loc_pong) if not e else True)
|
|
pong_node.listen(loc_pong, lambda v, e: pong(ping_node, loc_ping) if not e else True)
|
|
|
|
ping(pong_node, loc_ping)
|
|
|
|
loop.run_forever()
|
|
|
|
t2 = time.time()
|
|
|
|
print(MAX, "ping-pong done, took", t2 - t1, "s")
|
|
print(1000 * (t2 - t1)/MAX, "ms per rt", MAX/(t2 - t1), "rt per s")
|