mirror of
https://github.com/ahmadk953/poixpixel-discord-bot.git
synced 2025-06-22 06:04:20 +00:00
203 lines
No EOL
8.3 KiB
Bash
203 lines
No EOL
8.3 KiB
Bash
#!/bin/sh
|
|
# Based on https://raw.githubusercontent.com/brainsam/pgbouncer/master/entrypoint.sh
|
|
# and https://raw.githubusercontent.com/edoburu/docker-pgbouncer/master/entrypoint.sh
|
|
|
|
set -e
|
|
|
|
# Here are some parameters. See all on
|
|
# https://pgbouncer.github.io/config.html
|
|
|
|
PG_CONFIG_DIR=/etc/pgbouncer
|
|
PG_CONFIG_FILE="${PG_CONFIG_DIR}/pgbouncer.ini"
|
|
_AUTH_FILE="${AUTH_FILE:-$PG_CONFIG_DIR/userlist.txt}"
|
|
|
|
# Workaround userlist.txt missing issue
|
|
# https://github.com/edoburu/docker-pgbouncer/issues/33
|
|
if [ ! -e "${_AUTH_FILE}" ]; then
|
|
touch "${_AUTH_FILE}"
|
|
fi
|
|
|
|
# Extract all info from a given URL. Sets variables because shell functions can't return multiple values.
|
|
#
|
|
# Parameters:
|
|
# - The url we should parse
|
|
# Returns (sets variables): DB_USER, DB_PASSWORD, DB_HOST, DB_PORT, DB_NAME
|
|
function parse_url() {
|
|
# Thanks to https://stackoverflow.com/a/17287984/146289
|
|
|
|
# Allow to pass values like dj-database-url / django-environ accept
|
|
proto="$(echo $1 | grep :// | sed -e's,^\(.*://\).*,\1,g')"
|
|
url="$(echo $1 | sed -e s,$proto,,g)"
|
|
|
|
# extract the user and password (if any)
|
|
userpass="$(echo $url | grep @ | sed -r 's/^(.*)@([^@]*)$/\1/')"
|
|
DB_PASSWORD="$(echo $userpass | grep : | cut -d: -f2)"
|
|
if [ -n "${DB_PASSWORD}" ]; then
|
|
DB_USER="$(echo $userpass | grep : | cut -d: -f1)"
|
|
else
|
|
DB_USER="${userpass}"
|
|
fi
|
|
|
|
# extract the host -- updated
|
|
hostport=`echo $url | sed -e s,$userpass@,,g | cut -d/ -f1`
|
|
port=`echo $hostport | grep : | cut -d: -f2`
|
|
if [ -n "$port" ]; then
|
|
DB_HOST=`echo $hostport | grep : | cut -d: -f1`
|
|
DB_PORT="${port}"
|
|
else
|
|
DB_HOST="${hostport}"
|
|
fi
|
|
|
|
DB_NAME="$(echo $url | grep / | cut -d/ -f2-)"
|
|
}
|
|
|
|
# Grabs variables set by `parse_url` and adds them to the userlist if not already set in there.
|
|
function generate_userlist_if_needed() {
|
|
if [ -n "${DB_USER}" -a -n "${DB_PASSWORD}" -a -e "${_AUTH_FILE}" ] && ! grep -q "^\"${DB_USER}\"" "${_AUTH_FILE}"; then
|
|
if [ "${AUTH_TYPE}" == "plain" ] || [ "${AUTH_TYPE}" == "scram-sha-256" ]; then
|
|
pass="${DB_PASSWORD}"
|
|
else
|
|
pass="md5$(echo -n "${DB_PASSWORD}${DB_USER}" | md5sum | cut -f 1 -d ' ')"
|
|
fi
|
|
echo "\"${DB_USER}\" \"${pass}\"" >> "${_AUTH_FILE}"
|
|
echo "Wrote authentication credentials for '${DB_USER}' to ${_AUTH_FILE}"
|
|
fi
|
|
}
|
|
|
|
# Grabs variables set by `parse_url` and adds them to the PG config file as a database entry.
|
|
function generate_config_db_entry() {
|
|
printf "\
|
|
${DB_NAME:-*} = host=${DB_HOST:?"Setup pgbouncer config error! You must set DB_HOST env"} \
|
|
port=${DB_PORT:-5432} auth_user=${DB_USER:-postgres}
|
|
${CLIENT_ENCODING:+client_encoding = ${CLIENT_ENCODING}\n}\
|
|
" >> "${PG_CONFIG_FILE}"
|
|
}
|
|
|
|
# Write the password with MD5 encryption, to avoid printing it during startup.
|
|
# Notice that `docker inspect` will show unencrypted env variables.
|
|
if [ -n "${DATABASE_URLS}" ]; then
|
|
echo "${DATABASE_URLS}" | tr , '\n' | while read url; do
|
|
parse_url "$url"
|
|
generate_userlist_if_needed
|
|
done
|
|
else
|
|
if [ -n "${DATABASE_URL}" ]; then
|
|
parse_url "${DATABASE_URL}"
|
|
fi
|
|
generate_userlist_if_needed
|
|
fi
|
|
|
|
if [ ! -f "${PG_CONFIG_FILE}" ]; then
|
|
echo "Creating pgbouncer config in ${PG_CONFIG_DIR}"
|
|
|
|
# Config file is in "ini" format. Section names are between "[" and "]".
|
|
# Lines starting with ";" or "#" are taken as comments and ignored.
|
|
# The characters ";" and "#" are not recognized when they appear later in the line.
|
|
printf "\
|
|
################## Auto generated ##################
|
|
[databases]
|
|
" > "${PG_CONFIG_FILE}"
|
|
|
|
if [ -n "$DATABASE_URLS" ]; then
|
|
echo "$DATABASE_URLS" | tr , '\n' | while read url; do
|
|
parse_url "$url"
|
|
generate_config_db_entry
|
|
done
|
|
else
|
|
if [ -n "$DATABASE_URL" ]; then
|
|
parse_url "$DATABASE_URL"
|
|
fi
|
|
generate_config_db_entry
|
|
fi
|
|
|
|
printf "\
|
|
[pgbouncer]
|
|
listen_addr = ${LISTEN_ADDR:-0.0.0.0}
|
|
listen_port = ${LISTEN_PORT:-5432}
|
|
unix_socket_dir = ${UNIX_SOCKET_DIR}
|
|
user = pgbouncer
|
|
auth_file = ${_AUTH_FILE}
|
|
${AUTH_HBA_FILE:+auth_hba_file = ${AUTH_HBA_FILE}\n}\
|
|
auth_type = ${AUTH_TYPE:-md5}
|
|
${AUTH_USER:+auth_user = ${AUTH_USER}\n}\
|
|
${AUTH_QUERY:+auth_query = ${AUTH_QUERY}\n}\
|
|
${AUTH_DBNAME:+auth_dbname = ${AUTH_DBNAME}\n}\
|
|
${POOL_MODE:+pool_mode = ${POOL_MODE}\n}\
|
|
${MAX_CLIENT_CONN:+max_client_conn = ${MAX_CLIENT_CONN}\n}\
|
|
${POOL_SIZE:+pool_size = ${POOL_SIZE}\n}\
|
|
${DEFAULT_POOL_SIZE:+default_pool_size = ${DEFAULT_POOL_SIZE}\n}\
|
|
${MIN_POOL_SIZE:+min_pool_size = ${MIN_POOL_SIZE}\n}\
|
|
${RESERVE_POOL_SIZE:+reserve_pool_size = ${RESERVE_POOL_SIZE}\n}\
|
|
${RESERVE_POOL_TIMEOUT:+reserve_pool_timeout = ${RESERVE_POOL_TIMEOUT}\n}\
|
|
${MAX_DB_CONNECTIONS:+max_db_connections = ${MAX_DB_CONNECTIONS}\n}\
|
|
${MAX_USER_CONNECTIONS:+max_user_connections = ${MAX_USER_CONNECTIONS}\n}\
|
|
${SERVER_ROUND_ROBIN:+server_round_robin = ${SERVER_ROUND_ROBIN}\n}\
|
|
ignore_startup_parameters = ${IGNORE_STARTUP_PARAMETERS:-extra_float_digits}
|
|
${DISABLE_PQEXEC:+disable_pqexec = ${DISABLE_PQEXEC}\n}\
|
|
${APPLICATION_NAME_ADD_HOST:+application_name_add_host = ${APPLICATION_NAME_ADD_HOST}\n}\
|
|
${TIMEZONE:+timezone = ${TIMEZONE}\n}\
|
|
${MAX_PREPARED_STATEMENTS:+max_prepared_statements = ${MAX_PREPARED_STATEMENTS}\n}\
|
|
|
|
# Log settings
|
|
${LOG_CONNECTIONS:+log_connections = ${LOG_CONNECTIONS}\n}\
|
|
${LOG_DISCONNECTIONS:+log_disconnections = ${LOG_DISCONNECTIONS}\n}\
|
|
${LOG_POOLER_ERRORS:+log_pooler_errors = ${LOG_POOLER_ERRORS}\n}\
|
|
${LOG_STATS:+log_stats = ${LOG_STATS}\n}\
|
|
${STATS_PERIOD:+stats_period = ${STATS_PERIOD}\n}\
|
|
${VERBOSE:+verbose = ${VERBOSE}\n}\
|
|
admin_users = ${ADMIN_USERS:-postgres}
|
|
${STATS_USERS:+stats_users = ${STATS_USERS}\n}\
|
|
|
|
# Connection sanity checks, timeouts
|
|
${SERVER_RESET_QUERY:+server_reset_query = ${SERVER_RESET_QUERY}\n}\
|
|
${SERVER_RESET_QUERY_ALWAYS:+server_reset_query_always = ${SERVER_RESET_QUERY_ALWAYS}\n}\
|
|
${SERVER_CHECK_DELAY:+server_check_delay = ${SERVER_CHECK_DELAY}\n}\
|
|
${SERVER_CHECK_QUERY:+server_check_query = ${SERVER_CHECK_QUERY}\n}\
|
|
${SERVER_LIFETIME:+server_lifetime = ${SERVER_LIFETIME}\n}\
|
|
${SERVER_IDLE_TIMEOUT:+server_idle_timeout = ${SERVER_IDLE_TIMEOUT}\n}\
|
|
${SERVER_CONNECT_TIMEOUT:+server_connect_timeout = ${SERVER_CONNECT_TIMEOUT}\n}\
|
|
${SERVER_LOGIN_RETRY:+server_login_retry = ${SERVER_LOGIN_RETRY}\n}\
|
|
${CLIENT_LOGIN_TIMEOUT:+client_login_timeout = ${CLIENT_LOGIN_TIMEOUT}\n}\
|
|
${AUTODB_IDLE_TIMEOUT:+autodb_idle_timeout = ${AUTODB_IDLE_TIMEOUT}\n}\
|
|
${DNS_MAX_TTL:+dns_max_ttl = ${DNS_MAX_TTL}\n}\
|
|
${DNS_NXDOMAIN_TTL:+dns_nxdomain_ttl = ${DNS_NXDOMAIN_TTL}\n}\
|
|
|
|
# TLS settings
|
|
${CLIENT_TLS_SSLMODE:+client_tls_sslmode = ${CLIENT_TLS_SSLMODE}\n}\
|
|
${CLIENT_TLS_KEY_FILE:+client_tls_key_file = ${CLIENT_TLS_KEY_FILE}\n}\
|
|
${CLIENT_TLS_CERT_FILE:+client_tls_cert_file = ${CLIENT_TLS_CERT_FILE}\n}\
|
|
${CLIENT_TLS_CA_FILE:+client_tls_ca_file = ${CLIENT_TLS_CA_FILE}\n}\
|
|
${CLIENT_TLS_PROTOCOLS:+client_tls_protocols = ${CLIENT_TLS_PROTOCOLS}\n}\
|
|
${CLIENT_TLS_CIPHERS:+client_tls_ciphers = ${CLIENT_TLS_CIPHERS}\n}\
|
|
${CLIENT_TLS_ECDHCURVE:+client_tls_ecdhcurve = ${CLIENT_TLS_ECDHCURVE}\n}\
|
|
${CLIENT_TLS_DHEPARAMS:+client_tls_dheparams = ${CLIENT_TLS_DHEPARAMS}\n}\
|
|
${SERVER_TLS_SSLMODE:+server_tls_sslmode = ${SERVER_TLS_SSLMODE}\n}\
|
|
${SERVER_TLS_CA_FILE:+server_tls_ca_file = ${SERVER_TLS_CA_FILE}\n}\
|
|
${SERVER_TLS_KEY_FILE:+server_tls_key_file = ${SERVER_TLS_KEY_FILE}\n}\
|
|
${SERVER_TLS_CERT_FILE:+server_tls_cert_file = ${SERVER_TLS_CERT_FILE}\n}\
|
|
${SERVER_TLS_PROTOCOLS:+server_tls_protocols = ${SERVER_TLS_PROTOCOLS}\n}\
|
|
${SERVER_TLS_CIPHERS:+server_tls_ciphers = ${SERVER_TLS_CIPHERS}\n}\
|
|
|
|
# Dangerous timeouts
|
|
${QUERY_TIMEOUT:+query_timeout = ${QUERY_TIMEOUT}\n}\
|
|
${QUERY_WAIT_TIMEOUT:+query_wait_timeout = ${QUERY_WAIT_TIMEOUT}\n}\
|
|
${CLIENT_IDLE_TIMEOUT:+client_idle_timeout = ${CLIENT_IDLE_TIMEOUT}\n}\
|
|
${IDLE_TRANSACTION_TIMEOUT:+idle_transaction_timeout = ${IDLE_TRANSACTION_TIMEOUT}\n}\
|
|
${PKT_BUF:+pkt_buf = ${PKT_BUF}\n}\
|
|
${MAX_PACKET_SIZE:+max_packet_size = ${MAX_PACKET_SIZE}\n}\
|
|
${LISTEN_BACKLOG:+listen_backlog = ${LISTEN_BACKLOG}\n}\
|
|
${SBUF_LOOPCNT:+sbuf_loopcnt = ${SBUF_LOOPCNT}\n}\
|
|
${SUSPEND_TIMEOUT:+suspend_timeout = ${SUSPEND_TIMEOUT}\n}\
|
|
${TCP_DEFER_ACCEPT:+tcp_defer_accept = ${TCP_DEFER_ACCEPT}\n}\
|
|
${TCP_KEEPALIVE:+tcp_keepalive = ${TCP_KEEPALIVE}\n}\
|
|
${TCP_KEEPCNT:+tcp_keepcnt = ${TCP_KEEPCNT}\n}\
|
|
${TCP_KEEPIDLE:+tcp_keepidle = ${TCP_KEEPIDLE}\n}\
|
|
${TCP_KEEPINTVL:+tcp_keepintvl = ${TCP_KEEPINTVL}\n}\
|
|
${TCP_USER_TIMEOUT:+tcp_user_timeout = ${TCP_USER_TIMEOUT}\n}\
|
|
################## end file ##################
|
|
" >> "${PG_CONFIG_FILE}"
|
|
cat "${PG_CONFIG_FILE}"
|
|
fi
|
|
|
|
echo "Starting $*..."
|
|
exec "$@" |