在编写脚本时,有些脚本可以反复使用,可以调用函数来解决。
语句块定义成函数约等于别名。
函数的作用:
- 使用函数可以避免代码重复;
- 使用函数可以将一个大的工程分割为若干小的功能模块,代码的可读性更强。
函数的使用方法:
- 先定义函数
- 再引用函数
1、定义、查看、删除函数
1.1 定义函数的方法
shell中定义函数有两种格式:
#格式一:
function 函数名 { //三部分之间都要有空格
命令序列
}
#格式二:
函数名 () { //函数名和小括号之间的空格可有可无
命令序列
}
###### main #####
#可以在主代码区域直接使用函数名调用函数
函数名
复制代码
1.2 删除函数unset
格式: unset 函数名
[root@yuji sh]# unset hello //删除函数
[root@yuji sh]# hello //删除后无法再调用
bash: hello: 未找到命令...
复制代码
1.3 查看函数declare
格式:
declare -F //查看函数列表
declare -f //查看函数具体的定义
复制代码
查看函数列表:
[root@yuji sh]# declare -F
declare -f __HOSTNAME
declare -f __SIZE
declare -f __SLAVEURL
declare -f __VOLNAME
......
复制代码
2、函数返回值
获得函数返回值的两种方式:
- return表示退出函数并返回一个退出值,脚本中可以用$?变量显示该值。
- 在函数体中用 echo 输出返回值。并在函数体外使用变量赋值后,可再进一步对函数的返回值进行加工操作。
2.1 使用return
return使用原则:
- 函数一结束就取返回值,因为$?变量只返回执行的最后一条命令的退出状态码。
- 退出状态码必须是0~255,超出时值将为除以256取余。
示例:
#!/bin/bash
#定义函数
fun1 () {
read -p "请输入一个数字:" num
return $[$num*2]
}
##### main ######
#调用函数
fun1
#输出返回值
echo $?
复制代码
2.2 使用echo
因为return的返回值的范围是0-255,超过部分除以256取余,得不到我们想要的结果,此时可以直接在函数体中使用echo命令。
示例:
#!/bin/bash
#定义函数
fun2 () {
read -p "请输入一个数字:" num
echo $[$num*2]
}
##### main ######
#在函数体外使用变量赋值,方便进一步对函数的返回值进行加工操作
result=`fun2`
#输出返回值
echo $result
复制代码
3、函数传参
示例1:
#!/bin/bash
sum1 () {
sum=$[$1+$2]
echo $sum
}
######## main #######
read -p "请输入第一个位置参数:" first
read -p "请输入第一个位置参数:" second
sum1 $first $second
复制代码
示例2:
#!/bin/bash
sum2 () {
sum=$[$1+$2]
echo $sum
}
######## main #######
sum2 $1 $2
复制代码
示例3:
#!/bin/bash
sum3 () {
#函数中的$1,$2 代表调用函数时,函数后面跟的位置参数
echo $1
echo $2
}
####### main #######
#主体代码中的$1,$2 代表执行脚本时,脚本后面跟的位置参数
sum3 $2 $1
复制代码
4、函数的作用范围
函数变量的作用范围:
-
函数在Shell脚本中仅在当前Shell环境中有效。(即定义和调用函数要在同一个shell环境中)
-
脚本中定义的变量,仅在当前shell环境中有效。
- 如果使用source或 . 执行脚本,那么该变量在命令行也会生效,因为这两种方式没有打开子shell环境,而是在当前shell环境中执行脚本。一般建议不要使用source和. 来执行脚本,可能会影响系统资源配置。
-
Shell脚本中变量默认全局有效。(即在整个脚本中都生效,并非是环境变量)
-
将变量限定在函数内部使用 local 命令。这样变量只在当前函数内有效,不影响脚本中的其他函数和变量。
示例1:
#!/bin/bash
i=8
i=9
echo $i
复制代码
示例2:
#!/bin/bash
myfun1 () {
#没有被local定义的变量是全局有效的
i=9
echo $i
}
########## main ###########
i=8
myfun1
echo $i
复制代码
示例3:
#!/bin/bash
myfun2 () {
#local定义的变量只能在local命令之后到函数结束之前有效
local i=9
echo $i
}
########## main ###########
i=8
myfun2
echo $i
复制代码
示例4:
#!/bin/bash
myfun3 () {
echo $i
#local定义的变量只能在local命令之后到函数结束之前有效
local i=9
echo $i
}
########## main ###########
i=8
myfun3
echo $i
复制代码
示例5:
#!/bin/bash
myfun4 () {
#没有被local定义的变量是全局有效的
i=7
echo $i
#local定义的变量只能在local命令之后到函数结束之前有效
local i=9
echo $i
}
########## main ###########
i=8
myfun4
echo $i
复制代码
5、函数的递归
递归就是一个函数在它的函数体内调用它自身。执行递归函数将反复调用其自身,每调用一次就进入新的一层。所以递归要有两个基本要素,结束条件与递推关系。
递归的两个基本要素:
(1)边界条件:确定递归到何时终止,也称为递归出口。
(2)递归模式:大问题是如何分解为小问题的,也称为递归体。递归函数只有具备了这两个要素,才能在有限次计算后得出结果 。
示例1:使用递归计算阶乘
创建脚本:
#!/bin/bash
fact() {
#$1表示调用函数时,后面跟的位置参数。如果值是1,就输出1。
if [ $1 -eq 1 ]
then
echo 1
else
#定义一个本地变量temp,变量值为每次传入的参数值减1
local temp=$[$1 - 1]
#不停地调用函数自身,直到temp的值等于1
local result=$(fact $temp)
echo $[$1 * $result]
fi
}
########## main ###########
read -p "请输入一个正整数:" num
a=$(fact $num)
echo "$num 的阶乘为:$a"
复制代码
实现过程:
以5的阶乘为例
#fact 5
$1=5 temp=4 result=$(fact 4) echo 5 * $(fact 4)
#fact 4
$1=4 temp=3 result=$(fact 3) echo 5 * 4*$(fact 3)
#fact 3
$1=3 temp=2 result=$(fact 2) echo 5 * 4 * 3*$(fact 2)
#fact 2
$1=2 temp=1 result=$(fact 1) echo 5 * 4 * 3 * 2*$(fact 1)
#fact 1
$1=1 echo 5 * 4 * 3 * 2 * 1
复制代码
执行脚本:
[root@yuji sh]# bash fact.sh
请输入一个正整数:5
5 的阶乘为:120
[root@yuji sh]# bash fact.sh
请输入一个正整数:7
7 的阶乘为:5040
[root@yuji sh]# bash fact.sh
请输入一个正整数:10
10 的阶乘为:3628800
复制代码
示例2:递归查找目录下的子目录和文件
先执行 mkdir -p /root/bin/aa/bb/cc/dd ; touch /root/bin/aa/bb/cc/dd/abc.txt。 之后输出环境变量PATH所包含的所有目录以及其中的子目录和所有不可执行文件。
#!/bin/bash
list () {
#逐个检查$1参数指定目录下的所有文件或子目录
for fd in $1
do
#判断如果是目录就输出,并且把这个目录下的所有文件或子目录再逐个检查一下,如果还有子目录,会一直递归下去。
if [ -d "$fd" ]
then
echo "$fd 是目录"
list "$fd/*"
elif [ -f $fd ]&&[ ! -x $fd ]
then
echo "$fd 文件没有执行权限"
fi
done
}
######main#####
IFS_OLD=$IFS
IFS=$IFS':'
for i in $PATH
do
list "$i"
done
IFS=$IFS_OLD
复制代码
执行脚本:
[root@yuji sh]# bash list.sh
/usr/local/sbin 是目录
/usr/local/bin 是目录
/usr/sbin 是目录
/usr/bin 是目录
/root/bin 是目录
/root/bin/aa 是目录
/root/bin/aa/bb 是目录
/root/bin/aa/bb/cc 是目录
/root/bin/aa/bb/cc/dd 是目录
/root/bin/aa/bb/cc/dd/abc.txt 文件没有执行权限
复制代码
6、函数库
我们可以事先创建一个函数库文件,在里面定义各种常用的函数,然后可以在别的shell脚本中直接引用这个函数库文件,使得不需要再次定义函数即可直接调用函数。
注意:
- "source"和 "." 是在当前shell环境中运行脚本。
- 如果函数库文件中定义了变量的话,切换bash环境就不生效了。所以在脚本中引用函数库文件时,一定要使用"source"或 "." 。
- 引用函数库文件时,建议使用绝对路径。避免找不到该文件。
创建一个函数库文件:
[root@yuji sh]# vim funbase.sh
#!/bin/bash
#加减乘除函数库
jiafa () {
echo $[$1+$2]
}
jianfa () {
echo $[$1-$2]
}
chengfa () {
echo $[$1*$2]
}
chufa () {
if [ $2 -eq 0 ]
then
echo "除数不能为0"
else
echo $[$1/$2]
fi
}
复制代码
在其他shell脚本中引用该函数库文件:
[root@yuji sh]# vim t1.sh
#!/bin/bash
#调用函数库文件,注意要使用source或.来调用
. funbase.sh
read -p "请输入第一个正整数:" a
read -p "请输入第二个正整数:" b
#调用函数
r1=$(jiafa $a $b)
r2=$(jianfa $a $b)
r3=$(chengfa $a $b)
r4=$(chufa $a $b)
#输出结果
echo "$a加$b的结果是:$r1"
echo "$a减$b的结果是:$r2"
echo "$a乘$b的结果是:$r3"
echo "$a除以$b的结果是:$r4"
[root@yuji sh]# bash t1.sh //执行脚本
请输入第一个正整数:6
请输入第二个正整数:3
6加3的结果是:9
6减3的结果是:3
6乘3的结果是:18
6除以3的结果是:2
复制代码
7、注意事项总结
- 脚本中调用函数,直接写函数名。
- 调用函数之前一定要先定义。
- 调用的函数时,如果有同名函数,后一个生效。
- 尽量避免定义同名函数。
- 函数不被调用时,是不生效的。
- 在脚本中引用函数库文件时,一定要使用"."或"source",同时要使用函数库文件的绝对路径。