206 lines
7.2 KiB
Bash
Executable File
206 lines
7.2 KiB
Bash
Executable File
#!/bin/bash
|
|
## EMQ docker image start script
|
|
# Huang Rui <vowstar@gmail.com>
|
|
# EMQ X Team <support@emqx.io>
|
|
|
|
## Shell setting
|
|
if [[ -n "$DEBUG" ]]; then
|
|
set -ex
|
|
else
|
|
set -e
|
|
fi
|
|
|
|
shopt -s nullglob
|
|
|
|
## Local IP address setting
|
|
|
|
LOCAL_IP=$(hostname -i | grep -oE '((25[0-5]|(2[0-4]|1[0-9]|[1-9]|)[0-9])\.){3}(25[0-5]|(2[0-4]|1[0-9]|[1-9]|)[0-9])' | head -n 1)
|
|
|
|
## EMQ Base settings and plugins setting
|
|
# Base settings in /opt/emqx/etc/emqx.conf
|
|
# Plugin settings in /opt/emqx/etc/plugins
|
|
|
|
_EMQX_HOME='/opt/emqx'
|
|
|
|
if [[ -z "$EMQX_NAME" ]]; then
|
|
EMQX_NAME="$(hostname)"
|
|
export EMQX_NAME
|
|
fi
|
|
|
|
if [[ -z "$EMQX_HOST" ]]; then
|
|
if [[ "$EMQX_CLUSTER__K8S__ADDRESS_TYPE" == "dns" ]] && [[ -n "$EMQX_CLUSTER__K8S__NAMESPACE" ]]; then
|
|
EMQX_CLUSTER__K8S__SUFFIX=${EMQX_CLUSTER__K8S__SUFFIX:-"pod.cluster.local"}
|
|
EMQX_HOST="${LOCAL_IP//./-}.$EMQX_CLUSTER__K8S__NAMESPACE.$EMQX_CLUSTER__K8S__SUFFIX"
|
|
elif [[ "$EMQX_CLUSTER__K8S__ADDRESS_TYPE" == 'hostname' ]] && [[ -n "$EMQX_CLUSTER__K8S__NAMESPACE" ]]; then
|
|
EMQX_CLUSTER__K8S__SUFFIX=${EMQX_CLUSTER__K8S__SUFFIX:-'svc.cluster.local'}
|
|
EMQX_HOST=$(grep -h "^$LOCAL_IP" /etc/hosts | grep -o "$(hostname).*.$EMQX_CLUSTER__K8S__NAMESPACE.$EMQX_CLUSTER__K8S__SUFFIX")
|
|
else
|
|
EMQX_HOST="$LOCAL_IP"
|
|
fi
|
|
export EMQX_HOST
|
|
fi
|
|
|
|
if [[ -z "$EMQX_NODE_NAME" ]]; then
|
|
export EMQX_NODE_NAME="$EMQX_NAME@$EMQX_HOST"
|
|
fi
|
|
|
|
# Set hosts to prevent cluster mode failed
|
|
|
|
if [[ -z "$EMQX_NODE__PROCESS_LIMIT" ]]; then
|
|
export EMQX_NODE__PROCESS_LIMIT=2097152
|
|
fi
|
|
|
|
if [[ -z "$EMQX_NODE__MAX_PORTS" ]]; then
|
|
export EMQX_NODE__MAX_PORTS=1048576
|
|
fi
|
|
|
|
if [[ -z "$EMQX_NODE__MAX_ETS_TABLES" ]]; then
|
|
export EMQX_NODE__MAX_ETS_TABLES=2097152
|
|
fi
|
|
|
|
if [[ -z "$EMQX_LISTENER__TCP__EXTERNAL__ACCEPTORS" ]]; then
|
|
export EMQX_LISTENER__TCP__EXTERNAL__ACCEPTORS=64
|
|
fi
|
|
|
|
if [[ -z "$EMQX_LISTENER__TCP__EXTERNAL__MAX_CONNECTIONS" ]]; then
|
|
export EMQX_LISTENER__TCP__EXTERNAL__MAX_CONNECTIONS=1024000
|
|
fi
|
|
|
|
if [[ -z "$EMQX_LISTENER__SSL__EXTERNAL__ACCEPTORS" ]]; then
|
|
export EMQX_LISTENER__SSL__EXTERNAL__ACCEPTORS=32
|
|
fi
|
|
|
|
if [[ -z "$EMQX_LISTENER__SSL__EXTERNAL__MAX_CONNECTIONS" ]]; then
|
|
export EMQX_LISTENER__SSL__EXTERNAL__MAX_CONNECTIONS=102400
|
|
fi
|
|
|
|
if [[ -z "$EMQX_LISTENER__WS__EXTERNAL__ACCEPTORS" ]]; then
|
|
export EMQX_LISTENER__WS__EXTERNAL__ACCEPTORS=16
|
|
fi
|
|
|
|
if [[ -z "$EMQX_LISTENER__WS__EXTERNAL__MAX_CONNECTIONS" ]]; then
|
|
export EMQX_LISTENER__WS__EXTERNAL__MAX_CONNECTIONS=102400
|
|
fi
|
|
|
|
if [[ -z "$EMQX_LISTENER__WSS__EXTERNAL__ACCEPTORS" ]]; then
|
|
export EMQX_LISTENER__WSS__EXTERNAL__ACCEPTORS=16
|
|
fi
|
|
|
|
if [[ -z "$EMQX_LISTENER__WSS__EXTERNAL__MAX_CONNECTIONS" ]]; then
|
|
export EMQX_LISTENER__WSS__EXTERNAL__MAX_CONNECTIONS=102400
|
|
fi
|
|
|
|
# Fix issue #42 - export env EMQX_DASHBOARD__DEFAULT_USER__PASSWORD to configure
|
|
# 'dashboard.default_user.password' in etc/plugins/emqx_dashboard.conf
|
|
if [[ -n "$EMQX_ADMIN_PASSWORD" ]]; then
|
|
export EMQX_DASHBOARD__DEFAULT_USER__PASSWORD=$EMQX_ADMIN_PASSWORD
|
|
fi
|
|
|
|
# echo value of $VAR hiding secrets if any
|
|
# SYNOPSIS
|
|
# echo_value KEY VALUE
|
|
echo_value() {
|
|
# get MASK_CONFIG
|
|
MASK_CONFIG_FILTER="$MASK_CONFIG_FILTER|password|passwd|key|token|secret"
|
|
FORMAT_MASK_CONFIG_FILTER=$(echo "$MASK_CONFIG_FILTER" | sed -r -e 's/^[^A-Za-z0-9_]+//' -e 's/[^A-Za-z0-9_]+$//' -e 's/[^A-Za-z0-9_]+/|/g')
|
|
local key=$1
|
|
local value=$2
|
|
# check if contains sensitive value
|
|
if echo "$key" | grep -iqwE "$FORMAT_MASK_CONFIG_FILTER"; then
|
|
echo "$key=***secret***"
|
|
else
|
|
echo "$key=$value"
|
|
fi
|
|
}
|
|
|
|
# fill config on specific file if the key exists
|
|
# SYNOPSIS
|
|
# try_fill_config FILE KEY VALUE
|
|
try_fill_config() {
|
|
local file=$1
|
|
local key=$2
|
|
local value=$3
|
|
local escaped_key
|
|
# shellcheck disable=SC2001
|
|
escaped_key=$(echo "$key" | sed 's/[^a-zA-Z0-9_]/\\&/g')
|
|
local escaped_value
|
|
escaped_value=$(echo "$value" | sed 's/[\/&]/\\&/g')
|
|
if grep -qE "^[#[:space:]]*$escaped_key\s*=" "$file"; then
|
|
echo_value "$key" "$value"
|
|
if [[ -z "$value" ]]; then
|
|
sed -r "s/^[#[:space:]]*($escaped_key)\s*=\s*(.*)/# \1 = \2/" "$file" > tmpfile && cat tmpfile > "$file"
|
|
else
|
|
sed -r "s/^[#[:space:]]*($escaped_key)\s*=\s*(.*)/\1 = $escaped_value/" "$file" > tmpfile && cat tmpfile > "$file"
|
|
fi
|
|
# Check if config has a numbering system, but no existing configuration line in file
|
|
elif echo "$key" | grep -qE '\.\d+|\d+\.'; then
|
|
if [[ -n "$value" ]]; then
|
|
local template
|
|
template="$(echo "$escaped_key" | sed -r -e 's/\\\.[0-9]+/\\.[0-9]+/g' -e 's/[0-9]+\\\./[0-9]+\\./g')"
|
|
if grep -qE "^[#[:space:]]*$template\s*=" "$file"; then
|
|
echo_value "$key" "$value"
|
|
sed '$a'\\ "$file" > tmpfile && cat tmpfile > "$file"
|
|
echo "$key = $value" >> "$file"
|
|
fi
|
|
fi
|
|
fi
|
|
}
|
|
|
|
# Catch all EMQX_ prefix environment variable and match it in configure file
|
|
CONFIG_FILE="$_EMQX_HOME/etc/emqx.conf"
|
|
CONFIG_PLUGINS="$_EMQX_HOME/etc/plugins"
|
|
for VAR in $(compgen -e); do
|
|
# Config normal keys such like node.name = emqx@127.0.0.1
|
|
if echo "$VAR" | grep -q '^EMQX_'; then
|
|
VAR_NAME=$(echo "$VAR" | sed -e 's/^EMQX_//' -e 's/__/./g' | tr '[:upper:]' '[:lower:]' | tr -d '[:cntrl:]')
|
|
VAR_VALUE=$(echo "${!VAR}" | tr -d '[:cntrl:]')
|
|
# Config in emqx.conf
|
|
try_fill_config "$CONFIG_FILE" "$VAR_NAME" "$VAR_VALUE"
|
|
# Config in plugins/*
|
|
for CONFIG_PLUGINS_FILE in "$CONFIG_PLUGINS"/*; do
|
|
try_fill_config "$CONFIG_PLUGINS_FILE" "$VAR_NAME" "$VAR_VALUE"
|
|
done
|
|
fi
|
|
done
|
|
|
|
# fill tuples on specific file
|
|
# SYNOPSIS
|
|
# fill_tuples FILE [ELEMENTS ...]
|
|
fill_tuples() {
|
|
local file=$1
|
|
local elements=${*:2}
|
|
for var in $elements; do
|
|
if grep -qE "\{\s*$var\s*,\s*(true|false)\s*\}\s*\." "$file"; then
|
|
sed -r "s/\{\s*($var)\s*,\s*(true|false)\s*\}\s*\./{\1, true}./1" "$file" > tmpfile && mv tmpfile "$file"
|
|
elif grep -q "$var\s*\." "$file"; then
|
|
# backward compatible.
|
|
sed -r "s/($var)\s*\./{\1, true}./1" "$file" > tmpfile && cat tmpfile > "$file"
|
|
else
|
|
sed '$a'\\ "$file" > tmpfile && cat tmpfile > "$file"
|
|
echo "{$var, true}." >> "$file"
|
|
fi
|
|
done
|
|
}
|
|
|
|
## EMQX Plugin load settings
|
|
# Plugins loaded by default
|
|
LOADED_PLUGINS="$_EMQX_HOME/data/loaded_plugins"
|
|
if [[ -n "$EMQX_LOADED_PLUGINS" ]]; then
|
|
EMQX_LOADED_PLUGINS=$(echo "$EMQX_LOADED_PLUGINS" | tr -d '[:cntrl:]' | sed -r -e 's/^[^A-Za-z0-9_]+//g' -e 's/[^A-Za-z0-9_]+$//g' -e 's/[^A-Za-z0-9_]+/ /g')
|
|
echo "EMQX_LOADED_PLUGINS=$EMQX_LOADED_PLUGINS"
|
|
# Parse module names and place `{module_name, true}.` tuples in `loaded_plugins`.
|
|
fill_tuples "$LOADED_PLUGINS" "$EMQX_LOADED_PLUGINS"
|
|
fi
|
|
|
|
## EMQX Modules load settings
|
|
# Modules loaded by default
|
|
LOADED_MODULES="$_EMQX_HOME/data/loaded_modules"
|
|
if [[ -n "$EMQX_LOADED_MODULES" ]]; then
|
|
EMQX_LOADED_MODULES=$(echo "$EMQX_LOADED_MODULES" | tr -d '[:cntrl:]' | sed -r -e 's/^[^A-Za-z0-9_]+//g' -e 's/[^A-Za-z0-9_]+$//g' -e 's/[^A-Za-z0-9_]+/ /g')
|
|
echo "EMQX_LOADED_MODULES=$EMQX_LOADED_MODULES"
|
|
# Parse module names and place `{module_name, true}.` tuples in `loaded_modules`.
|
|
fill_tuples "$LOADED_MODULES" "$EMQX_LOADED_MODULES"
|
|
fi
|
|
|
|
exec "$@"
|