blob: 62dd9b684ff79101085dffb712ef50e3b67602b4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
#!/bin/sh
RETRY_INTERVAL=5 # Interval between retries in seconds
elapsed=0 # Elapsed time
check_redis() {
host=$1
port=$2
while [ $elapsed -lt $TOTAL_RETRY_TIME ]; do
echo "Checking Redis at $host:$port... Elapsed time: ${elapsed}s"
if nc -z -w1 $TIMEOUT $host $port > /dev/null 2>&1; then
echo "Redis is up at $host:$port!"
return 0
else
echo "Redis is down at $host:$port. Retrying in $RETRY_INTERVAL seconds."
sleep $RETRY_INTERVAL
elapsed=$((elapsed + RETRY_INTERVAL))
fi
done
echo "Failed to connect to Redis at $host:$port after $TOTAL_RETRY_TIME seconds."
return 1
}
# For parsing and checking connections
parse_and_check() {
url=$1
# Strip either redis:// or rediss://
if [ $url = "rediss://*" ]; then
clean_url=${url#rediss://}
echo "Using secure Rediss connection..."
else
clean_url=${url#redis://}
echo "Using standard Redis connection..."
fi
host=$(echo $clean_url | cut -d':' -f1)
port=$(echo $clean_url | cut -d':' -f2)
check_redis $host $port
}
# Main
if [ -n "$OAUTH2_PROXY_REDIS_CLUSTER_CONNECTION_URLS" ]; then
echo "Checking Redis in cluster mode..."
echo "$OAUTH2_PROXY_REDIS_CLUSTER_CONNECTION_URLS" | tr ',' '\n' | while read -r addr; do
parse_and_check $addr || exit 1
done
elif [ -n "$OAUTH2_PROXY_REDIS_SENTINEL_CONNECTION_URLS" ]; then
echo "Checking Redis in sentinel mode..."
echo "$OAUTH2_PROXY_REDIS_SENTINEL_CONNECTION_URLS" | tr ',' '\n' | while read -r addr; do
parse_and_check $addr || exit 1
done
elif [ -n "$OAUTH2_PROXY_REDIS_CONNECTION_URL" ]; then
echo "Checking standalone Redis..."
parse_and_check "$OAUTH2_PROXY_REDIS_CONNECTION_URL" || exit 1
else
echo "Redis configuration not specified."
exit 1
fi
echo "Redis check completed."
|
