feat(metrics): Move minimetrics code to the SDK

Author: mitsuhiko

sentry_sdk/_types.py

@@ -13,7 +13,11 @@
     from typing import Any
     from typing import Callable
     from typing import Dict
+    from typing import Iterable
+    from typing import List
+    from typing import Mapping
     from typing import Optional
+    from typing import Sequence
     from typing import Tuple
     from typing import Type
     from typing import Union
@@ -87,3 +91,52 @@
     MeasurementUnit = Union[DurationUnit, InformationUnit, FractionUnit, str]
 
     ProfilerMode = Literal["sleep", "thread", "gevent", "unknown"]
+
+    # Unit of the metrics.
+    MetricUnit = Literal[
+        "none",
+        "nanosecond",
+        "microsecond",
+        "millisecond",
+        "second",
+        "minute",
+        "hour",
+        "day",
+        "week",
+        "bit",
+        "byte",
+        "kilobyte",
+        "kibibyte",
+        "mebibyte",
+        "gigabyte",
+        "terabyte",
+        "tebibyte",
+        "petabyte",
+        "pebibyte",
+        "exabyte",
+        "exbibyte",
+        "ratio",
+        "percent",
+    ]
+
+    # Type of the metric.
+    MetricType = Literal["d", "s", "g", "c"]
+
+    # Value of the metric.
+    MetricValue = Union[int, float, str]
+
+    # Tag key of a metric.
+    MetricTagKey = str
+
+    # Internal representation of tags as a tuple of tuples (this is done in order to allow for the same key to exist
+    # multiple times).
+    MetricTagsInternal = Tuple[Tuple[MetricTagKey, str], ...]
+
+    # External representation of tags as a dictionary.
+    MetricTagValue = Union[str, List[str], Tuple[str, ...]]
+    MetricTags = Mapping[MetricTagKey, MetricTagValue]
+
+    # Value inside the generator for the metric value.
+    FlushedMetricValue = Union[int, float]
+
+    BucketKey = Tuple[MetricType, str, MetricUnit, MetricTagsInternal]

sentry_sdk/client.py

@@ -229,6 +229,15 @@ def _capture_envelope(envelope):
 
             self.session_flusher = SessionFlusher(capture_func=_capture_envelope)
 
+            if self.options.get("_experiments", {}).get("enable_metrics"):
+                from sentry_sdk.metrics import MetricsAggregator
+
+                self.metrics_aggregator = MetricsAggregator(
+                    capture_func=_capture_envelope
+                )
+            else:
+                self.metrics_aggregator = None
+
             max_request_body_size = ("always", "never", "small", "medium")
             if self.options["max_request_body_size"] not in max_request_body_size:
                 raise ValueError(
@@ -610,6 +619,8 @@ def close(
         if self.transport is not None:
             self.flush(timeout=timeout, callback=callback)
             self.session_flusher.kill()
+            if self.metrics_aggregator is not None:
+                self.metrics_aggregator.kill()
             if self.monitor:
                 self.monitor.kill()
             self.transport.kill()
@@ -632,6 +643,8 @@ def flush(
             if timeout is None:
                 timeout = self.options["shutdown_timeout"]
             self.session_flusher.flush()
+            if self.metrics_aggregator is not None:
+                self.metrics_aggregator.flush()
             self.transport.flush(timeout=timeout, callback=callback)
 
     def __enter__(self):

sentry_sdk/consts.py

@@ -41,6 +41,7 @@
             "profiler_mode": Optional[ProfilerMode],
             "otel_powered_performance": Optional[bool],
             "transport_zlib_compression_level": Optional[int],
+            "enable_metrics": Optional[bool],
         },
         total=False,
     )

sentry_sdk/envelope.py

@@ -260,6 +260,8 @@ def data_category(self):
             return "internal"
         elif ty == "profile":
             return "profile"
+        elif ty == "statsd":
+            return "statsd"
         else:
             return "default"