test version

Author: mahlaghakafili

archipy/configs/config_template.py

@@ -193,6 +193,7 @@ class GrpcConfig(BaseModel):
     SERVE_PORT: int = Field(default=8100, description="Port to serve gRPC on")
     SERVE_HOST: str = Field(default="[::]", description="Host to serve gRPC on")  # IPv6 equivalent of 0.0.0.0
     THREAD_WORKER_COUNT: int | None = Field(default=None, description="Number of worker threads")
+    MAX_CONCURRENT_RPCS: int | None = Field(default=None, description="Maximum number of concurrent requests")
     THREAD_PER_CPU_CORE: int = Field(
         default=40,
         description="Threads per CPU core",

archipy/helpers/utils/app_utils.py

@@ -7,6 +7,7 @@
 from http import HTTPStatus
 from typing import Any, cast
 
+import grpc
 from pydantic import ValidationError
 
 from archipy.configs.base_config import BaseConfig
@@ -309,7 +310,12 @@ def create_fastapi_app(
         return app
 
     @classmethod
-    def create_async_grpc_app(cls, config: BaseConfig, interceptors: set[Any]) -> server:
+    def create_async_grpc_app(
+        cls,
+        config: BaseConfig,
+        interceptors: set[Any],
+        compression: grpc.Compression | None = None,
+    ) -> server:
         """Create and configure an async gRPC application."""
         from archipy.helpers.interceptors.grpc.exception import AsyncGrpcServerExceptionInterceptor
 
@@ -321,6 +327,9 @@ def create_async_grpc_app(cls, config: BaseConfig, interceptors: set[Any]) -> se
         app = server(
             futures.ThreadPoolExecutor(max_workers=config.GRPC.THREAD_WORKER_COUNT),
             interceptors=async_interceptors,
+            compression=compression,
+            options=config.GRPC.SERVER_OPTIONS_CONFIG_LIST,
+            maximum_concurrent_rpcs=config.GRPC.MAX_CONCURRENT_RPCS,
         )
 
         return app