|
| 1 | +#!/bin/sh |
| 2 | +set -e |
| 3 | + |
| 4 | +# Wait for service to be available using nc |
| 5 | +wait_for_service() { |
| 6 | + ( |
| 7 | + host="$1" |
| 8 | + port="$2" |
| 9 | + service_name="$3" |
| 10 | + timeout="${WAITSTART_TIMEOUT:-60s}" |
| 11 | + |
| 12 | + # Convert timeout to seconds (handle formats like "1m", "60s", "60") |
| 13 | + timeout_seconds=60 |
| 14 | + if echo "$timeout" | grep -q "m$"; then |
| 15 | + timeout_seconds=$(echo "$timeout" | sed 's/m$//' | awk '{print $1 * 60}') |
| 16 | + elif echo "$timeout" | grep -q "s$"; then |
| 17 | + timeout_seconds=$(echo "$timeout" | sed 's/s$//') |
| 18 | + else |
| 19 | + timeout_seconds="$timeout" |
| 20 | + fi |
| 21 | + |
| 22 | + elapsed=0 |
| 23 | + interval=1 |
| 24 | + |
| 25 | + echo "Waiting for $service_name at $host:$port..." |
| 26 | + while [ "$elapsed" -lt "$timeout_seconds" ]; do |
| 27 | + if nc -z "$host" "$port" 2>/dev/null; then |
| 28 | + echo "$service_name is available" |
| 29 | + exit 0 |
| 30 | + fi |
| 31 | + sleep "$interval" |
| 32 | + elapsed=$((elapsed + interval)) |
| 33 | + done |
| 34 | + |
| 35 | + echo "Error: Timeout waiting for $service_name at $host:$port" |
| 36 | + exit 1 |
| 37 | + ) |
| 38 | +} |
| 39 | + |
| 40 | +# Wait for MySQL |
| 41 | +if [ -z "$MYSQL_HOST" ] || [ -z "$MYSQL_PORT" ]; then |
| 42 | + echo "Error: MYSQL_HOST or MYSQL_PORT not set" |
| 43 | + exit 1 |
| 44 | +fi |
| 45 | +wait_for_service "$MYSQL_HOST" "$MYSQL_PORT" "MySQL" |
| 46 | + |
| 47 | +# Wait for MDA LMTP |
| 48 | +if [ -z "$MDA_LMTP_ADDRESS" ]; then |
| 49 | + echo "Error: MDA_LMTP_ADDRESS not set" |
| 50 | + exit 1 |
| 51 | +fi |
| 52 | +MDA_LMTP_HOST=$(echo "$MDA_LMTP_ADDRESS" | cut -d: -f1) |
| 53 | +MDA_LMTP_PORT=$(echo "$MDA_LMTP_ADDRESS" | cut -d: -f2) |
| 54 | +wait_for_service "$MDA_LMTP_HOST" "$MDA_LMTP_PORT" "MDA LMTP" |
| 55 | + |
| 56 | +# Wait for Filter Milter |
| 57 | +if [ -z "$FILTER_MILTER_ADDRESS" ]; then |
| 58 | + echo "Error: FILTER_MILTER_ADDRESS not set" |
| 59 | + exit 1 |
| 60 | +fi |
| 61 | +FILTER_MILTER_HOST=$(echo "$FILTER_MILTER_ADDRESS" | cut -d: -f1) |
| 62 | +FILTER_MILTER_PORT=$(echo "$FILTER_MILTER_ADDRESS" | cut -d: -f2) |
| 63 | +wait_for_service "$FILTER_MILTER_HOST" "$FILTER_MILTER_PORT" "Filter Milter" |
0 commit comments