docker-entrypoint.sh 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. #!/usr/bin/env bash
  2. ## EMQ docker image start script
  3. # Huang Rui <vowstar@gmail.com>
  4. # EMQX Team <support@emqx.io>
  5. ## Shell setting
  6. if [[ -n "$DEBUG" ]]; then
  7. set -ex
  8. else
  9. set -e
  10. fi
  11. shopt -s nullglob
  12. ## Local IP address setting
  13. 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)
  14. if [[ -z "$EMQX_NODE_NAME" ]]; then
  15. if [[ -z "$EMQX_NAME" ]]; then
  16. EMQX_NAME="$(hostname)"
  17. fi
  18. if [[ -z "$EMQX_HOST" ]]; then
  19. if [[ "$EMQX_CLUSTER__K8S__ADDRESS_TYPE" == "dns" ]] && [[ -n "$EMQX_CLUSTER__K8S__NAMESPACE" ]]; then
  20. EMQX_CLUSTER__K8S__SUFFIX=${EMQX_CLUSTER__K8S__SUFFIX:-"pod.cluster.local"}
  21. EMQX_HOST="${LOCAL_IP//./-}.$EMQX_CLUSTER__K8S__NAMESPACE.$EMQX_CLUSTER__K8S__SUFFIX"
  22. elif [[ "$EMQX_CLUSTER__K8S__ADDRESS_TYPE" == 'hostname' ]] && [[ -n "$EMQX_CLUSTER__K8S__NAMESPACE" ]]; then
  23. EMQX_CLUSTER__K8S__SUFFIX=${EMQX_CLUSTER__K8S__SUFFIX:-'svc.cluster.local'}
  24. EMQX_HOST=$(grep -h "^$LOCAL_IP" /etc/hosts | grep -o "$(hostname).*.$EMQX_CLUSTER__K8S__NAMESPACE.$EMQX_CLUSTER__K8S__SUFFIX")
  25. else
  26. EMQX_HOST="$LOCAL_IP"
  27. fi
  28. fi
  29. export EMQX_NODE_NAME="$EMQX_NAME@$EMQX_HOST"
  30. unset EMQX_NAME
  31. unset EMQX_HOST
  32. fi
  33. # fill tuples on specific file
  34. # SYNOPSIS
  35. # fill_tuples FILE [ELEMENTS ...]
  36. fill_tuples() {
  37. local file=$1
  38. local elements=${*:2}
  39. for var in $elements; do
  40. if grep -qE "\{\s*$var\s*,\s*(true|false)\s*\}\s*\." "$file"; then
  41. sed -r "s/\{\s*($var)\s*,\s*(true|false)\s*\}\s*\./{\1, true}./1" "$file" > tmpfile && cat tmpfile > "$file"
  42. elif grep -q "$var\s*\." "$file"; then
  43. # backward compatible.
  44. sed -r "s/($var)\s*\./{\1, true}./1" "$file" > tmpfile && cat tmpfile > "$file"
  45. else
  46. sed '$a'\\ "$file" > tmpfile && cat tmpfile > "$file"
  47. echo "{$var, true}." >> "$file"
  48. fi
  49. done
  50. }
  51. # The default rpc port discovery 'stateless' is mostly for clusters
  52. # having static node names. So it's troulbe-free for multiple emqx nodes
  53. # running on the same host.
  54. # When start emqx in docker, it's mostly one emqx node in one container
  55. export EMQX_RPC__PORT_DISCOVERY="${EMQX_RPC__PORT_DISCOVERY:-manual}"
  56. exec "$@"