Bash Check Arg Number

在编写 Shell 脚本时，很重要的一点是检查输入的参数是否正确。在本篇文章中，我们将介绍如何使用 Bash 脚本来检查输入的参数数量是否正确。

确定参数数量

在 Bash 脚本中，可以使用 $# 来获取输入参数的数量。例如，${#} 将返回传递给脚本的参数数量。

#!/bin/bash
if [[ ${#} -lt 2 ]]; then
  echo "Usage: ${0} <arg1> <arg2>"
  exit 1
fi

在以上示例中，我们检查传递给脚本的参数数量是否小于 2。如果是，则会显示脚本的使用方法，并立即退出。

带选项的参数

如果你的脚本需要处理带选项的参数，例如 -h 或 --help，你可以使用 getopts 来处理。下面是一个简单的示例：

#!/bin/bash
while getopts ":h" opt; do
  case ${opt} in
    h)
      echo "Usage: ${0} <arg1> <arg2>"
      exit 1
      ;;
    \?)
      echo "Invalid option -${OPTARG}"
      echo "Usage: ${0} <arg1> <arg2>"
      exit 1
      ;;
  esac
done

if [[ ${#} -lt $((OPTIND+2)) ]]; then
  echo "Usage: ${0} <arg1> <arg2>"
  exit 1
fi

echo "arg1: ${@:$OPTIND:1}"
echo "arg2: ${@:OPTIND+1}"

在以上示例中，我们使用 getopts 处理 -h 参数。如果用户传递了 -h 参数，将会显示脚本的使用方法，并立即退出。如果用户传递了无效的参数，脚本也会显示使用方法。

总结

检查输入的参数数量是否正确是编写 Bash 脚本时的一项基本任务。我们可以使用 $# 来获取参数数量，并使用 getopts 处理带选项的参数。无论你在写什么样的 Bash 脚本，都应该牢记这个基本任务，以确保脚本的正确性。