Compare commits
1
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
6c6607dd49 |
+8
-22
@@ -8,8 +8,7 @@
|
|||||||
set -e
|
set -e
|
||||||
|
|
||||||
BACKUP_FILE='/var/lib/postgresql/data/backup.sql'
|
BACKUP_FILE='/var/lib/postgresql/data/backup.sql'
|
||||||
PGDATA_DIR='/var/lib/postgresql/data'
|
HBA='/var/lib/postgresql/data/pg_hba.conf'
|
||||||
HBA="${PGDATA_DIR}/pg_hba.conf"
|
|
||||||
export PGPASSWORD=$(cat "${POSTGRES_PASSWORD_FILE:-/run/secrets/db_password}")
|
export PGPASSWORD=$(cat "${POSTGRES_PASSWORD_FILE:-/run/secrets/db_password}")
|
||||||
DB_USER="${POSTGRES_USER:-postgres}"
|
DB_USER="${POSTGRES_USER:-postgres}"
|
||||||
DB_NAME="${POSTGRES_DB:-immich}"
|
DB_NAME="${POSTGRES_DB:-immich}"
|
||||||
@@ -19,41 +18,28 @@ function backup {
|
|||||||
}
|
}
|
||||||
|
|
||||||
function restore {
|
function restore {
|
||||||
# Reload pg_hba WITHOUT depending on a client connection — SIGHUP the postmaster directly. This is
|
|
||||||
# what guarantees the app is re-admitted at the end even if a psql connection isn't available.
|
|
||||||
reload_pg() {
|
|
||||||
local pid
|
|
||||||
pid=$(head -1 "${PGDATA_DIR}/postmaster.pid" 2>/dev/null || true)
|
|
||||||
if [ -n "$pid" ]; then kill -HUP "$pid" 2>/dev/null || true
|
|
||||||
else psql -U "$DB_USER" -d postgres -c "SELECT pg_reload_conf();" >/dev/null 2>&1 || true; fi
|
|
||||||
}
|
|
||||||
restore_hba() {
|
restore_hba() {
|
||||||
if [ -f "${HBA}.ccci.bak" ]; then
|
if [ -f "${HBA}.ccci.bak" ]; then
|
||||||
cat "${HBA}.ccci.bak" > "$HBA"
|
cat "${HBA}.ccci.bak" > "$HBA"
|
||||||
rm -f "${HBA}.ccci.bak"
|
rm -f "${HBA}.ccci.bak"
|
||||||
reload_pg
|
psql -U "$DB_USER" -d postgres -c "SELECT pg_reload_conf();" >/dev/null 2>&1 || true
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
# Always re-admit the app, even if any step below fails or the hook is signalled.
|
|
||||||
trap restore_hba EXIT INT TERM
|
trap restore_hba EXIT INT TERM
|
||||||
|
|
||||||
# Lock the networked app out for the duration of the restore: a local-trust-only pg_hba + reload
|
# Lock the networked app out for the restore: local-trust-only pg_hba + reload rejects all TCP
|
||||||
# makes postgres reject every TCP connection, so immich-server can't reconnect and re-run its own
|
# connections, then terminate the ones it already holds so DROP DATABASE can proceed.
|
||||||
# migrations concurrently with (and conflicting with) the import. Then terminate the connections it
|
|
||||||
# already holds so DROP DATABASE can proceed.
|
|
||||||
cp "$HBA" "${HBA}.ccci.bak"
|
cp "$HBA" "${HBA}.ccci.bak"
|
||||||
printf 'local all all trust\n' > "$HBA"
|
printf 'local all all trust\n' > "$HBA"
|
||||||
reload_pg
|
psql -U "$DB_USER" -d postgres -c "SELECT pg_reload_conf();"
|
||||||
psql -U "$DB_USER" -d postgres -c \
|
psql -U "$DB_USER" -d postgres -c \
|
||||||
"SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname='${DB_NAME}' AND pid<>pg_backend_pid();" >/dev/null
|
"SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname='${DB_NAME}' AND pid<>pg_backend_pid();"
|
||||||
|
|
||||||
psql -U "$DB_USER" -d postgres -c "DROP DATABASE ${DB_NAME} WITH (FORCE);" >/dev/null
|
psql -U "$DB_USER" -d postgres -c "DROP DATABASE ${DB_NAME} WITH (FORCE);"
|
||||||
createdb -U "$DB_USER" "$DB_NAME"
|
createdb -U "$DB_USER" "$DB_NAME"
|
||||||
# Rewrite the empty search_path the VectorChord dump sets so vector/vchord type + operator
|
|
||||||
# references resolve during the import (immich upstream restore procedure), then import alone.
|
|
||||||
gunzip -c "$BACKUP_FILE" \
|
gunzip -c "$BACKUP_FILE" \
|
||||||
| sed "s/SELECT pg_catalog.set_config('search_path', '', false);/SELECT pg_catalog.set_config('search_path', 'public, pg_catalog', true);/g" \
|
| sed "s/SELECT pg_catalog.set_config('search_path', '', false);/SELECT pg_catalog.set_config('search_path', 'public, pg_catalog', true);/g" \
|
||||||
| psql -U "$DB_USER" -d "$DB_NAME" --single-transaction --set ON_ERROR_STOP=on >/dev/null
|
| psql -U "$DB_USER" -d "$DB_NAME" --single-transaction --set ON_ERROR_STOP=on
|
||||||
|
|
||||||
restore_hba
|
restore_hba
|
||||||
trap - EXIT INT TERM
|
trap - EXIT INT TERM
|
||||||
|
|||||||
Reference in New Issue
Block a user