varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve relay selection to prioritize active reservations

Browse Source 
Signed-off-by: varun-r-mallya <varunrmallya@gmail.com>
This commit is contained in:
varun-r-mallya
2025-08-19 04:53:50 +05:30
parent cff0bfc17d
commit 5de09ed8a1
1 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
libp2p/relay/circuit_v2/transport.py
View File

@ -219,11 +219,25 @@ class CircuitV2Transport(ITransport):
# Get a relay from the list of discovered relays
relays = self.discovery.get_relays()
if relays:
# TODO: Implement more sophisticated relay selection
# For now, just return the first available relay
return relays[0]
# Prioritize relays with active reservations
relays_with_reservations = []
other_relays = []
for relay_id in relays:
relay_info = self.discovery.get_relay_info(relay_id)
if relay_info and relay_info.has_reservation:
relays_with_reservations.append(relay_id)
else:
other_relays.append(relay_id)
# Return first available relay with reservation, or fallback to others
if relays_with_reservations:
return relays_with_reservations[
attempts % len(relays_with_reservations)
]
elif other_relays:
return other_relays[attempts % len(other_relays)]
# Wait and try discovery
await trio.sleep(1)
attempts += 1