Add strict option

Author: Lhuckaz

wait-for

@@ -23,6 +23,7 @@ usage() {
 Usage:
   $cmdname host:port [-t timeout] [-- command args]
   -q | --quiet                        Do not output any status messages
+  -s | --strict                       Only execute subcommand if the test succeeds
   -t TIMEOUT | --timeout=timeout      Timeout in seconds, zero for no timeout
   -- COMMAND ARGS                     Execute command with args after the test finishes
 USAGE
@@ -43,6 +44,12 @@ wait_for() {
     sleep 1
   done
   echo "Operation timed out" >&2
+  if [ $result -ne 0 ] && [ $STRICT -ne 1 ] ; then
+    if [ $# -gt 0 ] ; then
+      exec "$@"
+    fi
+    exit 0
+  fi
   exit 1
 }
 
@@ -58,6 +65,10 @@ do
     QUIET=1
     shift 1
     ;;
+    -s | --strict)
+    STRICT=1
+    shift 1
+    ;;
     -t)
     TIMEOUT="$2"
     if [ "$TIMEOUT" = "" ]; then break; fi
@@ -86,4 +97,6 @@ if [ "$HOST" = "" -o "$PORT" = "" ]; then
   usage 2
 fi
 
+STRICT=${STRICT:-0}
+
 wait_for "$@"