黑帽联盟

 找回密码
 会员注册
查看: 1612|回复: 0
打印 上一主题 下一主题

[基础服务] keepalived通知脚本

[复制链接]
yun 黑帽联盟官方人员 

920

主题

37

听众

1364

积分

超级版主

Rank: 8Rank: 8

  • TA的每日心情
    奋斗
    2019-10-18 11:20
  • 签到天数: 678 天

    [LV.9]以坛为家II

    下面的脚本可以接受选项,其中:

    -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. helpflag=0
    5. serviceflag=0
    6. modeflag=0
    7. addressflag=0
    8. notifyflag=0

    9. contact='root@localhost'

    10. Usage() {
    11.   echo "Usage: notify.sh [-m|--mode {mm|mb}] [-s|--service SERVICE1,...] <-a|--address VIP>  <-n|--notify {master|backup|falut}>"
    12.   echo "Usage: notify.sh -h|--help"
    13. }

    14. ParseOptions() {
    15.   local I=1;
    16.   if [ $# -gt 0 ]; then
    17.     while [ $I -le $# ]; do
    18.       case $1 in
    19.   -s|--service)
    20. [ $# -lt 2 ] && return 3
    21.          serviceflag=1
    22.          services=(`echo $2|awk -F"," '{for(i=1;i<=NF;i++) print $i}'`)
    23. shift 2 ;;
    24.   -h|--help)
    25.          helpflag=1
    26. return 0
    27.         shift
    28. ;;
    29.   -a|--address)
    30. [ $# -lt 2 ] && return 3
    31.     addressflag=1
    32. vip=$2
    33. shift 2
    34. ;;
    35.   -m|--mode)
    36. [ $# -lt 2 ] && return 3
    37. mode=$2
    38. shift 2
    39. ;;
    40.   -n|--notify)
    41. [ $# -lt 2 ] && return 3
    42. notifyflag=1
    43. notify=$2
    44. shift 2
    45. ;;
    46.   *)
    47. echo "Wrong options..."
    48. Usage
    49. return 7
    50. ;;
    51.        esac
    52.     done
    53.     return 0
    54.   fi
    55. }

    56. #workspace=$(dirname $0)

    57. RestartService() {
    58.   if [ ${#@} -gt 0 ]; then
    59.     for I in $@; do
    60.       if [ -x /etc/rc.d/init.d/$I ]; then
    61.         /etc/rc.d/init.d/$I restart
    62.       else
    63.         echo "$I is not a valid service..."
    64.       fi
    65.     done
    66.   fi
    67. }

    68. StopService() {
    69.   if [ ${#@} -gt 0 ]; then
    70.     for I in $@; do
    71.       if [ -x /etc/rc.d/init.d/$I ]; then
    72.         /etc/rc.d/init.d/$I stop
    73.       else
    74.         echo "$I is not a valid service..."
    75.       fi
    76.     done
    77.   fi
    78. }


    79. Notify() {
    80.     mailsubject="`hostname` to be $1: $vip floating"
    81.     mailbody="`date '+%F %H:%M:%S'`, vrrp transition, `hostname` changed to be $1."
    82.     echo $mailbody | mail -s "$mailsubject" $contact
    83. }


    84. # Main Function
    85. ParseOptions $@
    86. [ $? -ne 0 ] && Usage && exit 5

    87. [ $helpflag -eq 1 ] && Usage && exit 0

    88. if [ $addressflag -ne 1 -o $notifyflag -ne 1 ]; then
    89.   Usage
    90.   exit 2
    91. fi

    92. mode=${mode:-mb}

    93. case $notify in
    94. 'master')
    95.   if [ $serviceflag -eq 1 ]; then
    96.       RestartService ${services[*]}
    97.   fi
    98.   Notify master
    99.   ;;
    100. 'backup')
    101.   if [ $serviceflag -eq 1 ]; then
    102.     if [ "$mode" == 'mb' ]; then
    103.       StopService ${services[*]}
    104.     else
    105.       RestartService ${services[*]}
    106.     fi
    107.   fi
    108.   Notify backup
    109.   ;;
    110. 'fault')
    111.   Notify fault
    112.   ;;
    113. *)
    114.   Usage
    115.   exit 4
    116.   ;;
    117. 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"  
    复制代码


    帖子永久地址: 

    黑帽联盟 - 论坛版权1、本主题所有言论和图片纯属会员个人意见,与本论坛立场无关
    2、本站所有主题由该帖子作者发表,该帖子作者与黑帽联盟享有帖子相关版权
    3、其他单位或个人使用、转载或引用本文时必须同时征得该帖子作者和黑帽联盟的同意
    4、帖子作者须承担一切因本文发表而直接或间接导致的民事或刑事法律责任
    5、本帖部分内容转载自其它媒体,但并不代表本站赞同其观点和对其真实性负责
    6、如本帖侵犯到任何版权问题,请立即告知本站,本站将及时予与删除并致以最深的歉意
    7、黑帽联盟管理员和版主有权不事先通知发贴者而删除本文

    您需要登录后才可以回帖 登录 | 会员注册

    发布主题 !fastreply! 收藏帖子 返回列表 搜索
    回顶部