-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
28 additions
and
61 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,82 +1,49 @@ | ||
| #!/bin/sh | ||
| #! /bin/sh | ||
|
|
||
| set -euo pipefail | ||
|
|
||
| # Configure mysql env vars | ||
| : "${PDNS_gmysql_host:=${MYSQL_ENV_MYSQL_HOST:-mysql}}" | ||
| : "${PDNS_gmysql_port:=${MYSQL_ENV_MYSQL_PORT:-3306}}" | ||
| : "${PDNS_gmysql_user:=${MYSQL_ENV_MYSQL_USER:-root}}" | ||
| if [ "${PDNS_gmysql_user}" = 'root' ]; then | ||
| : "${PDNS_gmysql_password:=${MYSQL_ENV_MYSQL_ROOT_PASSWORD:-}}" | ||
| fi | ||
| : "${PDNS_gmysql_password:=${MYSQL_ENV_MYSQL_PASSWORD:-powerdns}}" | ||
| : "${PDNS_gmysql_dbname:=${MYSQL_ENV_MYSQL_DATABASE:-powerdns}}" | ||
|
|
||
| # use first part of node name as database name suffix | ||
| if [ "${NODE_NAME:-}" ]; then | ||
| NODE_NAME=$(echo ${NODE_NAME} | sed -e 's/\..*//' -e 's/-//') | ||
| PDNS_gmysql_dbname="${PDNS_gmysql_dbname}${NODE_NAME}" | ||
| fi | ||
| if [ "${PDNS_gmysql_host:-}" != "" ]; then | ||
| echo "Running migrations on the database..." | ||
|
|
||
| SQL_COMMAND="mysql -h ${PDNS_gmysql_host} -P ${PDNS_gmysql_port} -u ${PDNS_gmysql_user} -p${PDNS_gmysql_password} -D ${PDNS_gmysql_dbname}" | ||
|
|
||
| export PDNS_gmysql_host PDNS_gmysql_port PDNS_gmysql_user PDNS_gmysql_password PDNS_gmysql_dbname | ||
| # Fix DB schema | ||
| sed -i 's/CREATE TABLE/CREATE TABLE IF NOT EXISTS/g ; s/CREATE UNIQUE INDEX/CREATE UNIQUE INDEX IF NOT EXISTS/g ; s/CREATE INDEX/CREATE INDEX IF NOT EXISTS/g' /usr/share/doc/pdns/schema.mysql.sql | ||
|
|
||
| EXTRA="" | ||
| # Initialize DB if needed | ||
| $SQL_COMMAND </usr/share/doc/pdns/schema.mysql.sql | ||
|
|
||
| # Password Auth | ||
| if [ "${PDNS_gmysql_password}" != "" ]; then | ||
| EXTRA="${EXTRA} -p${PDNS_gmysql_password}" | ||
| echo "Done." | ||
| fi | ||
|
|
||
| # Allow socket connections | ||
| if [ "${PDNS_gmysql_socket:-}" != "" ]; then | ||
| export PDNS_gmysql_host="localhost" | ||
| EXTRA="${EXTRA} --socket=${PDNS_gmysql_socket}" | ||
| fi | ||
| if [ "${PDNS_gpgsql_host:-}" != "" ]; then | ||
| echo "Running migrations on the database..." | ||
|
|
||
| MYSQL_COMMAND="mysql -h ${PDNS_gmysql_host} -P ${PDNS_gmysql_port} -u ${PDNS_gmysql_user}${EXTRA}" | ||
| export PGPASSWORD=${PDNS_gpgsql_password} | ||
| SQL_COMMAND="psql -h ${PDNS_gpgsql_host} -p ${PDNS_gpgsql_port} -U ${PDNS_gpgsql_user} -w ${PDNS_gpgsql_dbname}" | ||
|
|
||
| # Wait for MySQL to respond | ||
| until $MYSQL_COMMAND -e ';' ; do | ||
| >&2 echo 'MySQL is unavailable - sleeping' | ||
| sleep 3 | ||
| done | ||
| # Fix DB schema | ||
| sed -i 's/CREATE TABLE/CREATE TABLE IF NOT EXISTS/g ; s/CREATE UNIQUE INDEX/CREATE UNIQUE INDEX IF NOT EXISTS/g ; s/CREATE INDEX/CREATE INDEX IF NOT EXISTS/g' /usr/share/doc/pdns/schema.pgsql.sql | ||
|
|
||
| # Initialize DB if needed | ||
| if [ "${SKIP_DB_CREATE:-false}" != 'true' ]; then | ||
| $MYSQL_COMMAND -e "CREATE DATABASE IF NOT EXISTS ${PDNS_gmysql_dbname}" | ||
| fi | ||
| # Initialize DB | ||
| $SQL_COMMAND </usr/share/doc/pdns/schema.pgsql.sql | ||
|
|
||
| MYSQL_CHECK_IF_HAS_TABLE="SELECT COUNT(DISTINCT table_name) FROM information_schema.columns WHERE table_schema = '${PDNS_gmysql_dbname}';" | ||
| MYSQL_NUM_TABLE=$($MYSQL_COMMAND --batch --skip-column-names -e "$MYSQL_CHECK_IF_HAS_TABLE") | ||
| if [ "$MYSQL_NUM_TABLE" -eq 0 ]; then | ||
| $MYSQL_COMMAND -D "$PDNS_gmysql_dbname" < /usr/share/doc/pdns/schema.mysql.sql | ||
| echo "Done." | ||
| fi | ||
|
|
||
| if [ "${PDNS_superslave:-no}" == "yes" ]; then | ||
| # Configure supermasters if needed | ||
| if [ "${SUPERMASTER_IPS:-}" ]; then | ||
| $MYSQL_COMMAND -D "$PDNS_gmysql_dbname" -e "TRUNCATE supermasters;" | ||
| MYSQL_INSERT_SUPERMASTERS='' | ||
| if [ "${SUPERMASTER_COUNT:-0}" == "0" ]; then | ||
| SUPERMASTER_COUNT=10 | ||
| fi | ||
| i=1; while [ $i -le ${SUPERMASTER_COUNT} ]; do | ||
| SUPERMASTER_HOST=$(echo ${SUPERMASTER_HOSTS:-} | awk -v col="$i" '{ print $col }') | ||
| SUPERMASTER_IP=$(echo ${SUPERMASTER_IPS} | awk -v col="$i" '{ print $col }') | ||
| if [ -z "${SUPERMASTER_HOST:-}" ]; then | ||
| SUPERMASTER_HOST=$(hostname -f) | ||
| fi | ||
| if [ "${SUPERMASTER_IP:-}" ]; then | ||
| MYSQL_INSERT_SUPERMASTERS="${MYSQL_INSERT_SUPERMASTERS} INSERT INTO supermasters VALUES('${SUPERMASTER_IP}', '${SUPERMASTER_HOST}', 'admin');" | ||
| fi | ||
| i=$(( i + 1 )) | ||
| done | ||
| $MYSQL_COMMAND -D "$PDNS_gmysql_dbname" -e "$MYSQL_INSERT_SUPERMASTERS" | ||
| if [ "${PDNS_autosecondary:-no}" == "yes" ]; then | ||
| # Configure masters for secondary if needed | ||
| if [ "${SUPERMASTER_IPS:-}" != "" ]; then | ||
| echo "Inserting masters for autosecondary..." | ||
| echo "TRUNCATE supermasters;" | $SQL_COMMAND | ||
|
|
||
| SUPERMASTERS=$($SUPERMASTER_IPS | sed 's/,/"), ("/g ; s/^/("/ ; s/$/")/ ; s/:/","/g') | ||
| echo "INSERT INTO supermasters VALUES $SUPERMASTERS;" | $SQL_COMMAND | ||
| echo "Done." | ||
| fi | ||
| fi | ||
|
|
||
| # Create config file from template | ||
| envtpl < /pdns.conf.tpl > /etc/pdns/pdns.conf | ||
| envtpl </pdns.conf.tpl >/etc/pdns/pdns.conf | ||
|
|
||
| exec "$@" |