varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-12 16:10:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

smol fix , adds degree

Browse Source
This commit is contained in:
guha-rahul
2025-06-27 04:41:52 +05:30
parent 6a92fa26eb
commit 7ed33e9a55
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/utils/pubsub/utils.py
View File

@ -24,16 +24,22 @@ def make_pubsub_msg(
) )
# TODO: Implement sparse connect
async def dense_connect(hosts: Sequence[IHost]) -> None: async def dense_connect(hosts: Sequence[IHost]) -> None:
await connect_some(hosts, 10) await connect_some(hosts, 10)
# FIXME: `degree` is not used at all
async def connect_some(hosts: Sequence[IHost], degree: int) -> None: async def connect_some(hosts: Sequence[IHost], degree: int) -> None:
"""
Connect each host to up to 'degree' number of other hosts.
Creates a sparse network topology where each node has limited connections.
"""
for i, host in enumerate(hosts): for i, host in enumerate(hosts):
for host2 in hosts[i + 1 :]: connections_made = 0
await connect(host, host2) for j in range(i + 1, len(hosts)):
if connections_made >= degree:
break
await connect(host, hosts[j])
connections_made += 1
async def one_to_all_connect(hosts: Sequence[IHost], central_host_index: int) -> None: async def one_to_all_connect(hosts: Sequence[IHost], central_host_index: int) -> None: