keepalived通知脚本

yun

下面的脚本可以接受选项，其中：

-s, --service SERVICE,...：指定服务脚本名称，当状态切换时可自动启动、重启或关闭此服务；

-a, --address VIP: 指定相关虚拟路由器的VIP地址；

-m, --mode {mm|mb}：指定虚拟路由的模型，mm表示主主，mb表示主备；它们表示相对于同一种服务而方，其VIP的工作类型；

-n, --notify {master|backup|fault}：指定通知的类型，即vrrp角色切换的目标角色；

-h, --help：获取脚本的使用帮助；

1. #!/bin/bash
   
2. # description: An example of notify script
   
3. # Usage: notify.sh -m|--mode {mm|mb} -s|--service SERVICE1,... -a|--address VIP  -n|--notify {master|backup|falut} -h|--help
   
4. 
   
5. helpflag=0
   
6. serviceflag=0
   
7. modeflag=0
   
8. addressflag=0
   
9. notifyflag=0
   
10. 
    
11. contact='root@localhost'
    
12. 
    
13. Usage() {
    
14.   echo "Usage: notify.sh [-m|--mode {mm|mb}] [-s|--service SERVICE1,...] <-a|--address VIP>  <-n|--notify {master|backup|falut}>"
    
15.   echo "Usage: notify.sh -h|--help"
    
16. }
    
17. 
    
18. ParseOptions() {
    
19.   local I=1;
    
20.   if [ $# -gt 0 ]; then
    
21.     while [ $I -le $# ]; do
    
22.       case $1 in
    
23.   -s|--service)
    
24. [ $# -lt 2 ] && return 3
    
25.          serviceflag=1
    
26.          services=(`echo $2|awk -F"," '{for(i=1;i<=NF;i++) print $i}'`)
    
27. shift 2 ;;
    
28.   -h|--help)
    
29.          helpflag=1
    
30. return 0
    
31.         shift
    
32. ;;
    
33.   -a|--address)
    
34. [ $# -lt 2 ] && return 3
    
35.     addressflag=1
    
36. vip=$2
    
37. shift 2
    
38. ;;
    
39.   -m|--mode)
    
40. [ $# -lt 2 ] && return 3
    
41. mode=$2
    
42. shift 2
    
43. ;;
    
44.   -n|--notify)
    
45. [ $# -lt 2 ] && return 3
    
46. notifyflag=1
    
47. notify=$2
    
48. shift 2
    
49. ;;
    
50.   *)
    
51. echo "Wrong options..."
    
52. Usage
    
53. return 7
    
54. ;;
    
55.        esac
    
56.     done
    
57.     return 0
    
58.   fi
    
59. }
    
60. 
    
61. #workspace=$(dirname $0)
    
62. 
    
63. RestartService() {
    
64.   if [ ${#@} -gt 0 ]; then
    
65.     for I in $@; do
    
66.       if [ -x /etc/rc.d/init.d/$I ]; then
    
67.         /etc/rc.d/init.d/$I restart
    
68.       else
    
69.         echo "$I is not a valid service..."
    
70.       fi
    
71.     done
    
72.   fi
    
73. }
    
74. 
    
75. StopService() {
    
76.   if [ ${#@} -gt 0 ]; then
    
77.     for I in $@; do
    
78.       if [ -x /etc/rc.d/init.d/$I ]; then
    
79.         /etc/rc.d/init.d/$I stop
    
80.       else
    
81.         echo "$I is not a valid service..."
    
82.       fi
    
83.     done
    
84.   fi
    
85. }
    
86. 
    
87. 
    
88. Notify() {
    
89.     mailsubject="`hostname` to be $1: $vip floating"
    
90.     mailbody="`date '+%F %H:%M:%S'`, vrrp transition, `hostname` changed to be $1."
    
91.     echo $mailbody | mail -s "$mailsubject" $contact
    
92. }
    
93. 
    
94. 
    
95. # Main Function
    
96. ParseOptions $@
    
97. [ $? -ne 0 ] && Usage && exit 5
    
98. 
    
99. [ $helpflag -eq 1 ] && Usage && exit 0
    
100. 
     
101. if [ $addressflag -ne 1 -o $notifyflag -ne 1 ]; then
     
102.   Usage
     
103.   exit 2
     
104. fi
     
105. 
     
106. mode=${mode:-mb}
     
107. 
     
108. case $notify in
     
109. 'master')
     
110.   if [ $serviceflag -eq 1 ]; then
     
111.       RestartService ${services[*]}
     
112.   fi
     
113.   Notify master
     
114.   ;;
     
115. 'backup')
     
116.   if [ $serviceflag -eq 1 ]; then
     
117.     if [ "$mode" == 'mb' ]; then
     
118.       StopService ${services[*]}
     
119.     else
     
120.       RestartService ${services[*]}
     
121.     fi
     
122.   fi
     
123.   Notify backup
     
124.   ;;
     
125. 'fault')
     
126.   Notify fault
     
127.   ;;
     
128. *)
     
129.   Usage
     
130.   exit 4
     
131.   ;;
     
132. esac

复制代码

在keepalived.conf配置文件中，其调用方法如下所示：

1. notify_master "/etc/keepalived/notify.sh -n master -a 172.16.100.1"  
   
2. notify_backup "/etc/keepalived/notify.sh -n backup -a 172.16.100.1"  
   
3. notify_fault "/etc/keepalived/notify.sh -n fault -a 172.16.100.1"  

复制代码