docker-entrypoint.sh 2.0 KB

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