mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-15 08:53:43 +08:00
firestar5683
e52e1a6a86
Implement Sentry scanning, notification state and worker cancellation through explicit policies and preferences.
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
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
|
|
|
|
exc = exc_info[1]
|
|
if integration._last_seen.get(None) is exc:
|
|
logger.info("DedupeIntegration dropped duplicated error event %s", exc)
|
|
return None
|
|
|
|
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)
|