shell高级用法--函数

bjghzly

　　在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 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用户重新执行看看结果
[root@dnsserver ~]# userdel -r hadoop　　
[root@dnsserver ~]# ./adduser.sh
　　
add user finished.
　　
[root@dnsserver ~]# ./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
　　执行脚本
[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."
　　
    fi
　　
done