Shell脚本 ——Shell 函数

Shell脚本 ——Shell 函数

---

基本介绍： Shell脚本中的函数用于组织和重用代码片段。通过定义函数，可以将复杂的脚本逻辑分解成更小、更易管理的部分。

---

基本语法：
格式1： 使用 function 关键字，用了function关键字时函数名后面的小括号可以忽略。

function function_name() {
	# commands
}

function function_name {
	# commands
}

示例：

#!/bin/bash

function greet() {
  echo "Hello, $1!"
}

greet "World"

输出：
Hello, World!

格式2： 不使用 function 关键字

function_name() {
  # commands
}

示例：

#!/bin/bash

greet() {
  echo "Hello, $1!"
}

greet "World"

输出：
Hello, World!

---

使用函数的参数： 函数可以接受参数，就像脚本本身可以接受命令行参数一样。参数通过 $1, $2 等引用。在函数内部，可以使用 $# 获取参数的个数，使用 $@ 获取所有参数。

示例：

#!/bin/bash
add() {
  local sum=$(( $1 + $2 ))
  echo "Sum: $sum"
}

add 3 5

输出：
Sum: 8

---

函数的返回值： 函数可以通过 return 语句返回一个整数值，表示函数的退出状态。返回值可以通过 $? 获取。需要注意的是，Shell函数的返回值只能是整数，通常用于表示成功（0）或失败（非0）的状态。

示例：

#!/bin/bash

is_even() {
  if (( $1 % 2 == 0 )); then
    return 0
  else
    return 1
  fi
}

is_even 4
if [ $? -eq 0 ]; then
  echo "Number is even"
else
  echo "Number is odd"
fi

输出：
Number is even

---

全局变量与局部变量： 在函数中定义的变量默认是全局变量，可以使用 local 关键字将其定义为局部变量，避免影响全局命名空间。局部变量的作用域仅限于函数内部。

示例：

#!/bin/bash

name="Global"

print_name() {
  local name="Local"
  echo "Inside function: $name"
}

print_name
echo "Outside function: $name"

输出：
Inside function: Local
Outside function: Global

---

示例Shell脚本： 以下是一个综合示例，展示了函数的定义、使用参数、返回值和局部变量。

#!/bin/bash

# 定义函数
greet() {
  echo "Hello, $1!"
}

# 使用函数参数
add() {
  local sum=$(( $1 + $2 ))
  echo "Sum: $sum"
}

# 使用返回值
is_even() {
  if (( $1 % 2 == 0 )); then
    return 0
  else
    return 1
  fi
}

# 全局变量与局部变量
name="Global"

print_name() {
  local name="Local"
  echo "Inside function: $name"
}

# 函数调用
greet "World"
add 3 5

is_even 4
if [ $? -eq 0 ]; then
  echo "Number is even"
else
  echo "Number is odd"
fi

print_name
echo "Outside function: $name"

---

递归函数： Shell脚本中也可以实现递归函数，即函数调用自身。这在需要解决分治问题或处理树形结构时非常有用。
示例： 计算阶乘

#!/bin/bash

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

result=$(factorial 5)
echo "Factorial of 5 is: $result"

输出：
Factorial of 5 is: 120

---

函数库： 为了提高代码的重用性，可以将常用函数放入一个单独的脚本文件中，并在需要时进行调用。这种方法类似于模块化编程。
示例： 定义函数库文件 mylib.sh

#!/bin/bash

greet() {
  echo "Hello, $1!"
}

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

示例： 在主脚本中调用函数库

#!/bin/bash

source mylib.sh

greet "World"
add 3 5

---

注意点：
1、函数命名： 选择具有描述性的函数名称，便于理解函数的作用。
2、参数引用： 通过 $1, $2 等引用函数参数，使用 $@ 可以引用所有参数。
3、局部变量： 使用 local 关键字将变量限定在函数内部，避免全局变量命名冲突。
4、退出状态： 通过 return 语句返回退出状态，用 $? 获取函数的返回值。
5、函数库： 将常用函数放入单独的文件中，提高代码的重用性和模块化。

---

 更多内容请参考 Shell 脚本专栏。