mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-11-21 04:36:53 -06:00
This alters the retry/backoff logic in the init-wait-for-db script to be more optimistic about database availability. During regular deployment and operations of paperless-ngx, it's common to restart the application server with the database instance already running, so we should optimize for this case. Instead of unconditionally delaying 5 seconds between each connection attempt, start with a minimum delay of 1 second and increase the delay linearly with each attempt, maxing out at 10 seconds. This makes the retry count-based failure mode less practical, so instead we just use a timeout-based approach.* *NOTE: the original implementation would have an effective timeout of 25s. This alters the behavior to 60s. Additionally, this removes an unnecessary 5s delay that was injected in the postgres case. The script uses a more comprehensive connection check for postgres than it does mariadb, so if anything this 5s delay after getting an "ok" response from the DB was extra unnecessary in the postgres case.
67 lines
1.6 KiB
Bash
Executable File
67 lines
1.6 KiB
Bash
Executable File
#!/command/with-contenv /usr/bin/bash
|
|
# shellcheck shell=bash
|
|
# vim: set ft=bash ts=4 sw=4 sts=4 et :
|
|
|
|
set -euo pipefail
|
|
|
|
declare -r LOG_PREFIX="[init-db-wait]"
|
|
|
|
declare -ri TIMEOUT=60
|
|
declare -i ATTEMPT=0
|
|
declare -i DELAY=0
|
|
declare -i STARTED_AT=${EPOCHSECONDS:?EPOCHSECONDS var unset}
|
|
|
|
delay_next_attempt() {
|
|
local -i elapsed=$(( EPOCHSECONDS - STARTED_AT ))
|
|
local -ri remaining=$(( TIMEOUT - elapsed ))
|
|
|
|
if (( remaining <= 0 )); then
|
|
echo "${LOG_PREFIX} Unable to connect after $elapsed seconds."
|
|
exit 1
|
|
fi
|
|
|
|
DELAY+=1
|
|
|
|
# clamp to remaining time
|
|
if (( DELAY > remaining )); then
|
|
DELAY=$remaining
|
|
fi
|
|
|
|
ATTEMPT+=1
|
|
echo "${LOG_PREFIX} Attempt $ATTEMPT failed! Trying again in $DELAY seconds..."
|
|
sleep "$DELAY"
|
|
}
|
|
|
|
wait_for_postgres() {
|
|
echo "${LOG_PREFIX} Waiting for PostgreSQL to start..."
|
|
|
|
local -r host="${PAPERLESS_DBHOST:-localhost}"
|
|
local -r port="${PAPERLESS_DBPORT:-5432}"
|
|
local -r user="${PAPERLESS_DBUSER:-paperless}"
|
|
|
|
while ! pg_isready -h "${host}" -p "${port}" --username "${user}"; do
|
|
delay_next_attempt
|
|
done
|
|
echo "${LOG_PREFIX} Connected to PostgreSQL"
|
|
}
|
|
|
|
wait_for_mariadb() {
|
|
echo "${LOG_PREFIX} Waiting for MariaDB to start..."
|
|
|
|
local -r host="${PAPERLESS_DBHOST:-localhost}"
|
|
local -r port="${PAPERLESS_DBPORT:-3306}"
|
|
|
|
while ! true > "/dev/tcp/$host/$port"; do
|
|
delay_next_attempt
|
|
done
|
|
echo "${LOG_PREFIX} Connected to MariaDB"
|
|
}
|
|
|
|
if [[ "${PAPERLESS_DBENGINE:-}" == "mariadb" ]]; then
|
|
wait_for_mariadb
|
|
elif [[ -n "${PAPERLESS_DBHOST:-}" ]]; then
|
|
wait_for_postgres
|
|
fi
|
|
|
|
echo "${LOG_PREFIX} Database is ready"
|