32 lines
806 B
Python
32 lines
806 B
Python
"""
|
|
Instrumentation for Redis Blaster (rb)
|
|
|
|
https://github.com/getsentry/rb
|
|
"""
|
|
|
|
from sentry_sdk.integrations.redis._sync_common import patch_redis_client
|
|
from sentry_sdk.integrations.redis.modules.queries import _set_db_data
|
|
|
|
|
|
def _patch_rb():
|
|
# type: () -> None
|
|
try:
|
|
import rb.clients # type: ignore
|
|
except ImportError:
|
|
pass
|
|
else:
|
|
patch_redis_client(
|
|
rb.clients.FanoutClient,
|
|
is_cluster=False,
|
|
set_db_data_fn=_set_db_data,
|
|
)
|
|
patch_redis_client(
|
|
rb.clients.MappingClient,
|
|
is_cluster=False,
|
|
set_db_data_fn=_set_db_data,
|
|
)
|
|
patch_redis_client(
|
|
rb.clients.RoutingClient,
|
|
is_cluster=False,
|
|
set_db_data_fn=_set_db_data,
|
|
)
|