shell高级用法--函数

Posted 2020-07-10

在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 $?

[[email protected] ~]# ./adduser.sh       
0
[[email protected] ~]# ./adduser.sh 
0

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

[[email protected] ~]# bash -x adduser.sh 
+ ADDUSER
+ USERNAME=hadoop
+ id -u hadoop
+ echo 0
0

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

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

上面的脚本改成这样：

#!/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用户重新执行看看结果

[[email protected] ~]# userdel -r hadoop
[[email protected] ~]# ./adduser.sh     
add user finished.
[[email protected] ~]# ./adduser.sh 
Failed

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

接受参数的函数：

#!/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、在主程序中调用此函数判定指定范围内的所有主机的在线情况

#!/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

执行脚本

[[email protected] ~]# ./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."
    fi
done

本文出自 “zengestudy” 博客，谢绝转载！