Shell 编程基础 - 函数

Shell 编程基础 - 函数

函数

函数是 Shell 脚本中自定义的一系列执行命令，一般来说函数应该设置有返回值（正确返回0，错误返回非0。对于错误返回，可以定义返回其他非0正值来细化错误）。使用函数最大的好处是可避免出现大量重复代码，同时增强了脚本的可读性。

函数的定义和调用

#function 关键字定义函数
function FUNCTINO_NAME() {
	command1
	command2
	......
}
#也可以省略关键字 function
FUNCTION_NAME() {
	command1
	command2
	......
}

[root@localhost shell]# cat func.sh
#!/bin/bash
#定义函数 sayHello
function sayHello() {
        echo -n "Hello "
}
#定义函数 sayWorld
sayWorld() {
        echo "World"
}
#调用函数 sayHello
sayHello
#调用函数 sayWorld
sayWorld
[root@localhost shell]# bash func.sh
Hello World

函数的返回值

函数的返回值又叫函数的退出状态，实际上是一种通信方式。

[root@localhost shell]# cat return.sh
#!/bin/bash
echo -n "输入一个文件："
read FILE
function checkFileExist() {
        if [ -f $FILE ]; then
                return 0
        else
                return 1
        fi
}
checkFileExist
if [ $? -eq 0 ]; then
        echo "文件 $FILE 存在"
else
        echo "文件 $FILE 不存在"
fi
[root@localhost shell]# bash return.sh
输入一个文件：/etc/passwd
文件 /etc/passwd 存在
[root@localhost shell]# bash return.sh
输入一个文件：/etc/nofile
文件 /etc/nofile 不存在

带参数的函数

[root@localhost shell]# cat func.sh
#!/bin/bash
function func() {
        echo "$0 $1 $2 $3"
        echo "$# $@"
}
func Hello World 12345
[root@localhost shell]# bash func.sh
func.sh Hello World 12345
3 Hello World 12345

函数库

对某些很常用的功能，必须考虑将其独立出来，集中存放在一些独立的文件中，这些文件就称为“函数库”。这么做的好处是在后期开发的过程中可以直接利用这些库函数写出高质量的代码。库函数的本质也是“函数”，所以它的定义方式和普通函数没有任何区别，但为了和一般函数区分开来，在实践中建议库函数使用下划线开头。

自定义函数库

由于 Shell 是一门面向过程的脚本型语言，而且用户主要是 Linux 系统管理人员，所以并没有非常活跃的社区，这也造成了 Shell 缺乏第三方函数库，所以在很多时候需要系统管理人员根据实际工作的需要自行开发函数库。

[root@localhost shell]# cat lib01.sh
#!/bin/bash
_checkFileExist() {
        if [ -f $1 ]; then
                echo "File $1 Exist"
        else
                echo "File $1 Not Exist"
        fi
}
[root@localhost shell]# cat test.sh
#!/bin/bash
source ./lib01.sh
_checkFileExist /etc/passwd
_checkFileExist /etc/nofile
[root@localhost shell]# bash test.sh
File /etc/passwd Exist
File /etc/nofile Not Exist

函数库 /etc/init.d/functions

很多 Linux 发行版中都有 /etc/init.d 目录，这是系统中放置所有开机启动脚本的目录，这些开机脚本在脚本开始运行时都会加载 /etc/init.d/functions 或 /etc/rc.d/init.d/functions 函数库（实际上这两个函数库的内容是完全一样的）。

递归函数

用阶乘举例

[root@localhost shell]# cat j.sh
#!/bin/bash
function j() {
        local num=$1
        if [ $num -le 0 ]; then
                return 1
        else
                j $((num-1))
                return $[$num*$?]
        fi
}
j 3
echo $?
[root@localhost shell]# bash j.sh
6