在 linux 环境中编写 shell 脚本时，掌握如何让函数返回执行状态而不退出整个脚本，以及如何获取脚本的实际路径，是非常实用的技能。本文将深入探讨如何实现函数返回状态而非退出脚本，并推荐几种获取脚本所在位置的方法。
一、让函数返回执行状态而不是退出脚本
在编写 Shell 脚本时，通常会定义函数来实现代码的模块化和重用。然而，如果不慎使用了 exit 命令，可能会导致整个脚本退出，这不是我们期望的结果。为了让函数返回执行状态而不退出脚本，可以使用 return 命令。以下是一个示例代码：
#!/bin/bash
<h1>定义一个示例函数</h1><p>my_function() {
echo "Running my_function..."</p><pre class="brush:php;toolbar:false;"><code># 假设某些操作失败
if [ "$1" -eq 0 ]; then
echo "Operation failed."
return 1
else
echo "Operation succeeded."
return 0
fi}
调用函数并捕获返回状态
my_function "$1" status=$?
检查函数的返回状态
if [ $status -eq 0 ]; then echo "Function executed successfully." else echo "Function execution failed with status $status." fi
echo "Script continues..."
在这个示例中：
- 定义函数：
my_function函数根据传入的参数返回不同的状态码。 - 使用
return命令：函数内部使用return返回状态码。 - 捕获返回状态：调用函数后，通过
$?变量获取返回状态。 - 检查返回状态：根据返回状态执行不同的逻辑，而不会影响脚本的继续执行。
二、设计和使用 main 函数
尽管 Shell 脚本不像某些编程语言那样强制要求使用 main 函数，但通过定义 main 函数可以使脚本逻辑更加清晰和结构化。以下是一个包含 main 函数的示例：
#!/bin/bash</p><h1>定义全局变量</h1><p>global_var="Hello, World!"</p><h1>定义一个辅助函数</h1><p>helper_function() {
echo "This is a helper function."
}</p><h1>定义 main 函数</h1><p>main() {
echo "Starting script..."</p><h1>调用辅助函数</h1><pre class="brush:php;toolbar:false;"><code>helper_function
# 使用全局变量
echo "Global variable: $global_var"
# 获取命令行参数
if [ $# -eq 0 ]; then
echo "No arguments provided."
return 1
else
echo "Arguments provided: $@"
fi
echo "Script execution completed."
return 0}
调用 main 函数并传递所有命令行参数
main "$@"
在这个示例中：
- 定义辅助函数：
helper_function用于执行辅助任务。 - 定义
main函数：在main函数中进行主要的脚本逻辑处理，包括调用辅助函数、处理全局变量和命令行参数。 - 调用
main函数：在脚本末尾调用main函数，并传递所有命令行参数。
三、使用 Heredoc 处理多行文本
在 Shell 脚本中，Heredoc（"Here Document" 的缩写）是一种方便的方法，用于将多行字符串传递给命令或赋值给变量。它可以让你在脚本中轻松地处理多行文本，而无需使用多个 echo 命令。
Heredoc 基本语法：
command <<- EOF 多行文本内容 EOF
其中：
-
command是你想要传递多行文本的命令，例如cat、echo、sed等。 -
EOF是一个自定义的结束标记符，可以是任意你选择的标识符，常用的有EOF、END等。 中的 <code>-表示忽略行首的制表符，方便在脚本中缩进文本内容。
示例用法：
- 使用
cat显示多行文本
#!/bin/bash cat <<- EOF 这是一个多行文本示例。 它可以包含多行内容， 并且可以进行缩进。 EOF
- 赋值给变量
#!/bin/bash my_var=$(cat <<- EOF 这是一个多行文本示例。 它可以包含多行内容， 并且可以进行缩进。 EOF ) echo "$my_var"
- 结合命令使用（例如，
grep）
#!/bin/bash grep "pattern" <<- EOF 这是一个多行文本示例。 它可以包含多行内容， 并且可以进行缩进。 EOF
忽略行首制表符
通过使用 而不是 <code>，可以忽略每行开头的制表符，这样可以使 Heredoc 内容更加美观和易读：
#!/bin/bash cat <<- EOF 这是一个多行文本示例。 它可以包含多行内容， 并且可以进行缩进。 EOF
在脚本中使用变量
你可以在 Heredoc 中使用脚本中的变量：
#!/bin/bash name="Alice" cat <<- EOF Hello, $name! 这是一个多行文本示例。 它可以包含多行内容， 并且可以进行缩进。 EOF
禁止变量扩展
如果希望 Heredoc 中的内容不进行变量扩展，可以使用引号（如 或 <code>）：
#!/bin/bash name="Alice" cat <<'EOF' Hello, $name! 这是一个多行文本示例。 它可以包含多行内容， 并且可以进行缩进。 EOF
应用场景
Heredoc 常用于：
- 向命令传递多行文本，例如
cat、grep、sed等。 - 生成多行配置文件或脚本内容。
- 提高脚本的可读性和维护性，避免使用多个
echo命令。
Heredoc 是 Shell 脚本中处理多行文本的强大工具，可以大大简化脚本的编写和维护。
四、获取脚本所在位置
在 Shell 脚本中，获取脚本所在的位置和调用者所在的位置是两个不同的概念。理解和正确使用这两者，可以让你的脚本更加灵活和强大。以下是常用的获取脚本所在位置的几种方法及其区别。
获取脚本所在位置
要获取脚本所在的位置，可以使用以下几种方法：
方法一：$0 和 dirname
#!/bin/bash script_dir=$(dirname "$0") echo "Script directory: $script_dir"
这种方法适用于大多数情况，但它有一个缺点：如果脚本是通过符号链接调用的，它将返回符号链接的目录，而不是实际脚本的位置。
方法二：BASH_SOURCE
#!/bin/bash
script_dir=$(dirname "${BASH_SOURCE[0]}")
echo "Script directory: $script_dir"BASH_SOURCE 数组包含脚本的实际路径，即使是通过符号链接调用也能获取到实际位置。这个方法是推荐的。
方法三：realpath
#!/bin/bash script_dir=$(dirname "$(realpath "$0")") echo "Script directory: $script_dir"
realpath 可以解析符号链接，返回脚本的实际路径。需要注意的是，这个方法依赖于 realpath 命令的存在。
获取调用者所在位置
要获取调用者所在的位置，可以使用以下方法：
#!/bin/bash caller_dir=$(pwd) echo "Caller directory: $caller_dir"
这个方法使用 pwd 命令获取当前工作目录，也就是调用脚本的位置。
完整示例
以下是一个完整的示例，展示如何同时获取脚本所在的位置和调用者所在的位置：
#!/bin/bash</p><h1>获取脚本所在位置</h1><p>script_dir=$(dirname "${BASH_SOURCE[0]}")</p><h1>获取调用者所在位置</h1><p>caller_dir=$(pwd)</p><p>echo "Script directory: $script_dir"
echo "Caller directory: $caller_dir"</p><h1>示例操作</h1><p>echo "Listing files in script directory:"
ls "$script_dir"
echo "Listing files in caller directory:"
ls "$caller_dir"在编写可移植的 Shell 脚本时，获取脚本和调用所在的位置，并区别它们非常重要。这可以确保脚本可以在任何目录下正确调用，而无需担心路径问题，提高脚本的可移植性和灵活性。
总结
在 Shell 脚本中，正确处理函数返回状态、设计和使用 main 函数、利用 Heredoc 处理多行文本以及获取脚本所在位置是编写高质量脚本的重要技巧。通过掌握这些技巧，可以编写出更加灵活、可读和易于维护的 Shell 脚本。
