99 lines
2.4 KiB
Bash
99 lines
2.4 KiB
Bash
|
#!/bin/sh
|
||
|
|
|
||
|
|
pingSentinel() {
|
||
|
|
resp=$(timeout -s 15 $1 \
|
||
|
|
valkey-cli \
|
||
|
|
-h ${VALKEY_SERVICE} \
|
||
|
|
-p ${VALKEY_SENTINEL_PORT} \
|
||
|
|
ping)
|
||
|
|
ret=${?}
|
||
|
|
echo $resp
|
||
|
|
return ${ret}
|
||
|
|
}
|
||
|
|
|
||
|
|
getPrimaryInfo() {
|
||
|
|
valkey-cli --csv -h ${VALKEY_SERVICE} -p ${VALKEY_SENTINEL_PORT} sentinel get-primary-addr-by-name "mymaster"| \
|
||
|
|
awk -F ',' '{ gsub(/"/,"",$0); print $1 " " $2 }'
|
||
|
|
return ${?}
|
||
|
|
}
|
||
|
|
|
||
|
|
waitForSentinel() {
|
||
|
|
tout=60
|
||
|
|
while true; do
|
||
|
|
response=$(pingSentinel 5)
|
||
|
|
if [ "${response}" = "PONG" ]; then
|
||
|
|
echo "Sentinel is responding"
|
||
|
|
break
|
||
|
|
return 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "Sentinel is not responding [${response}]"
|
||
|
|
sleep 1
|
||
|
|
tout=$((tout - 1))
|
||
|
|
if [ "${tout}" -le 0 ]; then
|
||
|
|
echo "Sentinel ping timed out"
|
||
|
|
return 124
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
sleep 10
|
||
|
|
}
|
||
|
|
|
||
|
|
startValkey() {
|
||
|
|
# Start Valkey
|
||
|
|
echo "Running : [valkey-server ${@}]"
|
||
|
|
valkey-server ${@}
|
||
|
|
ret=${?}
|
||
|
|
if [ "${ret}" -ne 0 ]; then
|
||
|
|
echo "Failed to start Valkey"
|
||
|
|
exit ${ret}
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
setupPrimary=0
|
||
|
|
primaryHost=""
|
||
|
|
primaryPort=""
|
||
|
|
|
||
|
|
|
||
|
|
if [[ -f /etc/valkey/sentinel.conf ]]; then
|
||
|
|
primaryHost="$(awk '/monitor/ {print $4}' /etc/valkey/sentinel.conf)"
|
||
|
|
primaryPort="$(awk '/monitor/ {print $5}' /etc/valkey/sentinel.conf)"
|
||
|
|
echo "Found previous primary ${primaryHost}:${primaryPort}"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ ! -f /etc/valkey/replication.conf ]]; then
|
||
|
|
cp /etc/valkey/replication.conf.orig /etc/valkey/replication.conf
|
||
|
|
fi
|
||
|
|
|
||
|
|
waitForSentinel
|
||
|
|
ret=${?}
|
||
|
|
if [ "${ret}" -ne 0 ]; then
|
||
|
|
exit ${ret}
|
||
|
|
fi
|
||
|
|
|
||
|
|
primaryInfo=$(getPrimaryInfo)
|
||
|
|
if [ "${?}" -ne 0 ]; then
|
||
|
|
echo "No primary found, seting up node as primary"
|
||
|
|
startPrimary=1
|
||
|
|
else
|
||
|
|
primaryHost=$(echo ${primaryInfo} | awk -F ' ' '{print $1}')
|
||
|
|
primaryPort=$(echo ${primaryInfo} | awk -F ' ' '{print $2}')
|
||
|
|
echo "Primary host is : ${primaryHost}, port: ${primaryPort}"
|
||
|
|
currentHost=$(hostname -f)
|
||
|
|
echo "Current host is : ${currentHost}"
|
||
|
|
if [ "${primaryHost}" != "${currentHost}" ]; then
|
||
|
|
echo "Not the primary, setting up as replica"
|
||
|
|
startPrimary=0
|
||
|
|
else
|
||
|
|
echo "This is the primary"
|
||
|
|
startPrimary=1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ "${startPrimary}" -eq 1 ]; then
|
||
|
|
echo "Starting Valkey as primary"
|
||
|
|
cat $1
|
||
|
|
startValkey ${@}
|
||
|
|
else
|
||
|
|
echo "Starting Valkey as replica"
|
||
|
|
startValkey ${@} "--replicaof" "${primaryHost}" "${primaryPort}"
|
||
|
|
fi
|