algo-backend-framework 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

algo_backend/metrics/time_cost_metrics.py

@@ -84,8 +84,8 @@ class ClientTimeCostMetrics(BasicTimeCostMetrics):
 
 class PrometheusTimeCostMetricSetting:
     __default_metrics = BasicTimeCostMetrics()
-    __api_metrics = None
-    __client_metrics = None
+
+    __metrics_dict = {"default": BasicTimeCostMetrics(), "api": None, "client": None}
 
     @classmethod
     def initialize(
@@ -94,35 +94,42 @@ class PrometheusTimeCostMetricSetting:
         api_metrics_buckets: Optional[List[int]] = None,
         client_metrics_buckets: Optional[List[int]] = None,
     ):
-        if api_metrics_buckets is None:
-            cls.__api_metrics = ApiTimeCostMetrics()
-        elif len(api_metrics_buckets) == 0:
-            cls.__api_metrics = BasicTimeCostMetrics()
-        else:
-            cls.__api_metrics = ApiTimeCostMetrics(buckets=list(api_metrics_buckets))
+        cls.init_metrics("api", ApiTimeCostMetrics, api_metrics_buckets)
+        cls.init_metrics("client", ClientTimeCostMetrics, client_metrics_buckets)
 
-        if client_metrics_buckets is None:
-            cls.__client_metrics = ClientTimeCostMetrics()
-        elif len(client_metrics_buckets) == 0:
-            cls.__client_metrics = BasicTimeCostMetrics()
+    @classmethod
+    def init_metrics(
+        cls, name: str, metrics_class, buckets: Optional[List[int]] = None
+    ):
+        if cls.__metrics_dict[name] is None:
+            cls.__metrics_dict[name] = metrics_class()
+        elif len(buckets) == 0:
+            cls.__metrics_dict[name] = cls.__default_metrics
         else:
-            cls.__client_metrics = ClientTimeCostMetrics(
-                buckets=list(client_metrics_buckets)
-            )
+            cls.__metrics_dict[name] = metrics_class(buckets=list(buckets))
+
+    @classmethod
+    def get_metrics(cls, name: str) -> BasicTimeCostMetrics:
+        return cls.__metrics_dict.get(name, cls.__default_metrics)
 
     @classmethod
     def api_metrics_instance(cls) -> BasicTimeCostMetrics:
-        return cls.__api_metrics or cls.__default_metrics
+        return cls.get_metrics("api")
 
     @classmethod
     def client_metrics_instance(cls) -> BasicTimeCostMetrics:
-        return cls.__client_metrics or cls.__default_metrics
+        return cls.get_metrics("client")
 
     @classmethod
-    def metrics_handler(cls, key: str, metrics: BasicTimeCostMetrics):
+    def metrics_handler(cls, key: str, metrics_cls_name: str):
+        """
+        装饰器只会初始化一次，因此要使用动态方式获取指标类，否则初始化装饰器时，指标类还没有被初始化
+        """
+
         def decorator(func: Callable[..., Awaitable]) -> Callable[..., Awaitable]:
             @wraps(func)
             async def wrapper(*args, **kwargs):
+                metrics = cls.get_metrics(metrics_cls_name)
                 try:
                     start = time.perf_counter()
                     result = await func(*args, **kwargs)
@@ -139,8 +146,8 @@ class PrometheusTimeCostMetricSetting:
 
     @classmethod
     def client_metrics_handler(cls, client_name: str):
-        return cls.metrics_handler(client_name, cls.client_metrics_instance())
+        return cls.metrics_handler(client_name, "client")
 
     @classmethod
     def api_metrics_handler(cls, api_name: str):
-        return cls.metrics_handler(api_name, cls.api_metrics_instance())
+        return cls.metrics_handler(api_name, "api")

{algo_backend_framework-0.0.1.dist-info → algo_backend_framework-0.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: algo-backend-framework
-Version: 0.0.1
+Version: 0.0.2
 Summary: Ctcdn algorithm backend framework
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{algo_backend_framework-0.0.1.dist-info → algo_backend_framework-0.0.2.dist-info}/RECORD

@@ -24,7 +24,7 @@ algo_backend/log/nblog/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 algo_backend/metrics/__init__.py,sha256=20CPEFgi0yJ2Fy-LTcFQ3aqlXHhLPtgPFboqOMBhl34,640
 algo_backend/metrics/http_metrics.py,sha256=3GYNhKJoS6oJ8L7FXnVkPZ_wa-SKj_444KLdBeGkbs4,1944
 algo_backend/metrics/prometheus_context.py,sha256=GqemLccNbTx_5DBijOyzySzWydbpWYfnhWh17rujjwE,1910
-algo_backend/metrics/time_cost_metrics.py,sha256=0v8Mu4lXbgKjAFH3aK0GPtVhw_70kIYMd-eH-ceyrkI,4645
+algo_backend/metrics/time_cost_metrics.py,sha256=wXSv3yN6_Eq55KduGy9tIX7sZWsdHI9YBnWoSsT9sZE,4895
 algo_backend/metrics/collector/__init__.py,sha256=t5Csl-Z9qpik-isEUVYDYBWYgf7xdM49548YfJr4X64,413
 algo_backend/metrics/collector/common.py,sha256=yeiZp9ufbYSW7N-R6ytk74UF0dCqJ8r338lL64TmA4g,475
 algo_backend/metrics/collector/gc_metrics.py,sha256=s2ariKimji9LC71eAQFlrFkpcF3hkSy5IBC6D8LFePc,2510
@@ -42,7 +42,7 @@ algo_backend/starter/event_list.py,sha256=vQHzQIpW8LZmQ93YyET-1gX6pQGVE5A6I_pLoY
 algo_backend/utils/__init__.py,sha256=oX6OyL-28jzc94u4fyH1TtntCzQySkfZ8jibMk1KPU8,168
 algo_backend/utils/meta_class.py,sha256=hcZPGF7EIHvJOXXR82_7Gah_AWbqkcSxUc473I_6maY,1850
 algo_backend/utils/utils.py,sha256=q3bxBrivndLRggWsLryloSpu-_Ecbj4mhZL8oYdiDTo,481
-algo_backend_framework-0.0.1.dist-info/METADATA,sha256=D4eriT_lp8IIMOwz5JLKBGca-6k7STDRBNKTLcmMVRE,2196
-algo_backend_framework-0.0.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-algo_backend_framework-0.0.1.dist-info/top_level.txt,sha256=zLsbLTRV1tO2hQfazqiBLO73VnjSAhJSUpMMBmQaLfw,13
-algo_backend_framework-0.0.1.dist-info/RECORD,,
+algo_backend_framework-0.0.2.dist-info/METADATA,sha256=U9d8H4QUHPR0DMGyRlbthlHnRROgg8X0mw1t0LdyokQ,2196
+algo_backend_framework-0.0.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+algo_backend_framework-0.0.2.dist-info/top_level.txt,sha256=zLsbLTRV1tO2hQfazqiBLO73VnjSAhJSUpMMBmQaLfw,13
+algo_backend_framework-0.0.2.dist-info/RECORD,,