| set -euo pipefail |
|
|
| PG_USER="asp_user" |
| PG_PASSWORD="asp_password" |
| PG_DB="asp" |
| BACKUP_DIR="/workspace/pg-backups" |
| BACKUP_FILE="${BACKUP_DIR}/latest.dump" |
| BACKUP_INTERVAL_SECONDS="${BACKUP_INTERVAL_SECONDS:-300}" |
|
|
| if [ "$(id -u)" -eq 0 ]; then |
| PGRUN="runuser -u postgres --" |
| else |
| PGRUN="sudo -u postgres" |
| fi |
|
|
| if [ ! -d /workspace ]; then |
| echo "!! /workspace not found. This script expects a RunPod network volume" |
| echo "mounted at /workspace. Aborting." |
| exit 1 |
| fi |
| mkdir -p "$BACKUP_DIR" |
|
|
| if ! command -v psql >/dev/null 2>&1; then |
| echo "Installing PostgreSQL" |
| apt-get update -y |
| apt-get install -y postgresql |
| fi |
|
|
| PG_VERSION="$(ls /etc/postgresql | sort -V | tail -n1)" |
| PG_CLUSTER="main" |
| NATIVE_DATA_DIR="/var/lib/postgresql/${PG_VERSION}/${PG_CLUSTER}" |
| echo "Found PostgreSQL ${PG_VERSION}, cluster '${PG_CLUSTER}'" |
|
|
| NEEDS_REPAIR=false |
| if [ -L "$NATIVE_DATA_DIR" ]; then |
| NEEDS_REPAIR=true |
| elif [ -d "$NATIVE_DATA_DIR" ] && [ "$(stat -c %U "$NATIVE_DATA_DIR" 2>/dev/null)" != "postgres" ]; then |
| NEEDS_REPAIR=true |
| fi |
| if [ "$NEEDS_REPAIR" = true ]; then |
| echo "Found a broken/root-owned data directory" |
| echo "recreating the local cluster from scratch" |
| pg_dropcluster --stop "${PG_VERSION}" "${PG_CLUSTER}" 2>/dev/null || true |
| rm -rf "$NATIVE_DATA_DIR" |
| rm -rf "/etc/postgresql/${PG_VERSION}/${PG_CLUSTER}" |
| pg_createcluster "${PG_VERSION}" "${PG_CLUSTER}" |
| fi |
|
|
| echo "Starting PostgreSQL..." |
| service postgresql start || pg_ctlcluster "${PG_VERSION}" "${PG_CLUSTER}" start |
| for i in $(seq 1 30); do |
| if ${PGRUN} pg_isready -q; then break; fi |
| sleep 1 |
| done |
|
|
| FRESH_CLUSTER=false |
| ROLE_EXISTS="$(${PGRUN} psql -tAc "SELECT 1 FROM pg_roles WHERE rolname='${PG_USER}'")" |
| if [ "$ROLE_EXISTS" != "1" ]; then |
| echo "Creating role '${PG_USER}' and database '${PG_DB}'." |
| ${PGRUN} psql -c "CREATE ROLE ${PG_USER} LOGIN PASSWORD '${PG_PASSWORD}';" |
| ${PGRUN} createdb -O "${PG_USER}" "${PG_DB}" |
| FRESH_CLUSTER=true |
| else |
| echo "Role '${PG_USER}' already exists; Skipping creation" |
| fi |
|
|
| if [ "$FRESH_CLUSTER" = true ] && [ -f "$BACKUP_FILE" ]; then |
| echo "Found existing backup at ${BACKUP_FILE} — restoring" |
| ${PGRUN} pg_restore -d "${PG_DB}" --clean --if-exists "$BACKUP_FILE" \ |
| && echo " Restore complete." \ |
| || echo "!! Restore reported errors. Check output above." |
| else |
| echo "No restore needed (either not a fresh cluster, or no backup found yet)." |
| fi |
|
|
| if ! pgrep -f "pg_dump.*${PG_DB}.*autobackup-loop" >/dev/null 2>&1; then |
| echo "Starting background auto-backup loop (every ${BACKUP_INTERVAL_SECONDS}s)..." |
| nohup bash -c " |
| # marker string 'autobackup-loop' below is just so pgrep can find this loop |
| while true; do |
| sleep ${BACKUP_INTERVAL_SECONDS} |
| ${PGRUN} pg_dump -Fc '${PG_DB}' -f '${BACKUP_FILE}.tmp' \ |
| && mv '${BACKUP_FILE}.tmp' '${BACKUP_FILE}' # autobackup-loop |
| done |
| " > "${BACKUP_DIR}/autobackup.log" 2>&1 & |
| disown |
| fi |
|
|
| echo "" |
| echo "To take a manual backup right now, run: bash backup_postgres.sh" |
|
|
