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

move concurrency_limit to identify_push

Browse Source 
Signed-off-by: sukhman <sukhmansinghsaluja@gmail.com>
This commit is contained in:
sukhman
2025-06-30 10:12:02 +05:30
parent 644fc77b6c
commit ad87e50eb7
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -40,7 +40,8 @@ logger = logging.getLogger(__name__)
ID_PUSH = TProtocol("/ipfs/id/push/1.0.0")
PROTOCOL_VERSION = "ipfs/0.1.0"
AGENT_VERSION = get_agent_version()
LIMIT = trio.Semaphore(10)
CONCURRENCY_LIMIT = 10
LIMIT = trio.Semaphore(CONCURRENCY_LIMIT)
def identify_push_handler_for(host: IHost) -> StreamHandlerFn:

7
tests/core/identity/identify_push/test_identify_push.py
View File

@ -20,6 +20,7 @@ from libp2p.identity.identify.pb.identify_pb2 import (
Identify,
)
from libp2p.identity.identify_push.identify_push import (
CONCURRENCY_LIMIT,
ID_PUSH,
_update_peerstore_from_identify,
identify_push_handler_for,
@ -449,8 +450,6 @@ async def test_push_identify_to_peers_respects_concurrency_limit():
It mocks `push_identify_to_peer` to simulate delay using sleep,
allowing the test to measure and assert actual concurrency behavior.
"""
CONCURRENCY_LIMIT = 10
LIMIT = trio.Semaphore(CONCURRENCY_LIMIT)
state = {
"concurrency_counter": 0,
"max_observed": 0,
@ -458,7 +457,7 @@ async def test_push_identify_to_peers_respects_concurrency_limit():
lock = trio.Lock()
async def mock_push_identify_to_peer(
host, peer_id, observed_multiaddr=None
host, peer_id, observed_multiaddr=None, limit=trio.Semaphore(CONCURRENCY_LIMIT)
) -> bool:
"""
Mock function to test concurrency by simulating an identify message.
@ -471,7 +470,7 @@ async def test_push_identify_to_peers_respects_concurrency_limit():
True if the push was successful, False otherwise.
"""
async with LIMIT:
async with limit:
async with lock:
state["concurrency_counter"] += 1
if state["concurrency_counter"] > CONCURRENCY_LIMIT: