在shell中如果一段代码需要反复被调用，则可以将该段代码定义称为函数，直接在后面的脚本中调用即可。定义函数的方法有两种：
1、function functionname {
}
2、functionname() {
}
例如：定义一个添加用户hadoop的函数
|
1
2
3
4
5
6
7
8
9
|
ADDUSER() {
USERNAME=hadoop
if
!
id
-u $USERNAME &>
/dev/null
;
then
useradd
$USERNAME
echo
$USERNAME |
passwd
--stdin $USERNAME &>
/dev/null
fi
}
ADDUSER
echo
$?
|
|
1
2
3
4
|
[root@dnsserver ~]
# ./adduser.sh
0
[root@dnsserver ~]
# ./adduser.sh
0
|
连续两次执行脚本，返回的状态值都是0，这里有一个问题，第一次执行这个脚本，成功添加了用户hadoop，但是第二次执行时，hadoop已经存在，那么这个为什么还是返回状态值为0？
|
1
2
3
4
5
6
|
[root@dnsserver ~]
# bash -x adduser.sh
+ ADDUSER
+ USERNAME=hadoop
+
id
-u hadoop
+
echo
0
0
|
从上面的分析可以看出，由于脚本最后一条命名为id -u hadoop,而该命令是能成功执行的，所以返回值是0。显然，这是不合理的。那么我们如果根据不同的情况，获取不同的状态值呢？
可以利用return命令，在函数中用return来定义函数的状态值。
上面的脚本改成这样：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
#!/bin/bash
#
ADDUSER() {
USERNAME=hadoop
if
!
id
-u $USERNAME &>
/dev/null
;
then
useradd
$USERNAME
echo
$USERNAME |
passwd
--stdin $USERNAME &>
/dev/null
return
0
else
return
1
fi
}
ADDUSER
if
[ $? -
eq
0 ];
then
echo
"add user finished."
else
echo
"Failed"
fi
|
删除hadoop用户重新执行看看结果
|
1
2
3
4
5
|
[root@dnsserver ~]
# userdel -r hadoop
[root@dnsserver ~]
# ./adduser.sh
add user finished.
[root@dnsserver ~]
# ./adduser.sh
Failed
|
第二次添加用户hadoop的时候就报错了。
接受参数的函数：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
#!/bin/bash
#
ADDUSER() {
USERNAME=$1
if
!
id
-u $USERNAME &>
/dev/null
;
then
useradd
$USERNAME
echo
$USERNAME |
passwd
--stdin $USERNAME &>
/dev/null
return
0
else
return
1
fi
}
##调用函数时也要跟$1这个变量
ADDUSER $1
if
[ $? -
eq
0 ];
then
echo
"add user finished."
else
echo
"Failed"
fi
|
练习：写一个脚本，判定192.168.0.200~192.168.0.254之间的主机哪些在线，要求：
1、使用函数来实现一台主机的判定过程：
2、在主程序中调用此函数判定指定范围内的所有主机的在线情况
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
#!/bin/bash
#
PING() {
if
ping
-c 1 -w 1 192.168.0.$I &>
/dev/null
;
then
echo
"192.168.0.$I is online."
else
echo
"192.168.0.$I is offline."
fi
}
for
I
in
{200..254};
do
PING 192.168.0.$I
done
|
执行脚本
|
1
2
3
4
5
|
[root@dnsserver ~]
# ./ping.sh
192.168.0.200 is offline.
192.168.0.201 is offline.
192.168.0.202 is offline.
192.168.0.203 is offline.
|
这里脚本输出结果返回的信息是由函数PING来显示的，如果不希望函数来显示是否在线，而是由主程序来显示，则可以利用return来改进脚本，改进后的脚本如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
#!/bin/bash
#
PING() {
if
ping
-c 1 -w 1 192.168.0.$I &>
/dev/null
;
then
return
0
else
return
1
fi
}
for
I
in
{200..254};
do
PING 192.168.0.$I
if
[ $? -
eq
0 ];
then
echo
"192.168.0.$I is online."
else
echo
"192.168.0.$I is offline."
fi
done
|
本文转自 曾哥最爱 51CTO博客，原文链接：http://blog.51cto.com/zengestudy/1775892，如需转载请自行联系原作者
