feat: metric controller

metric can be enable or disable by controller

Author: Kl0ven

sentry_dynamic_sampling_lib/sampler.py

@@ -62,12 +62,17 @@ def update_config(self):
             return
 
         data = resp.json()
-        self.config.sample_rate = data["active_sample_rate"]
-        self.config.ignored_paths = data["wsgi_ignore_path"]
-        self.config.ignored_tasks = data["celery_ignore_task"]
+        self.config.update(data)
+        self.metrics.set_mode(MetricType.CELERY, data["celery_collect_metrics"])
+        self.metrics.set_mode(MetricType.WSGI, data["wsgi_collect_metrics"])
 
     def update_metrics(self):
         for metric_type in MetricType:
+            # check if metric is enable
+            mode = self.metrics.get_mode(metric_type)
+            if not mode:
+                continue
+
             counter = self.metrics.get_and_reset(metric_type)
             if len(counter) == 0:
                 return

sentry_dynamic_sampling_lib/shared.py

@@ -47,6 +47,12 @@ def ignored_tasks(self):
     def ignored_tasks(self, new_ignored_tasks):
         self._ignored_tasks = set(new_ignored_tasks)
 
+    @synchronized
+    def update(self, data):
+        self._sample_rate = data["active_sample_rate"]
+        self._ignored_paths = data["wsgi_ignore_path"]
+        self._ignored_tasks = data["celery_ignore_task"]
+
 
 class MetricType(Enum):
     WSGI = "WSGI"
@@ -56,18 +62,30 @@ class MetricType(Enum):
 class Metric:
     def __init__(self) -> None:
         self._lock = RLock()
+        self._activate = {
+            MetricType.WSGI: False,
+            MetricType.CELERY: False,
+        }
         self._counters = {
             MetricType.WSGI: Counter(),
             MetricType.CELERY: Counter(),
         }
 
+    def set_mode(self, _type, mode):
+        self._activate[_type] = mode
+
+    def get_mode(self, _type):
+        return self._activate[_type]
+
     @synchronized
     def count_path(self, path):
-        self._counters[MetricType.WSGI][path] += 1
+        if self._activate[MetricType.WSGI]:
+            self._counters[MetricType.WSGI][path] += 1
 
     @synchronized
     def count_task(self, path):
-        self._counters[MetricType.CELERY][path] += 1
+        if self._activate[MetricType.CELERY]:
+            self._counters[MetricType.CELERY][path] += 1
 
     @synchronized
     def get_and_reset(self, _type):