Shell函数

函数简介

在Shell编程中，函数是一种非常有用的工具，它允许我们将一系列命令组织在一起，形成一个可以重复调用的模块。这样做不仅可以提高代码的可读性和可维护性，还可以方便地在不同脚本中重用代码。

函数定义

Shell函数可以通过以下两种格式之一来定义：

function 函数名 {
    命令序列
}

# 或者

函数名() {
    命令序列
}

参数传递

函数可以通过位置参数来接收输入值。例如，我们可以定义一个简单的加法函数：

add() {
    local sum=$(( $1 + $2 ))
    echo $sum
}

echo "输入两个数:"
read num1
read num2
add $num1 $num2

返回值

使用return语句，函数可以返回一个值给调用者。return后面的值可以是0（表示成功）或者非0（表示失败）：

check_root() {
    if [ $(id -u) -eq 0 ]; then
        return 0
    else
        return 1
    fi
}

变量作用域

在Shell函数中，使用local关键字可以定义局部变量，这些变量只在函数内部有效：

increment() {
    local counter=$1
    counter=$((counter + 1))
    echo $counter
}

counter=5
increment $counter
echo "全局变量: $counter"  # 输出仍然是5

函数递归

递归函数是调用自身的函数，这在处理某些问题（如阶乘计算）时非常有用：

factorial() {
    if [ $1 -le 1 ]; then
        echo 1
    else
        local temp=$(( $1 - 1 ))
        local result=$(factorial $temp)
        echo $(( $1 * result ))
    fi
}

read -p "Enter a number: " num
result=$(factorial $num)
echo "Factorial of $num is $result"

函数库

函数库是包含多个函数定义的脚本文件，可以在其他脚本中通过source命令来加载和使用这些函数：

# 在math_functions.sh中定义函数
add() { echo $(( $1 + $2 )); }
subtract() { echo $(( $1 - $2 )); }

# 在主脚本中使用函数库
. ./math_functions.sh
result_add=$(add 10 5)
result_subtract=$(subtract 10 5)
echo "Addition result: $result_add"
echo "Subtraction result: $result_subtract"

通过这种方式，我们可以将相关的功能组织在一起，形成一个模块化的代码库，便于管理和使用。