Shell 编程进阶：流程控制与函数

原文链接
在掌握了 Shell 编程的基础概念和变量操作后，我们将深入探讨 Shell 脚本中的流程控制和函数。这些高级特性让我们能够编写更复杂、更强大的脚本，实现更高效的自动化任务。让我们一起学习如何使用条件语句、循环结构和函数来增强 Shell 脚本的功能。

Shell 编程入门：掌握基础，开启命令行编程之旅
Shell 编程基础：变量和数据类型

条件语句

条件语句允许脚本根据特定条件执行不同的代码块。Shell 脚本中主要使用 if、elif 和 else 来构建条件语句。

if 语句

最基本的条件语句是 if 语句：

if [ condition ]; then
    # 如果条件为真，执行这里的代码
fi

例子：检查文件是否存在

#!/bin/bash

file="/path/to/file"
if [ -f "$file" ]; then
    echo "文件存在"
fi

注释：

• -f 是一个文件测试操作符，用于检查文件是否存在且为普通文件。
• 使用双引号包裹变量可以防止路径中的空格导致错误。

if-else 语句

当我们需要在条件不满足时执行另一段代码，可以使用 if-else 结构：

if [ condition ]; then
    # 条件为真时执行
else
    # 条件为假时执行
fi

例子：检查用户输入的数字是奇数还是偶数

#!/bin/bash

read -p "请输入一个数字: " num

if [ $((num % 2)) -eq 0 ]; then
    echo "$num 是偶数"
else
    echo "$num 是奇数"
fi

注释：

• read -p 用于提示用户输入并将结果存储在变量中。
• $((num % 2)) 计算 num 除以 2 的余数。
• -eq 是比较操作符，用于检查两个数是否相等。

elif 语句

对于多条件判断，我们可以使用 elif（else if 的缩写）：

if [ condition1 ]; then
    # 条件1为真时执行
elif [ condition2 ]; then
    # 条件2为真时执行
else
    # 所有条件都为假时执行
fi

例子：根据分数给出等级评价

#!/bin/bash

read -p "请输入你的分数（0-100）: " score

if [ $score -ge 90 ]; then
    echo "优秀"
elif [ $score -ge 80 ]; then
    echo "良好"
elif [ $score -ge 70 ]; then
    echo "中等"
elif [ $score -ge 60 ]; then
    echo "及格"
else
    echo "不及格"
fi

注释：

• -ge 是比较操作符，表示"大于或等于"。
• 条件按照从上到下的顺序评估，一旦满足就执行相应的代码块。

循环结构

循环允许我们重复执行一段代码。Shell 脚本主要支持 for 循环和 while 循环。

for 循环

for 循环用于遍历一个列表中的元素：

for item in list
do
    # 对每个item执行的操作
done

例子：遍历目录中的文件

#!/bin/bash

for file in /path/to/directory/*
do
    echo "文件名: $file"
done

注释：

• * 是通配符，匹配目录中的所有文件。
• 这个循环会遍历指定目录中的所有文件，并打印它们的名称。

while 循环

while 循环在条件为真时重复执行代码块：

while [ condition ]
do
    # 循环体
done

例子：倒计时程序

#!/bin/bash

count=5
while [ $count -gt 0 ]
do
    echo $count
    count=$((count - 1))
    sleep 1
done
echo "时间到！"

注释：

• -gt 是比较操作符，表示"大于"。
• sleep 1 使脚本暂停1秒。

函数

函数是一段可重复使用的代码块，可以提高代码的可读性和可维护性。

定义和调用函数

function_name() {
    # 函数体
}

# 调用函数
function_name

例子：创建一个简单的问候函数

#!/bin/bash

greet() {
    echo "你好，$1！"
}

greet "Alice"
greet "Bob"

注释：

• $1 表示传递给函数的第一个参数。
• 函数可以多次调用，每次可以传入不同的参数。

返回值

Shell 函数可以使用 return 语句返回数值，或者通过 echo 输出结果：

#!/bin/bash

add_numbers() {
    local result=$((1 + $2))
    echo $result
}

sum=$(add_numbers 5 3)
echo "5 + 3 = $sum"

注释：

• local 关键字用于声明局部变量。
• 函数的输出可以通过命令替换 $() 捕获。

高级技巧

命令行参数

脚本可以接受命令行参数，使用 $1, $2 等访问：

#!/bin/bash

echo "脚本名称: $0"
echo "第一个参数: $1"
echo "第二个参数: $2"
echo "所有参数: $@"
echo "参数个数: $#"

错误处理

使用 set -e 可以在遇到错误时立即退出脚本：

#!/bin/bash
set -e

# 如果这个命令失败，脚本将立即退出
non_existent_command

echo "这行不会被执行"

调试技巧

使用 set -x 可以打印每个执行的命令：

#!/bin/bash
set -x

echo "这是一个调试示例"
ls /non_existent_directory

总结

本文深入探讨了 Shell 编程中的流程控制和函数，包括条件语句、循环结构、函数定义和调用等关键概念。我们通过实际例子展示了如何使用这些特性来编写更复杂、更强大的 Shell 脚本。此外，还介绍了一些高级技巧，如命令行参数处理、错误处理和调试方法。掌握这些知识将帮助您编写更高效、更可靠的 Shell 脚本，进一步提升自动化任务的能力。