deliveries/wait-for.sh

   1 #!/bin/sh
   2 # https://github.com/Eficode/wait-for.git
   3 # MIT License
   4
   5 TIMEOUT=15
   6 QUIET=0
   7
   8 echoerr() {
   9   if [ "$QUIET" -ne 1 ]; then printf "%s\n" "$*" 1>&2; fi
  10 }
  11
  12 usage() {
  13   exitcode="$1"
  14   cat << USAGE >&2
  15 Usage:
  16   $cmdname host:port [-t timeout] [-- command args]
  17   -q | --quiet                        Do not output any status messages
  18   -t TIMEOUT | --timeout=timeout      Timeout in seconds, zero for no timeout
  19   -- COMMAND ARGS                     Execute command with args after the test finishes
  20 USAGE
  21   exit "$exitcode"
  22 }
  23
  24 wait_for() {
  25   command="$*"
  26   if [ "$QUIET" -ne 1 ]; then echo "$0: probing host $HOST port $PORT"; fi
  27   for i in `seq $TIMEOUT` ; do
  28     nc -z "$HOST" "$PORT" > /dev/null 2>&1
  29     result=$?
  30     if [ $result -eq 0 ] ; then
  31       if [ "$QUIET" -ne 1 ]; then echo "$0: operation succeeded on try $i"; fi
  32       if [ -n "$command" ] ; then
  33         if [ "$QUIET" -ne 1 ]; then echo "$0: exec-ing command $command"; fi
  34         exec $command
  35       fi
  36       exit 0
  37     fi
  38     if [ "$QUIET" -ne 1 ]; then echo "$0: sleeping after try $i"; fi
  39     sleep 1
  40   done
  41   echo "$0: Operation timed out" >&2
  42   exit 1
  43 }
  44
  45 while [ $# -gt 0 ]
  46 do
  47   case "$1" in
  48     *:* )
  49     HOST=$(printf "%s\n" "$1"| cut -d : -f 1)
  50     PORT=$(printf "%s\n" "$1"| cut -d : -f 2)
  51     shift 1
  52     ;;
  53     -q | --quiet)
  54     QUIET=1
  55     shift 1
  56     ;;
  57     -t)
  58     TIMEOUT="$2"
  59     if [ "$TIMEOUT" = "" ]; then break; fi
  60     shift 2
  61     ;;
  62     --timeout=*)
  63     TIMEOUT="${1#*=}"
  64     shift 1
  65     ;;
  66     --)
  67     shift
  68     break
  69     ;;
  70     --help)
  71     usage 0
  72     ;;
  73     *)
  74     echoerr "Unknown argument: $1"
  75     usage 1
  76     ;;
  77   esac
  78 done
  79
  80 if [ "$HOST" = "" -o "$PORT" = "" ]; then
  81   echoerr "Error: you need to provide a host and port to test."
  82   usage 2
  83 fi
  84
  85 wait_for "$@"