entrypoint.sh

#!/bin/bash
set -e

## ProxySQL entrypoint
## ===================
PROXYSQL_CONF_CHECK_INTERVAL=${PROXYSQL_CONF_CHECK_INTERVAL:-60}
PROXYSQL_CONF_LIVE_RELOAD=${PROXYSQL_CONF_LIVE_RELOAD:-false}
PROXYSQL_WORKDIR=${PROXYSQL_WORKDIR:-/proxysql}
PROXYSQL_CONF_DIR=$PROXYSQL_WORKDIR/conf
PROXYSQL_CONF_TPL_FILE=$PROXYSQL_CONF_DIR/proxysql.cnf.tpl
PROXYSQL_CONF_FILE=/etc/proxysql.cnf
PROXYSQL_SECRETS_DIR=$PROXYSQL_WORKDIR/secrets
PROXYSQL_SECRETS_FILE=$PROXYSQL_SECRETS_DIR/secrets.env
PROXYSQL_CONF_RELOADER_LOCK=$PROXYSQL_WORKDIR/proxysql-config-reloader.lock
PROXYSQL_ADMIN_USERNAME=${PROXYSQL_ADMIN_USERNAME:-admin}
PROXYSQL_ADMIN_PASSWORD=${PROXYSQL_ADMIN_PASSWORD:-admin}
PROXYSQL_ADMIN_HOST=${PROXYSQL_ADMIN_HOST:-127.0.0.1}
PROXYSQL_ADMIN_PORT=${PROXYSQL_ADMIN_PORT:-6032}

function loadEnv {
    echo "Loading env vars from $PROXYSQL_SECRETS_FILE"
    eval $(cat $PROXYSQL_SECRETS_FILE | sed 's/^/export /') && \
    echo "Loaded ProxySQL env vars"
}

function renderProxySQLCnf {
    cat $PROXYSQL_CONF_TPL_FILE | envsubst > $PROXYSQL_CONF_FILE && \
    echo "Rendered $PROXYSQL_CONF_TPL_FILE"
}

function hardReloadProxySQLCnf {
    MYSQL_CMD='mysql -h ${PROXYSQL_ADMIN_HOST} -P ${PROXYSQL_ADMIN_PORT} -u ${PROXYSQL_ADMIN_USERNAME} -p${PROXYSQL_ADMIN_PASSWORD} -D main -e "
        DELETE FROM mysql_servers;
        DELETE FROM mysql_users;
        DELETE FROM mysql_query_rules;
        LOAD MYSQL VARIABLES FROM CONFIG;
        LOAD MYSQL SERVERS FROM CONFIG;
        LOAD MYSQL USERS FROM CONFIG;
        LOAD MYSQL QUERY RULES FROM CONFIG;
        LOAD MYSQL VARIABLES TO RUNTIME;
        LOAD MYSQL SERVERS TO RUNTIME;
        LOAD MYSQL USERS TO RUNTIME;
        LOAD MYSQL QUERY RULES TO RUNTIME;
        SAVE MYSQL VARIABLES TO DISK;
        SAVE MYSQL SERVERS TO DISK;
        SAVE MYSQL USERS TO DISK;
        SAVE MYSQL QUERY RULES TO DISK;"'
    echo "$MYSQL_CMD"
    eval $MYSQL_CMD || echo ""
}

function configReloader {
    if [ -f $PROXYSQL_CONF_RELOADER_LOCK ]; then
        return
    fi

    if [ "$PROXYSQL_CONF_LIVE_RELOAD" != true ]; then
        echo "ProxySQL live config reload is disabled."
        return
    fi
    echo "ProxySQL live config reload is enabled. Checking config updates every ${PROXYSQL_CONF_CHECK_INTERVAL}s..."

    touch $PROXYSQL_CONF_RELOADER_LOCK
    echo "Starting ProxySQL config reloader..."
    tpl_hash=$(md5sum $PROXYSQL_CONF_TPL_FILE)
    secrets_hash=$(md5sum $PROXYSQL_SECRETS_FILE)

    while true; do
        sleep $PROXYSQL_CONF_CHECK_INTERVAL
        echo "Checking for ProxySQL config changes..."
        tpl_hash_=$(md5sum $PROXYSQL_CONF_TPL_FILE)
        secrets_hash_=$(md5sum $PROXYSQL_SECRETS_FILE)
        if [ "$tpl_hash" != "$tpl_hash_" ] || [ "$secrets_hash" != "$secrets_hash_" ]; then
            echo "ProxySQL config changed..."
            loadEnv
            renderProxySQLCnf
            hardReloadProxySQLCnf
        fi
        tpl_hash=$tpl_hash_
        secrets_hash=$secrets_hash_
    done
    rm -rf $PROXYSQL_CONF_RELOADER_LOCK
}

function init {
    loadEnv
    renderProxySQLCnf
    configReloader &
}

init

exec $@