shell高级用法--函数

时间:2022-10-10 14:38:04

在shell中如果一段代码需要反复被调用,则可以将该段代码定义称为函数,直接在后面的脚本中调用即可。定义函数的方法有两种:

1、function  functionname {



}


2、functionname() {


}


例如:定义一个添加用户hadoop的函数

ADDUSER() {USERNAME=hadoop
if ! id -u $USERNAME &> /dev/null; then
    useradd $USERNAME
    echo $USERNAME | passwd --stdin $USERNAME &> /dev/null
fi
}
ADDUSER
echo $?
[root@dnsserver ~]# ./adduser.sh       0[root@dnsserver ~]# ./adduser.sh 0

连续两次执行脚本,返回的状态值都是0,这里有一个问题,第一次执行这个脚本,成功添加了用户hadoop,但是第二次执行时,hadoop已经存在,那么这个为什么还是返回状态值为0?

[root@dnsserver ~]# bash -x adduser.sh + ADDUSER+ USERNAME=hadoop+ id -u hadoop+ echo 00

从上面的分析可以看出,由于脚本最后一条命名为id -u hadoop,而该命令是能成功执行的,所以返回值是0。显然,这是不合理的。那么我们如果根据不同的情况,获取不同的状态值呢?

可以利用return命令,在函数中用return来定义函数的状态值。

上面的脚本改成这样:

#!/bin/bash#ADDUSER() {USERNAME=hadoopif ! id -u $USERNAME &> /dev/null; then    useradd $USERNAME    echo $USERNAME | passwd --stdin $USERNAME &> /dev/null    return 0else    return 1fi}ADDUSERif [ $? -eq 0 ]; then   echo "add user finished."else   echo "Failed"fi


删除hadoop用户重新执行看看结果

[root@dnsserver ~]# userdel -r hadoop[root@dnsserver ~]# ./adduser.sh     add user finished.[root@dnsserver ~]# ./adduser.sh Failed

第二次添加用户hadoop的时候就报错了。




接受参数的函数:

#!/bin/bash#ADDUSER() {USERNAME=$1if ! id -u $USERNAME &> /dev/null; then    useradd $USERNAME    echo $USERNAME | passwd --stdin $USERNAME &> /dev/null    return 0else    return 1fi}##调用函数时也要跟$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、在主程序中调用此函数判定指定范围内的所有主机的在线情况

#!/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.$Idone

执行脚本

[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来改进脚本,改进后的脚本如下:

#!/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."    fidone














本文出自 “zengestudy” 博客,谢绝转载!