续Shell脚本学习 Day6

shell函数

基本概念

区分return与exit

return只能在函数体中 使用

函数实践

函数定义和执行在同一个文件中

编写脚本func1.sh

1
2
3
4
5
6
7
8
9
10
#!/bin/bash
#函数定义
function write_music(){
	cd /tmp/
	echo "创建一个文件，并写入信息"
	echo "我是被写入的信息" > ./music.txt
	return 0
}
#函数执行
write_music

执行演示

1
2
3
4
[root@localhost ~]# source func1.sh
创建一个文件，并写入信息
[root@localhost tmp]# cat music.txt
我是被写入的信息

定义函数加载到环境变量中

编写脚本func2.sh

1
2
3
4
#!/bin/bash
func2(){
	echo "我是函数体，我被执行了！"
}

检查环境变量的方法

1
set

利用source或.命令执行函数，可以将函数加载到环境变量中

1
2
set | grep ^func2
#^表示以func2开头的文本

执行演示

1
2
3
4
5
6
7
8
9
[root@localhost ~]# cat func2.sh
#!/bin/bash
func2(){
        echo "我是函数体，我被执行了！"
}
[root@localhost ~]# set | grep ^func2
[root@localhost ~]# . func2.sh
[root@localhost ~]# set | grep ^func2
func2 ()

在环境变量中的函数可以直接被调用

1
2
[root@localhost ~]# func2
我是函数体，我被执行了！

当我们退出当前shell后，环境变量就消失了

定义函数和执行函数在不同文件中

编写func3.sh脚本

1
2
3
4
5
#!/bin/bash
#将fun2函数加载到环境变量
[ -f /func2.sh ] && . func2.sh || exit
#执行被加载到环境变量的函数func2
func2

引用系统自定义函数美化脚本

/etc/init.d/mysql

log_success_msg

log_failure_msg

1
#没找到这个文件，暂时不写

开发rsync启停管理脚本

函数版本

编写rsync_func文件

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#!/bin/bash

usage(){
        echo "Usage:$0 {start|stop|restart}"
        exit 1
}

start(){
        /usr/bin/rsync --daemon  #在后台运行
        sleep 1
        if [ `ss -tunlp|grep rsync|wc -l` -ge "1" ] #检查进程或者端口确保运行
        then
                #log_success_msg "Rsync is started!"
                echo "Rsync is started successfully!"
        else
                #log_failure_msg "Rsync failed to start!"
                echo "Rsync failed to start!"
        fi
}

stop(){
        killall rsync &>/dev/null
        sleep 1
        if [ `ss -tunlp|grep rsync|wc -l` -eq "0" ]
        then
                #log_success_msg "Rsync is stopped!"
                echo "Rsync is stopped successfully!"
        else
                #log_failure_msg "Rsync failed to stop!"
                echo "Rsync failed to stop!"
        fi
}

restart(){
        killall rsync &>/dev/null
        var1=`ss -tunlp|grep rsync|wc -l`
        /usr/bin/rsync --ddeamon
        var2=`ss -tunlp|grep rsync|wc -l`
        if [ "$var1" -lt "$var2" ]
        then
                 #log_success_msg "Rsync is started!"
                echo "Rsync is restarted successfully!"
        else
                #log_failure_msg "Rsync failed to restart!"
                echo "Rsync failed to restart!"
        fi
}

main(){
        if [ "$#" -ne "1" ]
        then
                usage
        fi

        if [ "$1" = "start" ]
        then
                start
        elif [ "$1" = "stop" ]
        then
                stop
        elif [ "$1" = "restart" ]
        then
                restart
        else
                usage
        fi

}
#调用函数接口
main $*

执行演示

1
2
3
4
5
6
7
8
9
10
[root@localhost ~]# /root/rsync_func start
Rsync is started successfully!
[root@localhost ~]# /root/rsync_func stop
Rsync is stopped successfully!
[root@localhost ~]# /root/rsync_func restart
Rsync is restarted successfully!
[root@localhost ~]# /root/rsync_func sta
Usage:/root/rsync_func {start|stop|restart}
[root@localhost ~]# /root/rsync_func
Usage:/root/rsync_func {start|stop|restart}

函数脚本传递参数

编写func4.sh脚本

1