|
| 1 | +#!/bin/bash |
| 2 | +set -eo pipefail |
| 3 | +shopt -s nullglob |
| 4 | + |
| 5 | +# if command starts with an option, prepend mysqld |
| 6 | +if [ "${1:0:1}" = '-' ]; then |
| 7 | + set -- mysqld "$@" |
| 8 | +fi |
| 9 | + |
| 10 | +# skip setup if they want an option that stops mysqld |
| 11 | +wantHelp= |
| 12 | +for arg; do |
| 13 | + case "$arg" in |
| 14 | + -'?'|--help|--print-defaults|-V|--version) |
| 15 | + wantHelp=1 |
| 16 | + break |
| 17 | + ;; |
| 18 | + esac |
| 19 | +done |
| 20 | + |
| 21 | +_check_config() { |
| 22 | + toRun=( "$@" --verbose --help ) |
| 23 | + if ! errors="$("${toRun[@]}" 2>&1 >/dev/null)"; then |
| 24 | + cat >&2 <<-EOM |
| 25 | +
|
| 26 | + ERROR: mysqld failed while attempting to check config |
| 27 | + command was: "${toRun[*]}" |
| 28 | +
|
| 29 | + $errors |
| 30 | + EOM |
| 31 | + exit 1 |
| 32 | + fi |
| 33 | +} |
| 34 | + |
| 35 | +_datadir() { |
| 36 | + "$@" --verbose --help 2>/dev/null | awk '$1 == "datadir" { print $2; exit }' |
| 37 | +} |
| 38 | + |
| 39 | +# allow the container to be started with `--user` |
| 40 | +if [ "$1" = 'mysqld' -a -z "$wantHelp" -a "$(id -u)" = '0' ]; then |
| 41 | + _check_config "$@" |
| 42 | + DATADIR="$(_datadir "$@")" |
| 43 | + mkdir -p "$DATADIR" |
| 44 | + chown -R mysql:mysql "$DATADIR" |
| 45 | + exec gosu mysql "$BASH_SOURCE" "$@" |
| 46 | +fi |
| 47 | + |
| 48 | +if [ "$1" = 'mysqld' -a -z "$wantHelp" ]; then |
| 49 | + # still need to check config, container may have started with --user |
| 50 | + _check_config "$@" |
| 51 | + # Get config |
| 52 | + DATADIR="$(_datadir "$@")" |
| 53 | + |
| 54 | + if [ ! -d "$DATADIR/mysql" ]; then |
| 55 | + if [ -z "$MYSQL_ROOT_PASSWORD" -a -z "$MYSQL_ALLOW_EMPTY_PASSWORD" -a -z "$MYSQL_RANDOM_ROOT_PASSWORD" ]; then |
| 56 | + echo >&2 'error: database is uninitialized and password option is not specified ' |
| 57 | + echo >&2 ' You need to specify one of MYSQL_ROOT_PASSWORD, MYSQL_ALLOW_EMPTY_PASSWORD and MYSQL_RANDOM_ROOT_PASSWORD' |
| 58 | + exit 1 |
| 59 | + fi |
| 60 | + |
| 61 | + mkdir -p "$DATADIR" |
| 62 | + |
| 63 | + echo 'Initializing database' |
| 64 | + "$@" --initialize-insecure |
| 65 | + echo 'Database initialized' |
| 66 | + |
| 67 | + "$@" --skip-networking & |
| 68 | + pid="$!" |
| 69 | + |
| 70 | + mysql=( mysql --protocol=socket -uroot ) |
| 71 | + |
| 72 | + for i in {30..0}; do |
| 73 | + if echo 'SELECT 1' | "${mysql[@]}" &> /dev/null; then |
| 74 | + break |
| 75 | + fi |
| 76 | + echo 'MySQL init process in progress...' |
| 77 | + sleep 1 |
| 78 | + done |
| 79 | + if [ "$i" = 0 ]; then |
| 80 | + echo >&2 'MySQL init process failed.' |
| 81 | + exit 1 |
| 82 | + fi |
| 83 | + |
| 84 | + if [ -z "$MYSQL_INITDB_SKIP_TZINFO" ]; then |
| 85 | + # sed is for https://bugs.mysql.com/bug.php?id=20545 |
| 86 | + mysql_tzinfo_to_sql /usr/share/zoneinfo | sed 's/Local time zone must be set--see zic manual page/FCTY/' | "${mysql[@]}" mysql |
| 87 | + fi |
| 88 | + |
| 89 | + if [ ! -z "$MYSQL_RANDOM_ROOT_PASSWORD" ]; then |
| 90 | + MYSQL_ROOT_PASSWORD="$(pwgen -1 32)" |
| 91 | + echo "GENERATED ROOT PASSWORD: $MYSQL_ROOT_PASSWORD" |
| 92 | + fi |
| 93 | + "${mysql[@]}" <<-EOSQL |
| 94 | + -- What's done in this file shouldn't be replicated |
| 95 | + -- or products like mysql-fabric won't work |
| 96 | + SET @@SESSION.SQL_LOG_BIN=0; |
| 97 | +
|
| 98 | + DELETE FROM mysql.user ; |
| 99 | + CREATE USER 'root'@'%' IDENTIFIED BY '${MYSQL_ROOT_PASSWORD}' ; |
| 100 | + GRANT ALL ON *.* TO 'root'@'%' WITH GRANT OPTION ; |
| 101 | + DROP DATABASE IF EXISTS test ; |
| 102 | + FLUSH PRIVILEGES ; |
| 103 | + EOSQL |
| 104 | + |
| 105 | + if [ ! -z "$MYSQL_ROOT_PASSWORD" ]; then |
| 106 | + mysql+=( -p"${MYSQL_ROOT_PASSWORD}" ) |
| 107 | + fi |
| 108 | + |
| 109 | + if [ "$MYSQL_DATABASE" ]; then |
| 110 | + echo "CREATE DATABASE IF NOT EXISTS \`$MYSQL_DATABASE\` ;" | "${mysql[@]}" |
| 111 | + mysql+=( "$MYSQL_DATABASE" ) |
| 112 | + fi |
| 113 | + |
| 114 | + if [ "$MYSQL_USER" -a "$MYSQL_PASSWORD" ]; then |
| 115 | + echo "CREATE USER '$MYSQL_USER'@'%' IDENTIFIED BY '$MYSQL_PASSWORD' ;" | "${mysql[@]}" |
| 116 | + |
| 117 | + if [ "$MYSQL_DATABASE" ]; then |
| 118 | + echo "GRANT ALL ON \`$MYSQL_DATABASE\`.* TO '$MYSQL_USER'@'%' ;" | "${mysql[@]}" |
| 119 | + fi |
| 120 | + |
| 121 | + echo 'FLUSH PRIVILEGES ;' | "${mysql[@]}" |
| 122 | + fi |
| 123 | + |
| 124 | + echo |
| 125 | + for f in /docker-entrypoint-initdb.d/*; do |
| 126 | + case "$f" in |
| 127 | + *.sh) echo "$0: running $f"; . "$f" ;; |
| 128 | + *.sql) echo "$0: running $f"; "${mysql[@]}" < "$f"; echo ;; |
| 129 | + *.sql.gz) echo "$0: running $f"; gunzip -c "$f" | "${mysql[@]}"; echo ;; |
| 130 | + *) echo "$0: ignoring $f" ;; |
| 131 | + esac |
| 132 | + echo |
| 133 | + done |
| 134 | + |
| 135 | + if [ ! -z "$MYSQL_ONETIME_PASSWORD" ]; then |
| 136 | + "${mysql[@]}" <<-EOSQL |
| 137 | + ALTER USER 'root'@'%' PASSWORD EXPIRE; |
| 138 | + EOSQL |
| 139 | + fi |
| 140 | + if ! kill -s TERM "$pid" || ! wait "$pid"; then |
| 141 | + echo >&2 'MySQL init process failed.' |
| 142 | + exit 1 |
| 143 | + fi |
| 144 | + |
| 145 | + echo |
| 146 | + echo 'MySQL init process done. Ready for start up.' |
| 147 | + echo |
| 148 | + fi |
| 149 | +fi |
| 150 | + |
| 151 | +exec "$@" |
0 commit comments