wait-for-elasticsearch

#!/bin/sh

WAIT_HOST=$1
WAIT_PORT=$2
WAIT_LOOP=60
WAIT_DELAY=1

function is_healthy () {
  host=$1
  port=$2

  res=$(nc -z $host $port > /dev/null 2>&1)
  suc=$?
  if [ $suc != "0" ]
  then
    return 1
  fi

  res=$(curl -fsSL "$host:$port/_cat/health?h=status" 2> /dev/null)
  if [ "$res" != "green" ]
  then
    return 2
  fi

  return 0
}

for i in $(seq 1 $WAIT_LOOP)
do
  is_healthy $WAIT_HOST $WAIT_PORT
  retval=$?
  if [ $retval == "0" ]
  then
    echo success
    break
  elif [ $retval == "1" ]
  then
    echo "port_closed"
  elif [ $retval == "2" ]
  then
    echo "not_green"
  else
    echo "???"
  fi
  sleep $WAIT_DELAY
done