67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
import weakref
|
|
|
|
import sentry_sdk
|
|
from sentry_sdk.utils import ContextVar, logger
|
|
from sentry_sdk.integrations import Integration
|
|
from sentry_sdk.scope import add_global_event_processor
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Optional
|
|
|
|
from sentry_sdk._types import Event, Hint
|
|
|
|
|
|
class DedupeIntegration(Integration):
|
|
identifier = "dedupe"
|
|
|
|
def __init__(self):
|
|
# type: () -> None
|
|
self._last_seen = ContextVar("last-seen")
|
|
|
|
@staticmethod
|
|
def setup_once():
|
|
# type: () -> None
|
|
@add_global_event_processor
|
|
def processor(event, hint):
|
|
# type: (Event, Optional[Hint]) -> Optional[Event]
|
|
if hint is None:
|
|
return event
|
|
|
|
integration = sentry_sdk.get_client().get_integration(DedupeIntegration)
|
|
if integration is None:
|
|
return event
|
|
|
|
exc_info = hint.get("exc_info", None)
|
|
if exc_info is None:
|
|
return event
|
|
|
|
last_seen = integration._last_seen.get(None)
|
|
if last_seen is not None:
|
|
# last_seen is either a weakref or the original instance
|
|
last_seen = (
|
|
last_seen() if isinstance(last_seen, weakref.ref) else last_seen
|
|
)
|
|
|
|
exc = exc_info[1]
|
|
if last_seen is exc:
|
|
logger.info("DedupeIntegration dropped duplicated error event %s", exc)
|
|
return None
|
|
|
|
# we can only weakref non builtin types
|
|
try:
|
|
integration._last_seen.set(weakref.ref(exc))
|
|
except TypeError:
|
|
integration._last_seen.set(exc)
|
|
|
|
return event
|
|
|
|
@staticmethod
|
|
def reset_last_seen():
|
|
# type: () -> None
|
|
integration = sentry_sdk.get_client().get_integration(DedupeIntegration)
|
|
if integration is None:
|
|
return
|
|
|
|
integration._last_seen.set(None)
|