1. import redis
  2. import logging
  3. import os
  5. # Configure logging
  6. logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
  8. class MessageQueueComponents:
  9. def __init__(self, redis_host='localhost', redis_port=6379):
  10. """
  11. Initializes components for message queue diagnostics with fallback logic.
  12. """
  13. self.redis = self._create_redis_connection(redis_host, redis_port)
  14. self.logger = logging.getLogger(__name__)
  16. def _create_redis_connection(self, host, port):
  17. """
  18. Creates a Redis connection with fallback.
  19. """
  20. try:
  21. redis_conn = redis.Redis(host=host, port=port, db=0)
  22. redis_conn.ping() # Test connection
  23. self.logger.info("Successfully connected to Redis.")
  24. return redis_conn
  25. except redis.exceptions.ConnectionError as e:
  26. self.logger.error(f"Failed to connect to Redis at {host}:{port}. Attempting fallback...")
  27. # Attempt to connect with default values
  28. try:
  29. redis_conn = redis.Redis(host='localhost', port=6379, db=0)
  30. redis_conn.ping()
  31. self.logger.info("Successfully connected to Redis using default settings after fallback.")
  32. return redis_conn
  33. except redis.exceptions.ConnectionError as fallback_e:
  34. self.logger.error(f"Failed to connect to Redis even with default settings. Using mock object.")
  35. return MockRedis() # Return a mock object if all fails
  37. def get(self, key):
  38. """
  39. Retrieves a value from the queue. Uses redis.
  40. """
  41. try:
  42. return self.redis.get(key)
  43. except Exception as e:
  44. self.logger.error(f"Error getting from Redis: {e}")
  45. return None
  47. def set(self, key, value):
  48. """
  49. Sets a value in the queue. Uses redis.
  50. """
  51. try:
  52. self.redis.set(key, value)
  53. except Exception as e:
  54. self.logger.error(f"Error setting in Redis: {e}")
  56. def delete(self, key):
  57. """
  58. Deletes a key from the queue. Uses redis.
  59. """
  60. try:
  61. self.redis.delete(key)
  62. except Exception as e:
  63. self.logger.error(f"Error deleting from Redis: {e}")
  65. class MockRedis:
  66. """
  67. A mock Redis object for testing and fallback scenarios.
  68. """
  69. def __init__(self):
  70. self.data = {}
  72. def get(self, key):
  73. return self.data.get(key)
  75. def set(self, key, value):
  76. self.data[key] = value
  78. def delete(self, key):
  79. if key in self.data:
  80. del self.data[key]

Add your comment