-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun.sh
68 lines (57 loc) · 1.84 KB
/
run.sh
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
62
63
64
65
66
67
68
#!/bin/bash
exec 2>&1
export PIDFILE="/tmp/haproxy.pid"
addFirewallRules() {
IFS=',' read -ra ADDR <<< "$PORTS"
for i in "${ADDR[@]}"; do
iptables -w -I INPUT -p tcp --dport $i --syn -j DROP
done
}
removeFirewallRules() {
IFS=',' read -ra ADDR <<< "$PORTS"
for i in "${ADDR[@]}"; do
while iptables -w -D INPUT -p tcp --dport $i --syn -j DROP 2>/dev/null; do :; done
done
}
kill () {
echo "Killing HaProxy `date +'%D %T'`"
(
flock 200
PID=`pidof haproxy`
echo "We block health check port and wait 30 seconds for the consul health checks fails"
iptables -w -I INPUT -p tcp --dport 4444 -j REJECT 2>/dev/null && sleep 30
echo "We remove the iptables rule and we will send SIGTERM to $PID "
iptables -w -D INPUT -p tcp --dport 4444 -j REJECT 2>/dev/null
kill -15 ${PID}
) 200>/var/run/haproxy/lock
}
reload() {
echo "Reloading haproxy `date +'%D %T'`"
(
flock 200
#Check configuration file before to reload process
haproxy -f /haproxy.cfg -c
# Begin to drop SYN packets with firewall rules
addFirewallRules
# Wait to settle
sleep 0.1
# Save the current HAProxy state
socat /var/run/haproxy/socket - <<< "show servers state" > /var/state/haproxy/global
# Trigger reload
LATEST_HAPROXY_PID=$(cat $PIDFILE)
haproxy -p $PIDFILE -f /haproxy.cfg -D -sf $LATEST_HAPROXY_PID 200>&-
if [ -n "${HAPROXY_RELOAD_SIGTERM_DELAY-}" ]; then
sleep $HAPROXY_RELOAD_SIGTERM_DELAY && kill $LATEST_HAPROXY_PID 200>&- 2>/dev/null &
fi
# Remove the firewall rules
removeFirewallRules
# Need to wait 1s to prevent TCP SYN exponential backoff
sleep 1
) 200>/var/run/haproxy/lock
}
mkdir -p /var/state/haproxy
mkdir -p /var/run/haproxy
reload
trap reload SIGHUP
trap kill SIGTERM
while true; do /bin/sleep 0.5; done