varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-11 15:40:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix round robin load balancing

Browse Source
This commit is contained in:
sukhman
2025-09-23 08:58:12 +05:30
parent d268393812
commit 3363f57338
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -90,6 +90,7 @@ class CircuitV2Transport(ITransport):
discovery_interval=config.discovery_interval, discovery_interval=config.discovery_interval,
max_relays=config.max_relays, max_relays=config.max_relays,
) )
self.relay_counter = 0 # for round robin load balancing
async def dial( async def dial(
self, self,
@ -231,13 +232,13 @@ class CircuitV2Transport(ITransport):
other_relays.append(relay_id) other_relays.append(relay_id)
# Return first available relay with reservation, or fallback to others # Return first available relay with reservation, or fallback to others
self.relay_counter += 1
if relays_with_reservations: if relays_with_reservations:
return relays_with_reservations[ return relays_with_reservations[
attempts % len(relays_with_reservations) (self.relay_counter-1) % len(relays_with_reservations)
] ]
elif other_relays: elif other_relays:
return other_relays[attempts % len(other_relays)] return other_relays[(self.relay_counter-1) % len(other_relays)]
await trio.sleep(1) await trio.sleep(1)
attempts += 1 attempts += 1