shell function（建议收藏）

什么是 Shell Function？从零开始掌握脚本中的“小函数”

在日常的 Linux 或 macOS 系统使用中，你可能已经写过不少 shell 脚本。它们能自动执行一系列命令，比如备份文件、检查服务状态、清理日志等。但当你需要重复执行某个操作时，直接复制粘贴命令会变得繁琐且容易出错。

这时，shell function 就像一个“工具箱里的小零件”——你可以把它封装起来，随时调用，既干净又高效。它不是独立的程序，而是嵌在当前 shell 环境中的一段可重用代码块，特别适合快速编写自动化任务。

想象一下，你每天都要执行 ls -la 查看文件详情，如果每次都要敲这行命令，久了也会烦。但如果你定义一个叫 ll 的 function，只需要输入 ll 就能完成同样的事，岂不省心？

如何定义一个 Shell Function？

定义一个 shell function 的语法非常简单，结构清晰：

function_name() {
    # 函数体：一系列要执行的命令
    echo "这是函数的内容"
    date
}

注意几点细节：

  • 函数名后必须带括号 ()，即使没有参数也必须写。
  • 大括号 {} 和括号之间至少要有一个空格。
  • 函数体内的命令会按顺序执行，就像脚本一样。
  • 不需要 return 关键字来返回值（后面会讲）。

举个实际例子，我们来创建一个叫 greet 的函数，用来打招呼：

greet() {
    # 输出欢迎信息
    echo "你好！欢迎使用本脚本"
    # 显示当前时间
    date "+%Y-%m-%d %H:%M:%S"
}

现在，只要你在终端里输入 greet，它就会自动执行这两个命令。是不是很清爽？

💡 小贴士：你可以在当前 shell 中直接定义函数，无需保存到文件。但如果你想长期使用，建议写进 .bashrc 或 .zshrc 文件中。

参数传递与变量作用域

函数最实用的功能之一，就是可以接收参数。就像你给手机发消息时要写“你好”，而对方回复时能根据你的名字个性化回应一样，函数也能“知道”你传了什么。

在 shell function 中，参数通过 $1, $2, ..., $n 来访问，其中 $1 是第一个参数，$2 是第二个，以此类推。

我们来定义一个简单的加法函数：

add() {
    # 检查是否传入了两个参数
    if [ $# -ne 2 ]; then
        echo "错误：必须传入两个数字"
        return 1
    fi

    # 将参数赋值给本地变量
    local num1=$1
    local num2=$2

    # 执行加法运算
    local sum=$((num1 + num2))

    # 输出结果
    echo "结果是：$sum"
}
  • # 表示参数个数，$# 就是传入参数的数量。
  • local 关键字用于声明局部变量，防止污染全局环境。
  • return 1 表示函数执行失败，返回非零状态码。

使用方式如下：

add 5 3

add 10 20

⚠️ 注意：return 只能返回整数状态码（0 表示成功，非 0 表示失败），不能返回字符串或数值结果。如果需要返回值，可以通过 echo 输出，再用命令替换获取。

函数返回值的正确处理方式

很多人会误以为 shell function 可以像 Python 或 Java 那样用 return 返回一个值。但在 shell 中，return 只能返回状态码。

那怎么获取函数的“返回值”呢？答案是：用 echo 输出结果，然后通过命令替换 $(...) 捕获。

来看一个例子：

get_greeting() {
    # 根据时间判断问候语
    local hour=$(date +%H)

    if [ $hour -lt 12 ]; then
        echo "早上好！"
    elif [ $hour -lt 18 ]; then
        echo "下午好！"
    else
        echo "晚上好！"
    fi
}

调用它并获取返回内容：

message=$(get_greeting)
echo "$message"

这个技巧非常重要：函数的“返回值”本质上就是它输出的内容。所以你要确保函数输出你想用的那部分。

常见应用场景与实用技巧

创建数组与初始化

虽然 shell 的数组语法略显原始，但配合 function 可以实现很多灵活操作。比如创建一个函数来初始化一个包含多个项目的数组：

init_projects() {
    # 定义一个数组，存储项目名称
    local projects=(
        "用户管理系统"
        "订单处理模块"
        "报表生成器"
        "权限控制服务"
    )

    # 遍历数组并打印每个项目
    for project in "${projects[@]}"; do
        echo "项目：$project"
    done
}

调用后会列出所有项目。这种写法在批量处理任务时非常有用。

文件操作自动化

假设你经常需要创建一组目录结构，比如为新项目准备 src, docs, tests 等文件夹。你可以写一个函数来一键完成：

setup_project_dir() {
    # 检查是否传入项目名
    if [ -z "$1" ]; then
        echo "错误：请指定项目名称"
        return 1
    fi

    local project_name=$1

    # 创建主目录
    mkdir -p "$project_name"

    # 在主目录下创建子目录
    mkdir -p "$project_name/src"
    mkdir -p "$project_name/docs"
    mkdir -p "$project_name/tests"
    mkdir -p "$project_name/config"

    # 输出成功信息
    echo "项目目录已创建：$project_name"
}

使用方式：

setup_project_dir myapp

你会发现，原本要敲好几行 mkdir 的操作，现在只用一个命令搞定。

条件判断与错误处理

函数中嵌套 if-else 是常见需求。下面这个函数用于检查某个服务是否运行：

check_service() {
    local service_name=$1

    # 如果没有传入服务名，报错
    if [ -z "$service_name" ]; then
        echo "请提供服务名称"
        return 1
    fi

    # 使用 systemctl 检查服务状态
    if systemctl is-active --quiet "$service_name"; then
        echo "✅ 服务 $service_name 正在运行"
        return 0
    else
        echo "❌ 服务 $service_name 未运行"
        return 1
    fi
}

调用示例：

check_service nginx
check_service mysql

这个函数结构清晰，错误处理完整，是编写健壮脚本的好模板。

最佳实践：让函数更可靠、更易维护

  1. 使用 local 声明变量
    避免函数内部变量污染全局环境，尤其是循环、条件判断中。

  2. 添加参数校验
    检查 $#，确保用户传入了足够的参数。

  3. 返回状态码
    return 0 表示成功，return 1 或更高表示失败，便于脚本链式调用。

  4. 避免在函数中使用全局变量
    若必须使用，建议加上前缀如 GLOBAL_ 以示区分。

  5. 文档化你的函数
    在函数上方加注释说明用途、参数、返回值，方便日后维护。

总结：Shell Function 是提升效率的秘密武器

shell function 看似简单，实则威力巨大。它让你不再重复敲命令，而是把常用逻辑封装成“小工具”，随时调用。

从打招呼的 greet，到项目目录的 setup_project_dir，再到服务状态检查的 check_service，每一个函数都像是你在命令行世界里安放的一颗螺丝钉——虽小，但不可或缺。

掌握 shell function，不仅意味着你能写出更简洁的脚本，更意味着你开始用“编程思维”去思考自动化问题。当你能用几行代码完成原本需要十分钟的手动操作时，那种掌控感，真的很爽。

所以，别再让重复劳动浪费你的时间了。从今天起，给你的终端加点“函数”吧。