Adding locks when manipulating the _session_cache object in case of multithreading

Signed-off-by: Harmouch101 <[email protected]>

Author: wiseaidev

web3/_utils/request.py

@@ -1,5 +1,14 @@
+from collections import (
+    defaultdict,
+)
+from functools import _make_key  # type: ignore
+from functools import lru_cache
+from threading import (
+    RLock,
+)
 from typing import (
     Any,
+    Callable,
 )
 
 from eth_typing import (
@@ -13,6 +22,22 @@
 )
 
 
+def thread_safe_lru(func: Callable[..., Any]) -> Callable[..., Any]:
+    """
+    A thread-safe cache decorator that prevents concurrent invocations
+    and allow reusing from cache.
+    """
+    caching_func = lru_cache(maxsize=8, typed=False)(func)
+    _keyed_locks = defaultdict(RLock)  # type: ignore
+
+    def inner(*args: Any, **kwargs: Any) -> Callable[..., Any]:
+        key = _make_key(args, kwargs, typed=False)
+        with _keyed_locks[key]:
+            return caching_func(*args, **kwargs)
+    return inner
+
+
+@thread_safe_lru
 def cache_session(endpoint_uri: URI, session: requests.Session) -> None:
     cache_key = generate_cache_key(endpoint_uri)
     _session_cache[cache_key] = session
@@ -32,6 +57,7 @@ def _get_session(endpoint_uri: URI) -> requests.Session:
     return _session_cache[cache_key]
 
 
+@thread_safe_lru
 def make_post_request(endpoint_uri: URI, data: bytes, *args: Any, **kwargs: Any) -> bytes:
     kwargs.setdefault('timeout', 10)
     session = _get_session(endpoint_uri)