shell基础(三)函数

一、函数

function 函数名(){
    指令。。
    return n
}

以上可以简写成:
函数名(){
    指令..
    return n
}

二、函数的执行

一:不带参数的执行
1.function和后面的小括号都不带吗,仅仅时函数名就可以执行
2.函数的定义必须在执行之前定义或加载(即先定义后执行)
3.函数执行时,会和脚本公用变量,也可以为函数设定局部变量及特殊位置参数
4.函数中的return和脚本中的exit功能类似,return用来退出函数,exit是用来退出脚本
5.return的返回值会给调用函数的程序,而exit的返回值给执行程序的shell
6.如果讲函数独立与脚本之外,被脚本加载使用时,需要使用source或者"."来提前加载使用
  例如:
        . /etc/init.d/functions  #加载系统函数中的命令或者参数变量,以供后面的程序使用
7.local定义函数内部的局部变量,变量在离开函数后会消失

二:带参数的执行:
  函数名 参数1 参数2
1.shell的位置参数$1、$2、..$# 、$*、$?及$@都可以作为函数的参数使用,此时,父脚本的参数临时被隐藏,而$0仍然时父脚本的名称
2.当函数执行完成时,原来的命令行脚本的参数恢复
3.函数的参数变量是在函数体里面定义的

小插曲

1.使用cat命令追加多行,如打印选项菜单
cat <

实战题目一:

用shell脚本检查某网站是否存在异常

方式一(普通):
#!/bin/bash
 if [ $# -ne 1 ];then
        usage $"usage $0 url"
 fi
wget --spider -q -o /dev/null --tries=1 -T 5 $1 
#--spider用于测试,不下载,-q不在命令中显示 -o 输入到后面的文件中 ,--tries=number为尝试次数和-t一样  -T超时时间和--timeout一样 -S显示响应头
if [ $? -eq 0 ];then
        echo "ok"
else
        echo "error"
fi
方式二(函数封装):
#!/bin/bash
function Usage(){

        echo $"Usage:$0 url"
        exit 1
}
function check_url(){
        wget --spider -q -o /dev/null -t 1 -T 5 $1 
        if [ $? -eq 0 ];then
                echo "ok"
        else
                echo "error"
        fi 
}
function main(){
        if [ $# -ne 1 ];then
                Usage
        else
                check_url $1
        fi
}
main $*  #将脚本传入的参数全部都传到主函数中

实战题目二:参数传入脚本、检查某网站是否存在异常,以更专业的方式输出

#!/bin/bash
. /etc/init.d/functions  #调用(加载)系统函数,因为下面要用action函数
function Usage(){
        echo $"usage:$0 url"
        exit 1
}
function check_url(){
        wget --spider -q -o /dev/null -t 1 -T 5 $1
        if [ $? -eq 0 ];then
                action "test $1" /bin/true
        else
                action "test $1" /bin/false
        fi
}
function main (){
        if [ $# -ne 1 ];then
                Usage
        else
                check_url $1
        fi
}
main $*

效果如下:
[root@mycentos shell]# sh 3.sh www.baidu.com   
test www.baidu.com                                         [  OK  ]
[root@mycentos shell]# sh 3.sh www.baidu.c
test www.baidu.c                                           [FAILED]

实战题目三:用shell开发模块化rsync服务启动脚本

#!/bin/bash
#chkconfig:2345 21 81
#description
#上面2行是将rsync加入开机自启动服务

#调用系统函数
. /etc/init.d/functions


#输入错误提示
function Usage(){
        echo $"usage $0 {start|stop|restart}"
        exit 1
}


#启动服务
function Start(){
        rsync --daemon  #启动服务
        sleep 2    #启动服务2秒后再做判断

        if [ $(netstat -pantu | grep rsync | wc -l) -ne 0 ];then
                action "rsyncd is started" /bin/true
        else
                action "rsyncd is started" /bin/false
        fi
}
#停止服务
function Stop(){

        killall rsync&>/dev/null

        sleep 2

        if [ $(netstat -apntu| grep rsync | wc -l) -eq 0 ];then
                action "rsyncd is stopped" /bin/true
        else
                action "rsyncd is stopped" /bin/false
        fi
}

case "$1" in
 "start")
        Start
        ;;
 "stop")
        Stop
        ;;
 "restart")
        Stop
        sleep 1
        Start
        ;;
  *)
        Usage
esac

结果如下:
[root@mycentos init.d]# /etc/init.d/rsyncd start
rsyncd is started                                          [  OK  ]
[root@mycentos init.d]# lsof -i:873
COMMAND  PID USER   FD   TYPE DEVICE SIZE/OFF NODE NAME
rsync   1478 root    4u  IPv4  13067      0t0  TCP *:rsync (LISTEN)
rsync   1478 root    5u  IPv6  13068      0t0  TCP *:rsync (LISTEN)
[root@mycentos init.d]# /etc/init.d/rsyncd stop
rsyncd is stopped                                          [  OK  ]
[root@mycentos init.d]# lsof -i:873
[root@mycentos init.d]# /etc/init.d/rsyncd restart
rsyncd is stopped                                          [  OK  ]
rsyncd is started                                          [  OK  ]
[root@mycentos init.d]# lsof -i:873               
COMMAND  PID USER   FD   TYPE DEVICE SIZE/OFF NODE NAME
rsync   2379 root    4u  IPv4  19734      0t0  TCP *:rsync (LISTEN)
rsync   2379 root    5u  IPv6  19735      0t0  TCP *:rsync (LISTEN)

注:
1.在安装或者启动时,如果遇到找不到“/etc/rsync.conf”文件时,要用touch建立一个,实际工作中要写入东西,此处为了方便就不写,为了启动rsync服务即可
2.rsync具体的脱离xinetd启动的方式见"shell基础二"
3.放在/etc/init.d目录下就可以用service命令启动,启动前需要是脚本有执行权限,否则无法执行,也无法自动补全。
4.要加入开机自启动,则需要加如脚本中解释器下面2行,然后“chkconfig --add rsyncd”加入自启动 "rsyncd"是/etc/init.d目录下的服务名称

实战四:执行shell脚本,打印如下菜单,柑橘选择,给选择的水果加一种颜色。

1.红色的苹果
2.绿色的苹果
3.黄色的苹果
4.蓝色的苹果

方法一(普通版)
#!/bin/sh

#定义好颜色,方便后面使用
RED_COLOR='\E[1;31m'
GREEN_COLOR='\E[1;32m'
YELLOW_COLOR='\E[1;33m'
BLUE_COLOR='\E[1;34m'

RES='\E[0m'

cat <

效果如图:


shell基础(三)函数_第1张图片
image.png

实战五:紧接着上题,请开发一个给制定内容加上制定颜色的脚本

#!/bin/bash

RED_COLOR='\E[1;31m'
GREEN_COLOR='\E[1;32m'
YELLOW_COLOR='\E[1;33m'
BLUE_COLOR='\E[1;34m'

RES='\E[0m'

function Usage(){

        echo $"usage:$0 txt {red|green|yellow|pink}"
        exit 1
}

if [ $# -ne 2 ];then
           Usage
           exit
fi 
function Choose(){

        case "$2" in
                "red")
                        echo -e ${RED_COLOR}$1${RES}
                        ;;
                "green")
                        echo -e ${GREEN_COLOR}$1${RES}

                        ;;
                "yellow")
                        echo -e ${YELLOW_COLOR}$1${RES}
                        ;;
                "blue")
                        echo -e ${BLUE_COLOR}$1${RES}
                        ;;
                *)
                Usage
        esac
}
function Main(){
        Choose $1 $2
}
Main $*
注:
1.将参数二的颜色付给参数一
2.精确匹配单词的三种方式
  1.grep -w 'oldboy' file 
  2.grep "\boldboy\b" file
  3.grep "^oldboy$" file
以上都是将出现oldboy单词的行显示出来,而不是将包含oldboy的行显示出来

实战五:

启动Nginx服务的命令:/application/nginx/sbin/nginx
关闭Nginx服务的命令:/application/nginx/sbin/nginx -s stop

请开发脚本,以实现Nginx服务启动和关闭功能,具体脚本命令为/etc/init.d/nginxd {start|stop|restart},并通过chkconfig进行开机自启动

思路:
1.判断开启或关闭服务(一是检测pid文件是否存在,存在就是开启,不存在就是服务已经关闭),或者使用netstat链接数也可以
2.start和stop分别构成函数
3.对函数和命令运行的返回值进行处理
4.chkconfig实现服务自启动
代码:
#!/bin/sh
#chkconfig:2345 27 83
#description

source /etc/init.d/functions  #加载系统函数库

#定义文件路径
PATH="/application/nginx/sbin"
PID_PATH="/application/nginx/logs/nginx.pid"
REVEAL=0
function Usage(){
        echo $"usage:$0 {start|stop|restart}"
        exit 1
}
#判断参数的个数
if [ $# -ne 1 ];then
        Usage
fi
#开始函数
function Start(){
        if [ ! -f $PID_PATH ];then  #若原来服务是关闭的
                $PATH/nginx  #开启服务
                REVEAL=$?  

                if [ $REVEAL -eq 0 ];then  #判断是否开启
                        action "nginx is started" /bin/true

                else
                        action "nginx is started" /bin/false
                fi

        else
                echo "nginx is running"

        fi
        return $REVEAL
}
function Stop(){#结束函数

        if [ -f $PID_PATH ];then  #若原来服务是启动的
                $PATH/nginx -s stop  #关闭服务
                REVEAL=$?

                if [ $REVEAL -eq 0 ];then  #判断是否关闭
                        action "nginx is stopped" /bin/true
                else
                        action "nginx is stopped" /bin/false
                fi
                return $REVEAL
        else
                echo "nginx is no running"
        fi
        return $REVEAL
}

        case "$1" in
           "start")
                 Start
                REVEAL=$?
                ;;
           "stop")
                Stop
                REVEAL=$?
                ;;
           "restart")
                Stop
                Start
                REVEAL=$?
                ;;
                *)
                Usage
        esac

 exit $REVEAL

效果如下:

[root@mycentos init.d]# /etc/init.d/nginxd start
nginx is running
[root@mycentos init.d]# /etc/init.d/nginxd stop
nginx is stopped                                           [  OK  ]
[root@mycentos init.d]# /etc/init.d/nginxd start
nginx is started                                           [  OK  ]
[root@mycentos init.d]# lsof -i:80
COMMAND  PID  USER   FD   TYPE DEVICE SIZE/OFF NODE NAME
nginx   2928  root    6u  IPv4  23795      0t0  TCP *:http (LISTEN)
nginx   2929 nginx    6u  IPv4  23795      0t0  TCP *:http (LISTEN)
[root@mycentos init.d]# /etc/init.d/nginxd restart
nginx is stopped                                           [  OK  ]
nginx is started                                           [  OK  ]
[root@mycentos init.d]# lsof -i:80                
COMMAND  PID  USER   FD   TYPE DEVICE SIZE/OFF NODE NAME
nginx   2940  root    6u  IPv4  24824      0t0  TCP *:http (LISTEN)
nginx   2941 nginx    6u  IPv4  24824      0t0  TCP *:http (LISTEN)

彩蛋一枚

return :
    1.用来退出函数,后面的函数里面的内容不再执行
exit:
     2.用来退出脚本,后面的内容不再执行

function test(){
  return 1
}  
test    #函数执行完成后,$?会得到test中return后面的返回值
ls -l ./  #这条命令执行成功,$?变成0
function test2(){
  exit 99
} 
test2 #函数执行完成后,$?变成99,也就是exit后面的数值
$?的值会不断的改变,有别于其他语言的函数调用的返回值。

结果:

#!/bin/bash
function test(){
        return 8
}
echo $?  #0
test
echo $? #8
function test2(){
        return 9
}
echo $? #0
test2
echo $? #9
ls -l ./
echo $? #0
exit 10

此shell执行结束后,echo $? 结果是10

你可能感兴趣的:(shell基础(三)函数)