aspyx-service 0.10.7__py3-none-any.whl → 0.11.0__py3-none-any.whl

aspyx_service/__init__.py

@@ -12,6 +12,7 @@ from .healthcheck import health_checks, health_check, HealthCheckManager, Health
 from .restchannel import RestChannel, post, get, put, delete, QueryParam, Body, rest
 from .session import Session, SessionManager, SessionContext
 from .authorization import AuthorizationManager, AbstractAuthorizationFactory
+from .protobuf import ProtobufManager
 
 @module()
 class ServiceModule:
@@ -48,6 +49,10 @@ __all__ = [
     "MissingTokenException",
     "AuthorizationException",
 
+    # protobuf
+
+    "ProtobufManager",
+
     # authorization
 
     "AuthorizationManager",

aspyx_service/channels.py

@@ -196,6 +196,32 @@ class HTTPXChannel(Channel):
 
             headers["Authorization"] = f"Bearer {token}"
 
+        ### TEST
+
+        print_size = False
+        if print_size:
+            request = self.get_client().build_request(http_method, url, params=params, json=json, headers=headers, timeout=timeout, content=content)
+            # Measure body
+            body_size = len(request.content or b"")
+
+            # Measure headers (as raw bytes)
+            headers_size = sum(
+                len(k.encode()) + len(v.encode()) + 4  # ": " + "\r\n"
+                for k, v in request.headers.items()
+            ) + 2  # final \r\n
+
+            # Optional: estimate request line
+            request_line = f"{request.method} {request.url.raw_path.decode()} HTTP/1.1\r\n".encode()
+            request_line_size = len(request_line)
+
+            # Total estimated size
+            total_size = request_line_size + headers_size + body_size
+
+            print(f"Request line: {request_line_size} bytes")
+            print(f"Headers: {headers_size} bytes")
+            print(f"Body: {body_size} bytes")
+            print(f"Total request size: {total_size} bytes")
+
         try:
             response = self.get_client().request(http_method, url, params=params, json=json, headers=headers, timeout=timeout, content=content)
             response.raise_for_status()
@@ -397,10 +423,11 @@ class DispatchMSPackChannel(HTTPXChannel):
                 if "invalid_token" in www_auth:
                     if 'expired' in www_auth:
                         raise TokenExpiredException() from e
-                    elif 'missing' in www_auth:
+
+                    if 'missing' in www_auth:
                         raise MissingTokenException() from e
-                    else:
-                        raise InvalidTokenException() from e
+
+                    raise InvalidTokenException() from e
 
             raise RemoteServiceException(str(e)) from e
         except httpx.HTTPError as e:

aspyx_service/healthcheck.py

@@ -15,7 +15,7 @@ from aspyx.reflection import Decorators, TypeDescriptor
 
 def health_checks():
     """
-    Instances of classes that are annotated with @injectable can be created by an Environment.
+    Instances of classes that are annotated with @health_checks contain healt mehtods.
     """
     def decorator(cls):
         Decorators.add(cls, health_checks)
@@ -31,7 +31,7 @@ def health_checks():
 
 def health_check(name="", cache = 0, fail_if_slower_than = 0):
     """
-    Methods annotated with `@on_init` will be called when the instance is created.
+    Methods annotated with `@health_check` specify health checks that will be executed.
     """
     def decorator(func):
         Decorators.add(func, health_check, name, cache, fail_if_slower_than)