File size: 3,075 Bytes
6dde7dd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | 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"
|