shell编程之函数

概述

在 Shell 脚本中，函数是一种组织代码的方式，它允许你将一组相关的命令组合在一起，以便在脚本的不同地方重复使用

简单点说就是将shell脚本中重复的代码提取出来，精简脚本的内容，让脚本更加规范，可读性更高。

应用场景

在书写脚本的时候，尽量使用函数，规范脚本
让脚本的开发，更加模块化
方便后期的调试，调试的时候只需要注释函数即可

函数的语法

函数定义使用function关键字，

注意：func_name代表函数名称

第一种（最完整的）

复制代码

function func_name() {
  shell内容
  shell内容
  ...
  return 返回结果
}

第二种

复制代码

func_name() {
  shell内容
  shell内容
  ...
  return 返回结果
}

第三种

复制代码

function func_name {
  shell内容
  shell内容
  ...
  return 返回结果
}

如何调用函数

在脚本中直接使用函数名即可。
调用函数需要在函数声明后调用，如果是声明前调用则会出现异常

示例：

复制代码

# 编写脚本
[root@master01 ~]# cat shell01.sh 
#!/bin/bash
function hello() {
  echo "hello shell"
}
hello

# 加上执行权限
[root@master01 ~]# chmod +x shell01.sh 

# 执行脚本
[root@master01 ~]# ./shell01.sh 
hello shell

如何给函数传递参数

在 Shell 中，函数可以接受参数，这些参数通过位置传递，就像脚本接受命令行参数一样。

$1,$ 2, ... ：表示传递给函数的第 1 个、第 2 个... 参数
$# ：传递给函数的参数个数
$@ ：所有参数列表
$* ：所有参数列表（以单个字符串形式）

示例：

复制代码

# 定义脚本
[root@master01 ~]# cat shell02.sh 
#!/bin/bash
function add_number() {
 echo '你输入的第一个参数：'$1
 echo '你输入的第二个参数：'$2
 count=`awk -v num1=$1 -v num2=$2 'BEGIN {print num1+num2}'`
 echo '和为：'$count
}
add_number 2 4

# 添加执行权限
[root@master01 ~]# chmod +x shell02.sh 

# 执行脚本
[root@master01 ~]# ./shell02.sh 
你输入的第一个参数：2
你输入的第二个参数：4
和为：6

如何获取函数返回值

Shell 函数的返回值通过退出状态码（exit status）传递，范围是 0-255，其中 0 表示成功，非零表示失败。

使用 return 语句设置返回值
使用 $? 获取函数的返回值

示例：

复制代码

# 定义脚本
[root@master01 ~]# cat shell03.sh 
#!/bin/bash
function add_number() {
 echo '你输入的第一个参数：'$1
 echo '你输入的第二个参数：'$2
 count=`awk -v num1=$1 -v num2=$2 'BEGIN {print num1+num2}'`
 echo '和为：'$count
 return 222
}
add_number 2 4
echo '函数返回值：'$?

# 给脚本添加执行权限
[root@master01 ~]# chmod +x shell03.sh 

# 执行脚本
[root@master01 ~]# ./shell03.sh
你输入的第一个参数：2
你输入的第二个参数：4
和为：6
函数返回值：222

变量作用域

变量根据所在位置不同分为两类：

全局变量：在函数外部定义的变量，函数内部可以直接访问
局部变量：在函数内部使用 local 关键字定义的变量，只在函数内部可见

示例：

复制代码

global_var="global"

function test_scope {
    local local_var="local"  # 局部变量
    echo "Inside function: global_var=$global_var, local_var=$local_var"
}

# 输出：Inside function: global_var=global, local_var=local
test_scope 

# 输出：Outside function: global_var=global, local_var=
echo "Outside function: global_var=$global_var, local_var=$local_var"

注意事项：

函数必须在调用之前定义
函数内部的命令执行失败（返回非零状态码）不会自动终止函数，需要显式处理
函数返回值只能是 0-255 的整数，如果需要返回其他类型的值，可以通过 echo 输出并捕获
使用 local 关键字定义局部变量，避免意外修改全局变量