keepalive笔记之三:keepalived通知脚本进阶示例

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

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

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

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

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

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

#!/bin/bash
# Author: MageEdu <linuxedu@foxmail.com>
# description: An example of notify script
# Usage: notify.sh -m|--mode {mm|mb} -s|--service SERVICE1,... -a|--address VIP  -n|--notify {master|backup|falut} -h|--help 

#contact='linuxedu@foxmail.com'
helpflag=
serviceflag=
modeflag=
addressflag=
notifyflag=

contact='root@localhost'

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

ParseOptions() {
  local I=;
  if [ $# -gt  ]; then
    while [ $I -le $# ]; do
      case $ in
  -s|--service)
[ $# -lt  ] && return
         serviceflag=
         services=(`echo $|awk -F"," '{for(i=1;i<=NF;i++) print $i}'`)
shift  ;;
  -h|--help)
         helpflag=
return
        shift
;;
  -a|--address)
[ $# -lt  ] && return
    addressflag=
vip=$
shift
;;
  -m|--mode)
[ $# -lt  ] && return
mode=$
shift
;;
  -n|--notify)
[ $# -lt  ] && return
notifyflag=
notify=$
shift
;;
  *)
echo "Wrong options..."
Usage
return
;;
       esac
    done
    return
  fi
}

#workspace=$(dirname $)

RestartService() {
  if [ ${#@} -gt  ]; then
    for I in $@; do
      if [ -x /etc/rc.d/init.d/$I ]; then
        /etc/rc.d/init.d/$I restart
      else
        echo "$I is not a valid service..."
      fi
    done
  fi
}

StopService() {
  if [ ${#@} -gt  ]; then
    for I in $@; do
      if [ -x /etc/rc.d/init.d/$I ]; then
        /etc/rc.d/init.d/$I stop
      else
        echo "$I is not a valid service..."
      fi
    done
  fi
}

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

# Main Function
ParseOptions $@
[ $? -ne  ] && Usage && exit 

[ $helpflag -eq  ] && Usage && exit 

if [ $addressflag -ne  -o $notifyflag -ne  ]; then
  Usage
  exit
fi

mode=${mode:-mb}

case $notify in
'master')
  if [ $serviceflag -eq  ]; then
      RestartService ${services[*]}
  fi
  Notify master
  ;;
'backup')
  if [ $serviceflag -eq  ]; then
    if [ "$mode" == 'mb' ]; then
      StopService ${services[*]}
    else
      RestartService ${services[*]}
    fi
  fi
  Notify backup
  ;;
'fault')
  Notify fault
  ;;
*)
  Usage
  exit
  ;;
esac

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

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