155 lines
5.6 KiB
Python
155 lines
5.6 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Healthcheck script for Puppeteer API
|
|
Monitors the API endpoint and restarts the container if it fails
|
|
"""
|
|
|
|
import os
|
|
import time
|
|
import logging
|
|
import requests
|
|
import docker
|
|
from urllib.parse import quote
|
|
from datetime import datetime
|
|
|
|
# Configure logging
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(levelname)s - %(message)s',
|
|
handlers=[
|
|
logging.StreamHandler(),
|
|
logging.FileHandler('/app/healthcheck.log')
|
|
]
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class PuppeteerHealthcheck:
|
|
def __init__(self):
|
|
self.base_url = os.getenv('BASE_URL', 'https://puppeteer.workwithkora.com')
|
|
self.test_url = os.getenv('TEST_URL', 'https://www.google.com')
|
|
self.api_key = os.getenv('API_KEY', 'Q7Sd#hhFkyHy*T')
|
|
self.target_container = os.getenv('TARGET_CONTAINER', 'puppeteer-api')
|
|
self.check_interval = int(os.getenv('CHECK_INTERVAL', '60'))
|
|
|
|
# Initialize Docker client
|
|
try:
|
|
self.docker_client = docker.from_env()
|
|
logger.info("Docker client initialized successfully")
|
|
except Exception as e:
|
|
logger.error(f"Failed to initialize Docker client: {e}")
|
|
raise
|
|
|
|
logger.info(f"Healthcheck initialized with:")
|
|
logger.info(f" Base URL: {self.base_url}")
|
|
logger.info(f" Test URL: {self.test_url}")
|
|
logger.info(f" Target Container: {self.target_container}")
|
|
logger.info(f" Check Interval: {self.check_interval} seconds")
|
|
|
|
def perform_health_check(self):
|
|
"""Perform the health check by making a request to the puppeteer API"""
|
|
try:
|
|
# Construct the URL with the test URL as a parameter
|
|
encoded_test_url = quote(self.test_url, safe='')
|
|
url = f"{self.base_url}/?url={encoded_test_url}"
|
|
|
|
headers = {
|
|
'x-api-key': self.api_key,
|
|
'User-Agent': 'Puppeteer-Healthcheck/1.0'
|
|
}
|
|
|
|
logger.info(f"Performing health check: {url}")
|
|
|
|
response = requests.get(
|
|
url,
|
|
headers=headers,
|
|
timeout=30,
|
|
allow_redirects=True
|
|
)
|
|
|
|
if response.status_code == 200:
|
|
logger.info("Health check passed - API is responding correctly")
|
|
return True
|
|
else:
|
|
logger.warning(f"Health check failed - Status code: {response.status_code}")
|
|
logger.warning(f"Response: {response.text[:200]}...")
|
|
return False
|
|
|
|
except requests.exceptions.RequestException as e:
|
|
logger.error(f"Health check failed - Request error: {e}")
|
|
return False
|
|
except Exception as e:
|
|
logger.error(f"Health check failed - Unexpected error: {e}")
|
|
return False
|
|
|
|
def restart_container(self):
|
|
"""Restart the target puppeteer-api container"""
|
|
try:
|
|
logger.info(f"Attempting to restart container: {self.target_container}")
|
|
|
|
# Get the container
|
|
container = self.docker_client.containers.get(self.target_container)
|
|
|
|
# Restart the container
|
|
container.restart(timeout=30)
|
|
|
|
logger.info(f"Successfully restarted container: {self.target_container}")
|
|
return True
|
|
|
|
except docker.errors.NotFound:
|
|
logger.error(f"Container not found: {self.target_container}")
|
|
return False
|
|
except docker.errors.APIError as e:
|
|
logger.error(f"Docker API error while restarting container: {e}")
|
|
return False
|
|
except Exception as e:
|
|
logger.error(f"Unexpected error while restarting container: {e}")
|
|
return False
|
|
|
|
def run(self):
|
|
"""Main loop for the healthcheck"""
|
|
logger.info("Starting Puppeteer API healthcheck service")
|
|
|
|
consecutive_failures = 0
|
|
max_consecutive_failures = 3 # Restart after 3 consecutive failures
|
|
|
|
while True:
|
|
try:
|
|
# Perform health check
|
|
if self.perform_health_check():
|
|
consecutive_failures = 0
|
|
logger.info("Health check passed - resetting failure counter")
|
|
else:
|
|
consecutive_failures += 1
|
|
logger.warning(f"Health check failed - consecutive failures: {consecutive_failures}")
|
|
|
|
# Restart container if we've had too many consecutive failures
|
|
if consecutive_failures >= max_consecutive_failures:
|
|
logger.error(f"Health check failed {consecutive_failures} times consecutively - restarting container")
|
|
if self.restart_container():
|
|
consecutive_failures = 0
|
|
logger.info("Container restarted successfully - resetting failure counter")
|
|
else:
|
|
logger.error("Failed to restart container")
|
|
|
|
# Wait before next check
|
|
logger.info(f"Waiting {self.check_interval} seconds before next health check...")
|
|
time.sleep(self.check_interval)
|
|
|
|
except KeyboardInterrupt:
|
|
logger.info("Healthcheck service stopped by user")
|
|
break
|
|
except Exception as e:
|
|
logger.error(f"Unexpected error in main loop: {e}")
|
|
time.sleep(self.check_interval)
|
|
|
|
def main():
|
|
"""Main entry point"""
|
|
try:
|
|
healthcheck = PuppeteerHealthcheck()
|
|
healthcheck.run()
|
|
except Exception as e:
|
|
logger.error(f"Failed to start healthcheck service: {e}")
|
|
exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
main() |