shell编程（二）

第三十二次课 shell编程（二）

目录

十五、shell中的函数
十六、shell中的数组
十七、告警系统需求分析
十八、告警系统主脚本
十九、告警系统配置文件
二十、告警系统监控项目
二十一、告警系统邮件引擎
二十二、运行告警系统

十五、shell中的函数

函数就是把一段代码整理到了一个小单元中，并给这个小单元起一个名字，当用到这段代码时直接调用这个小单元的名字即可。定义函数必须要放在最前面。定义好的函数相当于是命令。

语法格式

//name属性定义了赋予函数的唯一名称。脚本中定义的每个函数都必须有一个唯一的名称。
function name {
	commands
}

或者

//函数名称后的括号表明正在定义的是一个函数。
name(){
	commands
}

例1

#!/bin/bash

function print_para{
    echo "The first pamameter is $1"
    echo "The second pamameter is $2"
    echo "The third pamameter is $r3"
    echo "The script name  is $0"
    echo "The num pamameter is $#"
}

print_para 1 2 3

[root@lanquark sh]# sh para_list.sh 
The first pamameter is 1
The second pamameter is 2
The third pamameter is 
The script name  is para_list.sh
The num pamameter is 3

例2

[root@lanquark sh]# vim sum.sh
#!/bin/bash

function sum {
    sum=$(($1+$2))
    echo $sum
}

sum 2 3

[root@lanquark sh]# sh sum.sh 
5

例3

[root@lanquark sh]# vim ip.sh
#!/bin/bash

getip() {
        ifconfig $1 | awk 'NR==2{print $2}'
}

read -p "Please enter the eth name: " interface

myip=$(getip $interface)

echo "$interface ip addr is $myip"

[root@lanquark sh]# sh ip.sh            
Please enter the eth name: ens32
ens32 ip addr is 192.168.1.29

规范用户输入信息 并打出对应的IP地址 https://github.com/aminglinux/shell/blob/master/if_ip.txt

十六、shell中的数组

数组允许将一列词放到一个变量名中，如一列数、一列名称或一列文件。

语法格式

//declare明确声明部分可以省略
declare -a variable_name
variable = (item1 item2 item3 ...)

[root@lanquark sh]# declare -a friends
[root@lanquark sh]# friends=(shery1 Peter Louise)
[root@lanquark sh]# echo ${friends[0]}
shery1
[root@lanquark sh]# echo ${friends[1]}
Peter
[root@lanquark sh]# echo ${friends[2]}
Louise

或者

直接给变量名一个下标来创建

[root@lanquark sh]# x[3]=100
[root@lanquark sh]# echo ${x[*]}
100
[root@lanquark sh]# echo ${x[0]}

[root@lanquark sh]# echo ${x[1]}

[root@lanquark sh]# echo ${x[3]}
100

调用数组 ${a[@]}或者${a[*]}

[root@lanquark sh]# a=(1 2 3 4 5)
[root@lanquark sh]# echo ${a[*]}
1 2 3 4 5
[root@lanquark sh]# echo ${a[@]}
1 2 3 4 5

查看对应下标的元素值 默认从0开始 ${a[下标]}

[root@lanquark sh]# a=(1 2 3 4 5)
[root@lanquark sh]# echo ${a[0]}
1
[root@lanquark sh]# echo ${a[1]} 
2
[root@lanquark sh]# echo ${a[2]}
3

获取元素的个数

[root@lanquark sh]# echo ${#a[@]}
5

添加/更新元素 a[下标]=值

[root@lanquark sh]# a[1]=100
[root@lanquark sh]# echo ${a[@]}
1 100 3 4 5

添加/更新元素 a[下标]=值

[root@lanquark sh]# arr_tt=($(seq 1 10))
[root@lanquark sh]# echo ${arr_tt[@]}
1 2 3 4 5 6 7 8 9 10
//从第5个元素开始取，取3个元素
[root@lanquark sh]# echo ${arr_tt[@]:5:3}
6 7 8
//从倒数第9个元素开始取，取2个
[root@lanquark sh]# echo ${arr_tt[@]:0-9:2}
2 3

数组替换 ${a[@]/用来的值/替换后的值}

[root@lanquark sh]# echo ${arr_tt[@]/3/100}
1 2 100 4 5 6 7 8 9 10

十七、告警系统需求分析

需求：使用shell定制各种个性化告警工具，但需要统一化管理、规范化管理。

思路：指定一个脚本包，包含主程序、子程序、配置文件、邮件引擎、输出日志等。

主程序：作为整个脚本的入口，是整个系统的命脉。

配置文件：是一个控制中心，用它来开关各个子程序，指定各个相关联的日志文件。

子程序：这个才是真正的监控脚本，用来监控各个指标。

邮件引擎：是由一个python程序来实现，它可以定义发邮件的服务器、发邮件人以及发件人密码

输出日志：整个监控系统要有日志输出

要求：我们的机器角色多种多样，但是所有机器上都要部署同样的监控系统，也就说所有机器不管什么角色，整个程序框架都是一致的，不同的地方在于根据不同的角色，定制不同的配置文件

程序架构：

bin下是主程序

conf下是配置文件

shares下是各个监控脚本

mail下是邮件引擎

log下是日志。

十八、告警系统主脚本

根据程序架构，创建相应的目录结构

[root@lanquark sh]# cd /usr/local/sbin/
[root@lanquark sbin]# mkdir -p mon/{bin,conf,shares,mail,log}
[root@lanquark sbin]# tree
.
└── mon
    ├── bin
    ├── conf
    ├── log
    ├── mail
    └── shares
在/usr/local/sbin/ 目录下存放监控脚本，方便查找脚本

在mon/bin下创建主程序

[root@lanquark sbin]# vim mon/bin/mail.sh

#!/bin/bash
#export 表示该变量会应用到所有的子脚本中（主脚本所调用的脚本）
#发送邮件的开关
export send=1
#发送邮件机器的ip地址
export addr=`/usr/sbin/ifconfig |grep -A1 "ens33" |tail -1 |awk '{print $2}'`
#当前所在的路径
dir=`pwd`
#找出当前路径的最后一个目录
last_dir=`echo $dir |awk -F '/' '{print $NF}'`
#下面的判断目的是，保证执行脚本的时候，我们在bin目录里，不然监控脚本、邮件和日志很有可能找不到
#因为他们使用的是相对路径
if [ $last_dir == "bin" ] || [ $last_dir == "bin/" ]
then
#定义配置文件路径
        conf_file="../conf/mon.conf"
else
        echo "Please cd bin dir"
        exit
fi
#定义正确和错误日志的路径
exec 1>>../log/mon.log 2>>../log/err.log
#把下面内容输入到日志里
echo "`date +"%F %T"` load average"
#调用load.sh脚本来监控系统负载
/bin/bash ../shares/load.sh
#先检查配置文件中是否需要监控502
if grep -q 'to_mon_502=1' $conf_file
then
#定义网站的访问日志路径
    export log=`grep 'logfile=' $conf_file |awk -F '=' '{print $2}' |sed 's/ //g'`
#调用502脚本
    /bin/bash  ../shares/502.sh
fi
~
"mon/bin/mail.sh" [New] 35L, 1202C written
[root@lanquark sbin]# chmod +x mon/bin/mail.sh 

十九、告警系统配置文件

告警系统配置文件如下

## to config the options if to monitor
## 定义mysql的服务器地址、端口以及user、password
to_mon_cdb=0   ##0 or 1, default 0,0 not monitor, 1 monitor
db_ip=10.20.3.13
db_port=3315
db_user=username
db_pass=passwd
## httpd   如果是1则监控，为0不监控
to_mon_httpd=0
## php 如果是1则监控，为0不监控
to_mon_php_socket=0
## http_code_502  需要定义访问日志的路径
to_mon_502=1
logfile=/data/log/xxx.xxx.com/access.log
## request_count   定义日志路径以及域名
to_mon_request_count=0
req_log=/data/log/www.discuz.net/access.log
domainname=www.discuz.net

二十、告警系统监控项目

创建load.sh 监控系统负载脚本

[root@lanquark sbin]# vim mon/shares/load.sh
#! /bin/bash
##Writen by aming##
#load=$(uptime |awk -F 'average:' '{print $2}'|cut -d',' -f1|sed 's/ //g' |cut -d. -f1)
load=$(uptime | awk -F 'load average' '{print $2}' | awk -F'[: .]' '{print $3}')
if [ $load -gt 10 ] && [ $send -eq "1" ]
then
    echo "$addr $(date +%T) load is $load" >../log/load.tmp
    /bin/bash ../mail/mail.sh kennminn@139.com "${addr}_load:$load" $(cat ../log/load.tmp)
fi
echo "$(date +%T) load is $load"

创建502.sh 监控502脚本

[root@lanquark sbin]# vim mon/shares/502.sh
#! /bin/bash
d=$(date -d "-1 min" +%H:%M)
c_502=$(grep :$d:  $log  |grep ' 502 '|wc -l)
if [ $c_502 -gt 10 ] && [ $send == 1 ]; then
     echo "$addr $d 502 count is $c_502">../log/502.tmp
     /bin/bash ../mail/mail.sh ${addr}_502 $c_502  ../log/502.tmp
fi
echo "$(date +%T) 502 $c_502"

创建disk.sh 监控磁盘使用率脚本

[root@lanquark shares]# vim disk.sh 
#! /bin/bash
##Writen by aming##
rm -f ../log/disk.tmp
for r in $(df -h |awk -F '[ %]+' '{print $5}'|grep -v Use)
do
    if [ $r -gt 90 ] && [ $send -eq "1" ]
    then
        echo "$addr $(date +%T) disk usage is $r" >>../log/disk.tmp
    fi
done
if [ -f ../log/disk.tmp ]
then
    df -h >>../log/disk.tmp
    /bin/bash ../mail/mail.sh ${addr}_disk $r ../log/disk/disk.tmp
    echo "$(date +%T) disk usage is too high."
else
    echo "$(date +%T) disk usage is ok"
fi

二十一、告警系统邮件引擎

mail.py参考内容在 https://coding.net/u/aminglinux/p/aminglinux-book/git/blob/master/D22Z/mail.py

#!/usr/bin/env python
#-*- coding: UTF-8 -*-
import os,sys
reload(sys)
sys.setdefaultencoding('utf8')
import getopt
import smtplib
from email.MIMEText import MIMEText
from email.MIMEMultipart import MIMEMultipart
from  subprocess import *

def sendqqmail(username,password,mailfrom,mailto,subject,content):
    gserver = 'smtp.qq.com'
    gport = 25

    try:
        msg = MIMEText(unicode(content).encode('utf-8'))
        msg['from'] = mailfrom
        msg['to'] = mailto
        msg['Reply-To'] = mailfrom
        msg['Subject'] = subject

        smtp = smtplib.SMTP(gserver, gport)
        smtp.set_debuglevel(0)
        smtp.ehlo()
        smtp.login(username,password)

        smtp.sendmail(mailfrom, mailto, msg.as_string())
        smtp.close()
    except Exception,err:
        print "Send mail failed. Error: %s" % err


def main():
    to=sys.argv[1]
    subject=sys.argv[2]
    content=sys.argv[3]
##定义QQ邮箱的账号和密码，你需要修改成你自己的账号和密码（请不要把真实的用户名和密码放到网上公开，否则你会死的很惨）
    sendqqmail('1234567@qq.com','aaaaaaaaaa','1234567@qq.com',to,subject,content)

if __name__ == "__main__":
    main()
    
//脚本需要接收三个参数： 收件人，主题，内容
//用法  mail.py 收件人 主题  内容

创建mail.sh 调用发邮件和邮件收敛的脚本

[root@lanquark sbin]# vim mon/mail/mail.sh 
log=$1
t_s=$(date +%s)
t_s2=$(date -d "2 hours ago" +%s)
if [ ! -f /tmp/$log ]
then
    echo $t_s2 > /tmp/$log
fi
t_s2=$(tail -1 /tmp/$log | awk '{print $1}')
echo $t_si >>/tmp/$log
v=$[$t_s-$t_s2]
echo $v
if [ $v -gt 3600 ]
then
    ./mail.py  $1  $2  $3
    echo "0" > /tmp/$log.txt
else
    if [ ! -f /tmp/$log.txt ]
    then
        echo "0" > /tmp/$log.txt
    fi
    nu=$(cat /tmp/$log.txt)
    nu2=$((nu+1))
    echo $nu2 >/tmp/$log.txt
    if [ $nu2 -gt 10 ]
    then
         ./mail.py  $1 "trouble continue 10 min $2" "$3"
         echo "0" > /tmp/$log.txt
    fi
fi  

//tmp/$log是计时器日志 /tmp/$log.txt是计数器日志
//第一次执行mail.sh t_s是当前时间戳 t_s2是两个小时前的时间戳 判断差值大于3600s 发邮件 并计数器为0
//1分钟后再次执行mail.sh t_s是当前时间戳 t_s2为1分钟前的时间戳判断差值小于3600s 计数器为1
//每分钟都执行mail.sh 到了第11分钟 因为 t_s和t_s2之前的差值都是小于3600s是不会发邮件的 当时 此时计数器为11 大于10 就发一份“问题持续10分//钟”的邮件 并计数器清零
//假如超多一个小时后再此执行mail.sh t_s和t_s2的时间戳 判断差值大于3600s 就会发邮件并会清空原来的计数器

二十二、运行告警系统

为了让系统每分钟都执行main.sh来监控，可使用任务计划

[root@lanquark sbin]# crontab -e
no crontab for root - using an empty one
* * * * * cd /usr/local/sbin/mon/bin;bash main.sh