From 719777a4d4e9b33781c1df7903a6d64abe3cf3d3 Mon Sep 17 00:00:00 2001 From: Piotr Biernat Date: Tue, 6 Dec 2022 06:40:56 +0100 Subject: [PATCH] Added wait-for-it.sh script --- Dockerfile | 5 ++ api-eventbus/entrypoint.sh | 22 +++++ api-eventbus/wait-for-it.sh | 165 ++++++++++++++++++++++++++++++++++++ 3 files changed, 192 insertions(+) create mode 100755 api-eventbus/entrypoint.sh create mode 100755 api-eventbus/wait-for-it.sh diff --git a/Dockerfile b/Dockerfile index bdcdd46..b33f320 100644 --- a/Dockerfile +++ b/Dockerfile @@ -6,7 +6,12 @@ LABEL dev.egommerce.image.service="api-eventubus" LABEL dev.egommerce.image.version="1.0" COPY ./api-eventbus/etc /etc/rabbitmq +COPY ./api-eventbus/entrypoint.sh ./api-eventbus/wait-for-it.sh / RUN rabbitmq-plugins --offline enable rabbitmq_peer_discovery_consul +RUN chmod 755 /entrypoint.sh EXPOSE 5672 + +ENTRYPOINT ["/entrypoint.sh"] +CMD ["rabbitmq-server"] diff --git a/api-eventbus/entrypoint.sh b/api-eventbus/entrypoint.sh new file mode 100755 index 0000000..568b72c --- /dev/null +++ b/api-eventbus/entrypoint.sh @@ -0,0 +1,22 @@ +#!/usr/bin/env sh +set +e + +waitForService() +{ + ./wait-for-it.sh $1 -t 2 1>/dev/null 2>&1 + status=$? + while [ $status != 0 ] + do + echo "[x] wating for $1..." + sleep 1 + ./wait-for-it.sh $1 -t 2 1>/dev/null 2>&1 + status=$? + done +} + +waitForService "api-registry:8500" +waitForService "api-logger:24224" + +set -euo pipefail + +exec "$@" diff --git a/api-eventbus/wait-for-it.sh b/api-eventbus/wait-for-it.sh new file mode 100755 index 0000000..fff13c9 --- /dev/null +++ b/api-eventbus/wait-for-it.sh @@ -0,0 +1,165 @@ +#!/usr/bin/env sh +# Use this script to test if a given TCP host/port are available + +set -e + +cmdname=$(basename "$0") + +echoerr() { + if [ "$QUIET" -ne 1 ]; then + printf "%s\n" "$*" 1>&2; + fi +} + +usage() +{ + exitcode="$1" + cat << USAGE >&2 +Usage: + $cmdname host:port [-s] [-t timeout] [-- command args] + -h HOST | --host=HOST Host or IP under test + -p PORT | --port=PORT TCP port under test + Alternatively, you specify the host and port as host:port + -s | --strict Only execute subcommand if the test succeeds + -q | --quiet Don't output any status messages + -t TIMEOUT | --timeout=TIMEOUT + Timeout in seconds, zero for no timeout + -- COMMAND ARGS Execute command with args after the test finishes +USAGE + exit "$exitcode" +} + +wait_for() +{ + if [ "$TIMEOUT" -gt 0 ]; then + echoerr "$cmdname: waiting $TIMEOUT seconds for $HOST:$PORT" + else + echoerr "$cmdname: waiting for $HOST:$PORT without a timeout" + fi + start_ts=$(date +%s) + while true + do + nc -z "$HOST" "$PORT" >/dev/null 2>&1 + result=$? + if [ $result -eq 0 ]; then + end_ts=$(date +%s) + echoerr "$cmdname: $HOST:$PORT is available after $((end_ts - start_ts)) seconds" + break + fi + sleep 1 + done + return $result +} + +wait_for_wrapper() +{ + # In order to support SIGINT during timeout: http://unix.stackexchange.com/a/57692 + if [ "$QUIET" -eq 1 ]; then + timeout "$TIMEOUT" "$0" -q -child "$HOST":"$PORT" -t "$TIMEOUT" & + else + timeout "$TIMEOUT" "$0" --child "$HOST":"$PORT" -t "$TIMEOUT" & + fi + PID=$! + trap 'kill -INT -$PID' INT + wait $PID + RESULT=$? + if [ $RESULT -ne 0 ]; then + echoerr "$cmdname: timeout occurred after waiting $TIMEOUT seconds for $HOST:$PORT" + fi + return $RESULT +} + +TIMEOUT=15 +STRICT=0 +CHILD=0 +QUIET=0 +# process arguments +while [ $# -gt 0 ] +do + case "$1" in + *:* ) + HOST=$(printf "%s\n" "$1"| cut -d : -f 1) + PORT=$(printf "%s\n" "$1"| cut -d : -f 2) + shift 1 + ;; + --child) + CHILD=1 + shift 1 + ;; + -q | --quiet) + QUIET=1 + shift 1 + ;; + -s | --strict) + STRICT=1 + shift 1 + ;; + -h) + HOST="$2" + if [ "$HOST" = "" ]; then break; fi + shift 2 + ;; + --host=*) + HOST=$(printf "%s" "$1" | cut -d = -f 2) + shift 1 + ;; + -p) + PORT="$2" + if [ "$PORT" = "" ]; then break; fi + shift 2 + ;; + --port=*) + PORT="${1#*=}" + shift 1 + ;; + -t) + TIMEOUT="$2" + if [ "$TIMEOUT" = "" ]; then break; fi + shift 2 + ;; + --timeout=*) + TIMEOUT="${1#*=}" + shift 1 + ;; + --) + shift + break + ;; + --help) + usage 0 + ;; + *) + echoerr "Unknown argument: $1" + usage 1 + ;; + esac +done + +if [ "$HOST" = "" -o "$PORT" = "" ]; then + echoerr "Error: you need to provide a host and port to test." + usage 2 +fi + +if [ $CHILD -gt 0 ]; then + wait_for + RESULT=$? + exit $RESULT +else + if [ "$TIMEOUT" -gt 0 ]; then + wait_for_wrapper + RESULT=$? + else + wait_for + RESULT=$? + fi +fi + +if [ "$*" != "" ]; then + if [ $RESULT -ne 0 -a $STRICT -eq 1 ]; then + echoerr "$cmdname: strict mode, refusing to execute subprocess" + exit $RESULT + fi + exec "$@" +else + exit $RESULT +fi