【Linux入门】shell基础篇——函数

shell中的函数

Shell函数是Shell脚本编程中的一个基本概念，它允许你将一段代码封装起来，以便在脚本中多次调用，从而提高代码的复用性和可读性。

1. 定义函数

在Shell中，你可以使用function关键字或者更常见的直接使用函数名加括号的方式来定义函数。函数的定义通常放在脚本的开始部分，尽管这不是强制性的。
语法1（使用function关键字）:

function function_name {
    # 函数体
}

语法2（不使用function关键字）:

function_name() {
    # 函数体
}

2. 调用函数

定义了函数之后，你可以在脚本中的任何位置通过函数名加括号（可选的参数列表）来调用它。
调用函数：

function_name [arguments...]

3. 函数参数

Shell函数可以接受参数，这些参数在函数内部通过特殊变量$1, $2, …, ${10}, …来访问，其中$1是第一个参数，$2是第二个参数，依此类推。$#表示传递给函数的参数个数，$*和$@都表示所有传递给函数的参数列表，但在双引号中使用时它们的行为略有不同（"$*"将所有参数视为一个整体，而"$@"将每个参数视为独立的字符串）。
示例：

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

greet "World"

4. 返回值

Shell函数可以通过return语句返回一个退出状态码（一个0到255之间的整数），其中0通常表示成功，非0值表示出现了某种错误或异常情况。但是，函数也可以通过标准输出（echo, printf等命令）来“返回”数据给调用者。
使用return语句：

check_number() {
    if [ "$1" -gt 100 ]; then
        return 0
    else
        return 1
    fi
}

check_number 150
if [ $? -eq 0 ]; then
    echo "The number is greater than 100."
else
    echo "The number is not greater than 100."
fi

通过标准输出返回数据：

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

greeting=$(get_greeting "Alice")
echo "$greeting"

5. 局部变量与全局变量

在Shell函数中定义的变量默认是全局的，即它们在函数外部也是可见的。但是，你可以使用local关键字在函数内部定义局部变量，这些变量仅在函数内部可见。
示例：

myfunc() {
    local myvar="I am local"
    echo "$myvar"
}

myfunc
echo "$myvar"  # 这将不会打印任何内容，因为myvar是局部变量

6. 递归函数

典型递归函数实例：阶乘的计算

阶乘的定义：阶乘是指从1乘以2乘以3乘以…一直乘到所要求的数。例如，4的阶乘是1×2×3×4=24，6的阶乘是1×2×3×…×6=720。阶乘通常用"!"表示，如5!表示5的阶乘。
阶乘可以通过递归或循环的方式进行计算。下面是一个使用递归计算阶乘的Bash脚本示例：

#!/bin/bash
fact () {
  if [ $1 -eq 1 ]
  then
    echo 1
  else
    local temp=$[$1 - 1]
    local result=$(fact $temp)
    echo "$[$1*$result]"
  fi
}
read -p "请输入阶乘数：" n
result=fact $n
echo "$result"

递归函数实例：递归目录

递归目录是指通过递归遍历目录结构，输出目录和文件。下面是一个使用递归遍历目录结构的Bash脚本示例：

#!/bin/bash
function list_files {
  for f in $(ls $1)
  do
    if [ -d "$1/$f" ]
    then
      echo "$f"
      list_files "$1/$f"
    else
      echo "$f"
    fi
  done
}
list_files $1

函数库

函数库是指只包含函数定义的文件，可以在其他脚本中引入并使用这些函数。下面是一个简单的函数库示例：

#!/bin/bash
jiafa () {
  result=$[$1 + $2]
  echo $result
}
jianfa () {
  result=$[$1 - $2]
  echo $result
}
chengfa () {
  result=$[$1 * $2]
  echo $result
}
chufa () {
  if [ $2 -ne 0 ]
  then
    result=$[$1 / $2]
    echo $result
  else
    echo "$2不能等于0！"
  fi
}

在其他脚本中引入函数库的示例：

#!/bin/bash
. /opt/jiaoben1/test8.sh
read -p "输入第一个参数值：" first
read -p "输入第二个参数值：" second
result1=$(jiafa $first $second)
result2=$(jianfa $first $second)
result3=$(chengfa $first $second)
result4=$(chufa $first $second)
echo $result1
echo $result2
echo $result3
echo $result4