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

make limit configurable in push_identify_to_peers

Browse Source
This commit is contained in:
sukhman
2025-07-02 13:11:34 +05:30
parent cbe2b4bd99
commit ad0b5505ba
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
libp2p/identity/identify_push/identify_push.py
View File

@ -41,7 +41,6 @@ ID_PUSH = TProtocol("/ipfs/id/push/1.0.0")
PROTOCOL_VERSION = "ipfs/0.1.0" PROTOCOL_VERSION = "ipfs/0.1.0"
AGENT_VERSION = get_agent_version() AGENT_VERSION = get_agent_version()
CONCURRENCY_LIMIT = 10 CONCURRENCY_LIMIT = 10
LIMIT = trio.Semaphore(CONCURRENCY_LIMIT)
def identify_push_handler_for(host: IHost) -> StreamHandlerFn: def identify_push_handler_for(host: IHost) -> StreamHandlerFn:
@ -134,7 +133,10 @@ async def _update_peerstore_from_identify(
async def push_identify_to_peer( async def push_identify_to_peer(
host: IHost, peer_id: ID, observed_multiaddr: Multiaddr | None = None host: IHost,
peer_id: ID,
observed_multiaddr: Multiaddr | None = None,
limit=trio.Semaphore(CONCURRENCY_LIMIT),
) -> bool: ) -> bool:
""" """
Push an identify message to a specific peer. Push an identify message to a specific peer.
@ -148,7 +150,7 @@ async def push_identify_to_peer(
True if the push was successful, False otherwise. True if the push was successful, False otherwise.
""" """
async with LIMIT: async with limit:
try: try:
# Create a new stream to the peer using the identify/push protocol # Create a new stream to the peer using the identify/push protocol
stream = await host.new_stream(peer_id, [ID_PUSH]) stream = await host.new_stream(peer_id, [ID_PUSH])