使用if语句编写Shell脚本

Posted 2020-10-18

    在简单的shell脚本程序中，各条语句将按先后顺序依次执行，从而实现批处理的自动化过程。然而，单一的顺序结构使得脚本不够智能，难以处理更加灵活的系统任务。通过正确的使用if语句，使得shell脚本具有一定的判断能力，以根据不同的条件来完成不同的任务管理。

条件测试操作

    要使shell脚本具有一定的智能，就必须使其区分在不同的条件下执行哪种操作。例如，当磁盘使用率超过95%时发送告警消息等

    可使用专门的测试工具test命令，对特定的条件进行测试，并根据返回值（$?）来判断条件是否成立,返回值为0表示条件成立

使用test命令的格式如下：

格式一：test 条件表达式

格式二：[ 条件表达式 ]

    通常情况下，使用格式二更符合编程习惯，但是方括号与条件表达式至少需要一个空格进行分隔

1. 文件测试

    文件测试指的是根据指定的路径名称，判断对应的是文件还是目录，是否可读、可写、可执行。常见的操作选项如下：

-d：测试是否为目录（Directory）

-f：测试是否为文件（File）

-e：测试目录或文件是否存在（Exist）

-r：测试当前用户是否有权限读取（Read）

-w：测试当前用户是否有权限写入（Write）

-x：测试当前用户是否有权限执行（Excute）

2. 整数值比较

    整数值比较指的是根据指定的两个整数值，判断第一个数和第二个数的关系，常用操作选项如下：

-eq：第一个数等于第二个数

-ne：第一个数不等于第二个数

-gt：第一个数大于第二个数

-lt：第一个数小于第二个数

-ge：第一个数大于或等于第二个数

-le：第一个数小于或等于第二个数

3. 字符串比较

    字符串比较通常检测用户输入、系统环境是否满足条件，常用操作选项如下：

=：第一个字符串与第二个字符串相同

!=：第一个字符串与第二个字符串不相同

-z：检测字符串是否为空，对于未定义或赋予空值的变量将视为空串

4. 逻辑测试

    逻辑测试指的是判断两个或多个条件之间的依赖关系，常用的逻辑测试操作如下：

&&：逻辑与，只有当前后两个条件都成立时，整个测试命令的返回值才为0。使用test命令时，&&可改为-a

||：逻辑或，只要前后条件中有一个成立，整个测试命令的返回值即为0。使用test命令时，||可改为-o

！：逻辑否，只有当指定的条件不成立时，整个测试命令的返回值才为0

if语句的结构

    在shell脚本中，if语句是最常用的一种流程控制方式，用来根据指定的条件测试结构，分别执行不同的操作（如果……那么……）    

1. 单分支的if语句

案例1：判断/media/cdrom这个目录是否存在，若不存在则新建此目录

[[email protected] ~]# vim dir.sh
#!/bin/bash
DIR="/media/cdrom/"
if [ ! -d $DIR ]
then
    mkdir -p $DIR
fi

案例2：判断当前用户是否为root，如果不是root用户就报错，并退出脚本，当root用户执行脚本时，if语句不执行，正常执行fi之后的脚本代码

[[email protected] ~]# vim root.sh
#!/bin/bash
if [ "$USER" != "root" ]
then
    echo "错误：非root用户，权限不足"
exit 1
fi
fdisk -l /dev/sda

2. 双分支的if语句

案例1：编写一个测试连通性脚本，根据显示结果给出相应的提示

[[email protected] ~]# vim pinghost.sh
#!/bin/bash
ping -c 3 -i 0.2 -W 3 $1 &> /dev/null
if [ $? -eq 0 ]
then
    echo "Host $1 is up."
else
    echo "Host $1 is down."
fi

[[email protected] ~]# . pinghost.sh 192.168.4.11
Host 192.168.4.11 is up.
[[email protected] ~]# . pinghost.sh 192.168.4.13
Host 192.168.4.13 is down.

案例2：通过脚本检测vsftpd服务是否运行，如果以运行则列出其监听地址、PID号，否则输出提示“警告：vsftpd服务不可用！”

[[email protected] ~]# vim ftpd.sh
#!/bin/bash
/etc/init.d/vsftpd status &> /dev/null
if [ $? -eq 0 ]
then
    echo "监听地址：$(netstat -anpt | grep vsftpd | awk '{print $4}')"
    echo "进程PID：$(pgrep -x vsftpd)"
else
    echo "警告：vsftpd服务不可用！"
fi

[[email protected] ~]# sh ftpd.sh
警告：vsftpd服务不可用！
[[email protected] ~]# /etc/init.d/vsftpd start
[[email protected] ~]# sh ftpd.sh
监听地址：0.0.0.0:21
进程PID：986

3. 多分支的if语句

案例：编写一个脚本，根据输入考试分数来区分优秀、合格、不及格

[[email protected] ~]# vim gradediv.sh
#!/bin/bash
read -p "请输入您的分数（0-100）：" GRADE
if [ $GRADE -ge 85 ] && [ $GRADE -le 100 ] 
then
    echo "$GRADE 分！优秀"
elif [ $GRADE -ge 70 ] && [ $GRADE -le 84 ]  
then
    echo "$GRADE 分，合格"
else
    echo "$GRADE 分？不合格”
fi

[[email protected] ~]# sh gradediv.sh
请输入您的分数（0-100）：89
89 分！优秀