Improve async client to use recv_task for response tracking

Author: Ananto30

Makefile

@@ -5,11 +5,11 @@ setup:
 	source venv/bin/activate && ( \
 		pip install -r requirements.txt; \
 		pip install -r tests/requirements.txt; \
-		pip install black isort flake8 pylint pytype mypy; \
+		pip install -r requirements-lint.txt; \
 		)
 
 test:
-	python3 -m pytest tests --cov=zero --cov-report=term-missing --cov-config=.coveragerc -vv --durations=10 --timeout=280 -s --exitfirst
+	python3 -m pytest tests --cov=zero --cov-report=term-missing --cov-config=.coveragerc -vv --durations=10 --timeout=280
 
 docker-test:
 	docker build -t zero-test -f Dockerfile.test .

benchmarks/local/zero/Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.11-slim
+FROM python:3.13-slim
 
 COPY requirements.txt requirements.txt
 RUN pip install -r requirements.txt

benchmarks/local/zero/docker-compose.yaml

@@ -16,9 +16,9 @@ services:
   # gateway:
   #   build: .
   #   container_name: gateway
-  #   command: gunicorn gateway_aiohttp:app --bind 0.0.0.0:8000 --worker-class aiohttp.worker.GunicornWebWorker --workers 8 --log-level  warning
+  #   command: gunicorn gateway_aiohttp:app --bind 0.0.0.0:8766 --worker-class aiohttp.worker.GunicornWebWorker --workers 8 --log-level  warning
   #   ports:
-  #     - "8000:8000"
+  #     - "8766:8766"
   #   depends_on:
   #     - server
   #     - redis

benchmarks/others/zero_publisher.py

@@ -4,7 +4,6 @@
 
 from benchmarks.async_redis_repository import save_order as saveOrder
 from benchmarks.model import CreateOrderReq, Order, OrderResp, OrderStatus
-
 from zero import ZeroSubscriber
 
 

requirements-lint.txt

@@ -4,4 +4,5 @@ flake8
 pylint
 pytype
 mypy
-pydantic
+pydantic
+uvloop

setup.py

@@ -24,5 +24,7 @@
     install_requires=["pyzmq", "msgspec"],
     extras_require={
         "pydantic": ["pydantic"],  # Optional dependency
+        "uvloop": ["uvloop"],  # Optional dependency for better performance
+        "all": ["pydantic", "uvloop"],  # Install all optional dependencies
     },
 )

tests/concurrency/rps_sync.py

@@ -18,7 +18,7 @@ def get_and_sum(msg):
 
 
 if __name__ == "__main__":
-    process_no = 32
+    process_no = 8
     pool = Pool(process_no)
 
     sum_items = []

tests/utils.py

@@ -67,6 +67,10 @@ def start_subprocess(module: str) -> subprocess.Popen:
 def kill_subprocess(process: subprocess.Popen):
     pid = process.pid
     process.terminate()
-    process.wait()
+    try:
+        process.wait(timeout=5)
+    except subprocess.TimeoutExpired:
+        process.kill()
+        process.wait()
     if pid and pid in process_map:
         del process_map[pid]

zero/codegen/codegen.py

@@ -20,10 +20,16 @@
 )
 
 import msgspec
-from pydantic import BaseModel
 
 from zero.utils.type_util import typing_types
 
+try:
+    from pydantic import BaseModel
+
+    PYDANTIC_AVAILABLE = True
+except ImportError:
+    PYDANTIC_AVAILABLE = False
+
 python_version = sys.version_info
 
 
@@ -217,16 +223,14 @@ def _generate_code_for_type(self, typ: Type, already_generated: Set[Type]) -> st
         for possible_typ in all_possible_typs:
             self._track_imports(possible_typ)
             if isinstance(possible_typ, type) and (
-                issubclass(  # pytype: disable=wrong-arg-types
-                    possible_typ,
-                    (
-                        msgspec.Struct,
-                        BaseModel,
-                        enum.Enum,
-                        enum.IntEnum,
-                    ),
-                )
+                issubclass(possible_typ, (msgspec.Struct, enum.Enum, enum.IntEnum))
                 or is_dataclass(possible_typ)
+                or (
+                    PYDANTIC_AVAILABLE
+                    and issubclass(  # pytype: disable=wrong-arg-types
+                        possible_typ, BaseModel
+                    )
+                )
             ):
                 code += self._generate_class_code(possible_typ, already_generated)
         return code

zero/protocols/zeromq/server.py

@@ -15,8 +15,6 @@
 
 from .worker import _Worker
 
-# import uvloop
-
 
 class ZMQServer:
     def __init__(