83 lines
2.0 KiB
Bash
83 lines
2.0 KiB
Bash
#!/bin/sh
|
|
set -e
|
|
|
|
# Common initialization functions for PieFed containers
|
|
|
|
log() {
|
|
echo "[$(date +'%Y-%m-%d %H:%M:%S')] $1"
|
|
}
|
|
|
|
# Wait for database to be available
|
|
wait_for_db() {
|
|
log "Waiting for database connection..."
|
|
until python -c "
|
|
import psycopg2
|
|
import os
|
|
from urllib.parse import urlparse
|
|
|
|
try:
|
|
# Parse DATABASE_URL
|
|
database_url = os.environ.get('DATABASE_URL', '')
|
|
if not database_url:
|
|
raise Exception('DATABASE_URL not set')
|
|
|
|
# Parse the URL to extract connection details
|
|
parsed = urlparse(database_url)
|
|
conn = psycopg2.connect(
|
|
host=parsed.hostname,
|
|
port=parsed.port or 5432,
|
|
database=parsed.path[1:], # Remove leading slash
|
|
user=parsed.username,
|
|
password=parsed.password
|
|
)
|
|
conn.close()
|
|
print('Database connection successful')
|
|
except Exception as e:
|
|
print(f'Database connection failed: {e}')
|
|
exit(1)
|
|
" 2>/dev/null; do
|
|
log "Database not ready, waiting 2 seconds..."
|
|
sleep 2
|
|
done
|
|
log "Database connection established"
|
|
}
|
|
|
|
# Wait for Redis to be available
|
|
wait_for_redis() {
|
|
log "Waiting for Redis connection..."
|
|
until python -c "
|
|
import redis
|
|
import os
|
|
|
|
try:
|
|
cache_redis_url = os.environ.get('CACHE_REDIS_URL', '')
|
|
if cache_redis_url:
|
|
r = redis.from_url(cache_redis_url)
|
|
else:
|
|
# Fallback to separate host/port for backwards compatibility
|
|
r = redis.Redis(host='redis', port=6379, password=os.environ.get('REDIS_PASSWORD', ''))
|
|
r.ping()
|
|
print('Redis connection successful')
|
|
except Exception as e:
|
|
print(f'Redis connection failed: {e}')
|
|
exit(1)
|
|
" 2>/dev/null; do
|
|
log "Redis not ready, waiting 2 seconds..."
|
|
sleep 2
|
|
done
|
|
log "Redis connection established"
|
|
}
|
|
|
|
# Common startup sequence
|
|
common_startup() {
|
|
log "Starting PieFed common initialization..."
|
|
|
|
# Change to application directory
|
|
cd /app
|
|
|
|
# Wait for dependencies
|
|
wait_for_db
|
|
wait_for_redis
|
|
|
|
log "Common initialization completed"
|
|
} |