feat(api): Added error_sampler option

sentry_sdk/client.py

@@ -456,10 +456,11 @@ def _should_sample_error(
         event,  # type: Event
     ):
         # type: (...) -> bool
-        not_in_sample_rate = (
-            self.options["sample_rate"] < 1.0
-            and random.random() >= self.options["sample_rate"]
-        )
+        try:
+            sample_rate = self.options["issues_sampler"](event)
+        except (KeyError, TypeError):
+            sample_rate = self.options["sample_rate"]
+        not_in_sample_rate = sample_rate < 1.0 and random.random() >= sample_rate
         if not_in_sample_rate:
             # because we will not sample this event, record a "lost event".
             if self.transport:

sentry_sdk/consts.py

@@ -261,6 +261,7 @@ def __init__(
         event_scrubber=None,  # type: Optional[sentry_sdk.scrubber.EventScrubber]
         max_value_length=DEFAULT_MAX_VALUE_LENGTH,  # type: int
         enable_backpressure_handling=True,  # type: bool
+        issues_sampler=None,  # type: Optional[Callable[[Event], Union[float, bool]]]
     ):
         # type: (...) -> None
         pass

tests/test_client.py

@@ -25,6 +25,12 @@
 from sentry_sdk.utils import logger
 from sentry_sdk.serializer import MAX_DATABAG_BREADTH
 from sentry_sdk.consts import DEFAULT_MAX_BREADCRUMBS, DEFAULT_MAX_VALUE_LENGTH
+from sentry_sdk._types import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+    from typing import Any, Optional, Union
+    from sentry_sdk._types import Event
 
 try:
     from unittest import mock  # python 3.3 and above
@@ -1196,3 +1202,97 @@ def test_debug_option(
         assert "something is wrong" in caplog.text
     else:
         assert "something is wrong" not in caplog.text
+
+
+class IssuesSamplerTestConfig:
+    def __init__(
+        self,
+        expected_events,
+        sampler_function=None,
+        sample_rate=None,
+        exception_to_raise=Exception,
+    ):
+        # type: (int, Optional[Callable[[Event], Union[float, bool]]], Optional[float], type[Exception]) -> None
+        self.sampler_function_mock = (
+            None
+            if sampler_function is None
+            else mock.MagicMock(side_effect=sampler_function)
+        )
+        self.expected_events = expected_events
+        self.sample_rate = sample_rate
+        self.exception_to_raise = exception_to_raise
+
+    def init_sdk(self, sentry_init):
+        # type: (Callable[[*Any], None]) -> None
+        sentry_init(
+            issues_sampler=self.sampler_function_mock, sample_rate=self.sample_rate
+        )
+
+    def raise_exception(self):
+        # type: () -> None
+        raise self.exception_to_raise()
+
+
+@mock.patch("sentry_sdk.client.random.random", return_value=0.618)
+@pytest.mark.parametrize(
+    "test_config",
+    (
+        # Baseline test with issues_sampler only, both floats and bools
+        IssuesSamplerTestConfig(sampler_function=lambda _: 1.0, expected_events=1),
+        IssuesSamplerTestConfig(sampler_function=lambda _: 0.7, expected_events=1),
+        IssuesSamplerTestConfig(sampler_function=lambda _: 0.6, expected_events=0),
+        IssuesSamplerTestConfig(sampler_function=lambda _: 0.0, expected_events=0),
+        IssuesSamplerTestConfig(sampler_function=lambda _: True, expected_events=1),
+        IssuesSamplerTestConfig(sampler_function=lambda _: False, expected_events=0),
+        # Baseline test with sample_rate only
+        IssuesSamplerTestConfig(sample_rate=1.0, expected_events=1),
+        IssuesSamplerTestConfig(sample_rate=0.7, expected_events=1),
+        IssuesSamplerTestConfig(sample_rate=0.6, expected_events=0),
+        IssuesSamplerTestConfig(sample_rate=0.0, expected_events=0),
+        # issues_sampler takes precedence over sample_rate
+        IssuesSamplerTestConfig(
+            sampler_function=lambda _: 1.0, sample_rate=0.0, expected_events=1
+        ),
+        IssuesSamplerTestConfig(
+            sampler_function=lambda _: 0.0, sample_rate=1.0, expected_events=0
+        ),
+        # Different sample rates based on exception
+        IssuesSamplerTestConfig(
+            sampler_function=lambda event: {
+                "ZeroDivisionError": 1.0,
+                "AttributeError": 0.0,
+            }[event["exception"]["values"][0]["type"]],
+            exception_to_raise=ZeroDivisionError,
+            expected_events=1,
+        ),
+        IssuesSamplerTestConfig(
+            sampler_function=lambda event: {
+                "ZeroDivisionError": 1.0,
+                "AttributeError": 0.0,
+            }[event["exception"]["values"][0]["type"]],
+            exception_to_raise=AttributeError,
+            expected_events=0,
+        ),
+    ),
+)
+def test_issues_sampler(_, sentry_init, capture_events, test_config):
+    test_config.init_sdk(sentry_init)
+
+    events = capture_events()
+
+    try:
+        test_config.raise_exception()
+    except Exception:
+        capture_exception()
+
+    assert len(events) == test_config.expected_events
+
+    try:
+        assert test_config.sampler_function_mock.call_count == 1
+
+        # Ensure one argument (the event) was passed to the sampler function
+        assert len(test_config.sampler_function_mock.call_args[0]) == 1
+    except AttributeError:
+        # sampler_function_mock should be None in this case,
+        # but let's double-check to ensure the test is working correctly
+        assert test_config.sampler_function_mock is None