rb.py 806 B

1234567891011121314151617181920212223242526272829303132
  1. """
  2. Instrumentation for Redis Blaster (rb)
  3. https://github.com/getsentry/rb
  4. """
  5. from sentry_sdk.integrations.redis._sync_common import patch_redis_client
  6. from sentry_sdk.integrations.redis.modules.queries import _set_db_data
  7. def _patch_rb():
  8. # type: () -> None
  9. try:
  10. import rb.clients # type: ignore
  11. except ImportError:
  12. pass
  13. else:
  14. patch_redis_client(
  15. rb.clients.FanoutClient,
  16. is_cluster=False,
  17. set_db_data_fn=_set_db_data,
  18. )
  19. patch_redis_client(
  20. rb.clients.MappingClient,
  21. is_cluster=False,
  22. set_db_data_fn=_set_db_data,
  23. )
  24. patch_redis_client(
  25. rb.clients.RoutingClient,
  26. is_cluster=False,
  27. set_db_data_fn=_set_db_data,
  28. )