验证 bash 脚本参数

Question

我正在尝试做这样的事情来制作脚本以在备份失败时执行备份。我将环境作为脚本的参数。

我不确定如何做的一件事是我想验证 $1 以仅包含一些预定义的值。预定义的值应该类似于 tst、prd、qa、rpt。任何人？

#!/bin/bash
ENVIRONMENT=$1 
BACKUPDATE=$(date +"%d_%m_%Y")
BACKUPFILE="$ENVIRONMENT".backup."$BACKUPDATE".tar.gz

if [ $1 ==  "" ] 
 then
 echo "No environment specified"
 exit
elif [ -f "$BACKUPFILE" ]; then
   echo "The file '$BACKUPFILE' exists."
else
   echo "The file '$BACKUPFILE' in not found."
   exec touch "$BACKUPFILE"
fi

Answer 1

你可以使用case:

case "$1" in
    tst) echo "Backing up Test style" ;;
    prd)
        echo "Production backup"
        /etc/init.d/myservice stop
        tar czf ...
        /etc/init.d/myservice start
        ;;
    qa) echo "Quality skipped" ;;
    rpt)
        echo "Different type of backup"
        echo "This could be another processing"
        ...
        ;;
    *)
        echo "Unknown backup type"
        exit 2
        ;;
esac

注意结束每个案例的双 ;;，以及 convenient use of pattern matching。

编辑：按照您的评论和@CharlesDuffy 的建议，如果您想在一个数组中包含所有有效选项并针对其中的 any 测试您的值（因此对于 所有有效值），您可以使用associative array：

declare -A valids=(["tst"]=1 ["prd"]=1 ["qa"]=1 ["rpt"]=1)
if [[ -z ${valids[$1]} ]] ; then
    echo "Invalid parameter value"
    # Any other processing here ...
    exit 1
fi
# Here your parameter is valid, proceed with processing ...

这通过将值（此处为 1 但在这种情况下可能是其他任何值）分配给 每个 有效参数来实现。所以任何 invalid 参数都将为空，-z 测试将触发。

功劳归他所有。

Answer 2

根据您有多少不同的值，case 语句呢？它甚至允许通配符。

case $1 in
  (John)   printf "Likes Yoko\n";;
  (Paul)   printf "Likes to write songs\n";;
  (George) printf "Harrison\n";;
  (Ringo)  printf "Da drumma\n";;
  (*)      printf "Management, perhaps?\n";;
esac

另一方面，如果可以，您应该避免使用不可移植的 bashism，例如 [[ 测试运算符（如果可以，请使用 [，例如 if [ "$1" = "John" ]; then ...; fi。）