lvs健康检查脚本第三版

如下是学习完马哥视频lvs后改写的健康检查脚本第三版。利用工作之余三四个小时时间才把整个逻辑搞清楚，有时候自己都有点蒙圈，尤其是在写到while循环的时候。总的来说非常感谢马哥的慷慨解囊！脚本原稿及思路均来自马哥，自己又折腾几小时终于完成第三版！如有不妥之处，欢迎指出批评。

#!/bin/bash
#

VIP=192.168.28.50
CPORT=80
FAIL_BACK=127.0.0.1
RS=("192.168.28.11" "192.168.28.12")
#RSTATUS=("1" "1")
RW=("2" "1")
RPORT=80
TYPE=g
CHKLOOP=3
LOG=/var/log/ipvschk.log

#添加RS，添加成功返回0，否则返回1
function addrs() {
 ipvsadm -a -t $VIP:$CPORT -r $1:$RPORT -$TYPE -w $2
 [ $? -eq 0 ] && return 0 || return 1
}
#删除RS，删除成功返回0，否则返回1
function delrs() {
 ipvsadm -d -t $VIP:$CPORT -r $1:$RPORT
 [ $? -eq 0 ] && return 0 || return 1
}

#检测RS服务是否在线，注意一下这里面指的RS的服务，如果连续三次都监测不通，则返回1，否则返回0
function checkrs(){
 local I=1
 while [ $I -le $CHKLOOP ];do
  if curl --connect-timeout 1 http://$1 &>/dev/null ; then
   return 0
  fi
 let I++
 done
 return 1
}

#初始化RS在线状态，如果在线，设置节点初始化状态为1，否则为0
function initstatus(){
 local I
 local COUNT=0
 for I in ${RS[*]};do
  if checkrs $I ; then
   RSSTATUS[${COUNT}]=1
  else
   RSSTATUS[${COUNT}]=0
  fi
 let COUNT++
 done
}
#ipvs状态检查，如果已添加至ipvs则返回0，否则返回1
function rrstatus(){
 realsta=`ipvsadm -L -n | grep $1 | grep -v grep | wc -l`
 [ $realsta -ne 0 ] && return 0 || return 1
}

 

#进行初始化

initstatus

#判断ipvsadm是否启动，如果没有启动则直接退出，否则继续

if [ ! -e /var/lock/subsys/ipvsadm ]; then
 echo "`date +'%F %H:%M:%S'`, ipvs is stop." >> $LOG
 exit 1
fi

 

#在这里逻辑关系有点复杂，总体思路是:监测rs是否加入到ipvs中。如果未添加并且在线则添加；如果已添加并且不在线，则删除
while :; do
 let COUNT=0
 for I in ${RS[*]} ; do
  rrstatus $I
  [ $? -eq 0 ] &&  RRSTATUS=1 || RRSTATUS=0
  if checkrs $I; then
   if [ $RRSTATUS -eq 0 ]; then
    addrs $I ${RW[$COUNT]}
    if [ $? -eq 0 ] ; then 
     RSSTATUS[$COUNT]=1
     echo "`date +'%F %H:%M:%S'`, $I is back." >> $LOG
    else
     echo "`date +'%F %H:%M:%S'`, add $I faild." >> $LOG
    fi
   fi
  else
   if [ $RRSTATUS -eq 1 ] ; then
    delrs $I
    if [ $? -eq 0 ] ; then
     RSSTATUS[$COUNT]=0
     echo "`date +'%F %H:%M:%S'`, $I is gone." >> $LOG
    else
     echo "`date +'%F %H:%M:%S'`, del $I faild or $I already has gone." >> $LOG
    fi
   fi
  fi
  let COUNT++
 done

#如果服务都不在线，并且备用服务未添加至ipvs，则添加备用服务；一旦有服务在线，则删除已添加的备用节点
 if [ ${RSSTATUS[0]} -eq 0 -a ${RSSTATUS[1]} -eq 0 ];then
  rrstatus $FAIL_BACK
  if [ $? -eq 1 ]; then
   ipvsadm -a -t $VIP:$CPORT -r $FAIL_BACK -w 1
  fi
 else
  rrstatus $FAIL_BACK
  if [ $? -eq 0 ]; then
   ipvsadm -d -t $VIP:$CPORT -r $FAIL_BACK
  fi
 fi

#5s监测一次
 sleep 5
done