docker-entrypoint.sh 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  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. export EMQX_NAME="${EMQX_NAME:-emqx}"
  15. ## EMQX_NODE_NAME or EMQX_NODE__NAME to indicate the full node name to be used by EMQX
  16. ## If both are set EMQX_NODE_NAME takes higher precedence than EMQX_NODE__NAME
  17. if [[ -z "${EMQX_NODE_NAME:-}" ]] && [[ -z "${EMQX_NODE__NAME:-}" ]]; then
  18. # No node name is provide from environment variables
  19. # try to resolve from other settings
  20. if [[ -z "$EMQX_HOST" ]]; then
  21. if [[ "$EMQX_CLUSTER__DISCOVERY_STRATEGY" == "dns" ]] && \
  22. [[ "$EMQX_CLUSTER__DNS__RECORD_TYPE" == "srv" ]] && \
  23. grep -q "$(hostname).$EMQX_CLUSTER__DNS__NAME" /etc/hosts; then
  24. EMQX_HOST="$(hostname).$EMQX_CLUSTER__DNS__NAME"
  25. elif [[ "$EMQX_CLUSTER__DISCOVERY_STRATEGY" == "k8s" ]] && \
  26. [[ "$EMQX_CLUSTER__K8S__ADDRESS_TYPE" == "dns" ]] && \
  27. [[ -n "$EMQX_CLUSTER__K8S__NAMESPACE" ]]; then
  28. EMQX_CLUSTER__K8S__SUFFIX=${EMQX_CLUSTER__K8S__SUFFIX:-"pod.cluster.local"}
  29. EMQX_HOST="${LOCAL_IP//./-}.$EMQX_CLUSTER__K8S__NAMESPACE.$EMQX_CLUSTER__K8S__SUFFIX"
  30. elif [[ "$EMQX_CLUSTER__DISCOVERY_STRATEGY" == "k8s" ]] && \
  31. [[ "$EMQX_CLUSTER__K8S__ADDRESS_TYPE" == 'hostname' ]] && \
  32. [[ -n "$EMQX_CLUSTER__K8S__NAMESPACE" ]]; then
  33. EMQX_CLUSTER__K8S__SUFFIX=${EMQX_CLUSTER__K8S__SUFFIX:-'svc.cluster.local'}
  34. EMQX_HOST=$(grep -h "^$LOCAL_IP" /etc/hosts | grep -o "$(hostname).*.$EMQX_CLUSTER__K8S__NAMESPACE.$EMQX_CLUSTER__K8S__SUFFIX")
  35. else
  36. EMQX_HOST="$LOCAL_IP"
  37. fi
  38. export EMQX_HOST
  39. fi
  40. export EMQX_NODE_NAME="$EMQX_NAME@$EMQX_HOST"
  41. fi
  42. # The default rpc port discovery 'stateless' is mostly for clusters
  43. # having static node names. So it's troulbe-free for multiple emqx nodes
  44. # running on the same host.
  45. # When start emqx in docker, it's mostly one emqx node in one container
  46. # i.e. use port 5369 (or per tcp_server_port | ssl_server_port config) for gen_rpc
  47. export EMQX_RPC__PORT_DISCOVERY="${EMQX_RPC__PORT_DISCOVERY:-manual}"
  48. exec "$@"