added redis and aiven ping

aiven_keep_alive_service.py

@@ -0,0 +1,80 @@
+import os
+import logging
+from dotenv import load_dotenv
+import psycopg2  # For PostgreSQL
+import pymysql  # For MySQL
+
+# Load environment variables
+load_dotenv()
+
+sql_uri = os.getenv("AIVEN_MYSQL_URI")
+postgres_uri = os.getenv("AIVEN_POSTGRES_URI")
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.StreamHandler(),
+        logging.FileHandler('keepalive.log')
+    ]
+)
+logger = logging.getLogger('KeepAlive')
+
+def ping_mysql():
+    """Ping MySQL database to keep connection alive"""
+    try:
+        from urllib.parse import urlparse, parse_qs
+        
+        # Parse the URI
+        parsed_uri = urlparse(sql_uri)
+        
+        # Extract connection parameters
+        host = parsed_uri.hostname
+        port = parsed_uri.port or 3306  # default MySQL port
+        user = parsed_uri.username
+        password = parsed_uri.password
+        database = parsed_uri.path.lstrip('/') or 'defaultdb'
+        
+        # Parse query parameters for SSL mode
+        query_params = parse_qs(parsed_uri.query)
+        ssl_mode = query_params.get('ssl-mode', ['REQUIRED'])[0]
+        
+        connection = pymysql.connect(
+            host=host,
+            port=port,
+            user=user,
+            password=password,
+            database=database,
+            ssl={'mode': ssl_mode} if ssl_mode else None
+        )
+        with connection.cursor() as cursor:
+            cursor.execute("SELECT 1")
+            result = cursor.fetchone()
+            logger.info(f"MySQL ping successful: {result}")
+        connection.close()
+    except Exception as e:
+        logger.error(f"MySQL ping failed: {str(e)}")
+
+def ping_postgres():
+    """Ping PostgreSQL database to keep connection alive"""
+    try:
+        connection = psycopg2.connect(postgres_uri)
+        cursor = connection.cursor()
+        cursor.execute("SELECT 1")
+        result = cursor.fetchone()
+        logger.info(f"PostgreSQL ping successful: {result}")
+        cursor.close()
+        connection.close()
+    except Exception as e:
+        logger.error(f"PostgreSQL ping failed: {str(e)}")
+
+
+
+def ping_aiven_projects():
+    """Ping Aiven projects to keep connections alive"""
+    logger.info("Starting Aiven service health check")
+    ping_mysql()
+    ping_postgres()
+    
+    

keep_projects_alive_service.py

@@ -1,8 +1,10 @@
 
 
+from aiven_keep_alive_service import ping_aiven_projects
 from mongodb_keep_alive_service import ping_mongodb_projects
 from neondb_postgres_keep_alive_service import ping_neondb_projects
 from pinecone_keep_alive_service import ping_all_pinecone_indexes
+from redis_keep_alive_service import ping_all_redis_projects
 from supabase_keep_alive_service import ping_all_supabase_projects
 
 
@@ -11,4 +13,6 @@ def ping_all_projects():
     ping_neondb_projects()
     ping_all_pinecone_indexes()
     ping_mongodb_projects()
+    ping_aiven_projects()
+    ping_all_redis_projects()
     

redis_keep_alive_service.py

@@ -0,0 +1,63 @@
+import os
+from dotenv import load_dotenv
+import redis
+import logging
+from datetime import datetime
+
+# Set up logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.FileHandler('redis_ping.log'),
+        logging.StreamHandler()
+    ]
+)
+logger = logging.getLogger(__name__)
+
+load_dotenv()
+
+REDIS_SERVICES = {
+    "redis_production": {
+        "host": os.getenv("REDIS_PRODUCTION_HOST"),
+        "port": os.getenv("REDIS_PRODUCTION_PORT"),
+        "password": os.getenv("REDIS_PRODUCTION_PASSWORD"),
+    },
+    "redis_development": {
+        "host": os.getenv("REDIS_DEVELOPMENT_HOST"),
+        "port": os.getenv("REDIS_DEVELOPMENT_PORT"),
+        "password": os.getenv("REDIS_DEVELOPMENT_PASSWORD"),
+    },
+}
+
+def ping_redis(redis_config):
+    try:
+        start_time = datetime.now()
+        r = redis.Redis(
+            host=redis_config["host"],
+            port=redis_config["port"],
+            password=redis_config["password"],
+            ssl=True if redis_config.get("ssl") else False,
+        )
+        response = r.ping()
+        latency = (datetime.now() - start_time).total_seconds() * 1000  # in milliseconds
+        logger.info(
+            f"Successfully pinged {redis_config['host']}:{redis_config['port']} - "
+            f"Response: {response}, Latency: {latency:.2f}ms"
+        )
+        return True
+    except Exception as e:
+        logger.error(f"Failed to ping {redis_config.get('host', 'unknown_host')}: {str(e)}")
+        return False
+
+def ping_all_redis_projects():
+    logger.info("Starting Redis service health check")
+    results = {}
+    
+    for service_name, config in REDIS_SERVICES.items():
+        logger.info(f"Pinging {service_name} at {config['host']}:{config['port']}")
+        results[service_name] = ping_redis(config)
+    
+    logger.info("Completed all Redis service checks")
+    return results
+

requirements.txt

@@ -8,4 +8,6 @@ twilio
 pymongo==4.11.2
 psycopg2-binary==2.9.10
 pinecone==6.0.2
-supabase==2.13.0
+supabase==2.13.0
+pymysql
+redis