如何为Bash脚本getopts的flag参数设置有效默认值

默认值不生效核心是3个代码错误：

1. 条件判断语法错误：[ $counts==true ] 这类写法不符合shell语法要求，[ ] 结构内部的运算符、变量两侧必须加空格，否则整段内容会被识别为单个非空字符串，判断结果永远为真，完全不会按变量值走分支。另外判断逻辑本身也不成立：你存入变量的是用户传入的执行次数、超时时间、用户名，永远不会等于true字符串。
2. 默认值逻辑错误：
   • -c 要求默认无限次执行，对应top命令的行为是不传-n参数（top默认就是无限循环刷新），不需要给counts强行赋值
   • -u 要求默认匹配任意用户，对应top命令的行为是不传-u参数（top默认显示所有用户进程），你之前设的默认值.不是top支持的用户名匹配规则，传了反而会报错
3. 缺少参数移位逻辑：getopts解析完flag后没有移除已解析的选项，无法正确读取后面跟的exe-name位置参数，同时没有做参数合法性校验，传非法值时会直接抛出晦涩的系统错误。

正确代码实现 #

#!/bin/bash
set -u

print_usage(){
    echo "usage: $0 [-c 执行次数] [-t 刷新间隔秒数] [-u 用户名] 进程名"
}

# ------------ 在这里初始化默认值，必须放在getopts解析之前 ------------
counts=""        # 空值代表不传-n参数，默认top无限次执行
timeout=1        # 默认刷新间隔1秒
user_name=""     # 空值代表不传-u参数，默认匹配所有用户
# -------------------------------------------------------------------

if [[ $# -eq 0 ]]; then
    print_usage
    exit 1
fi

# getopts参数解析，增加非法参数、缺参数处理
while getopts :c:t:u: flag; do
    case $flag in
        c)
            # 校验-c传入的是正整数
            if ! [[ $OPTARG =~ ^[1-9][0-9]*$ ]]; then
                echo "错误：-c参数必须传入正整数"
                print_usage
                exit 1
            fi
            counts=$OPTARG
            ;;
        t)
            # 校验-t传入的是合法数字
            if ! [[ $OPTARG =~ ^[0-9]+(\.[0-9]+)?$ ]]; then
                echo "错误：-t参数必须传入合法的秒数"
                print_usage
                exit 1
            fi
            timeout=$OPTARG
            ;;
        u)
            user_name=$OPTARG
            ;;
        :)
            echo "错误：-$OPTARG 参数缺少必填值"
            print_usage
            exit 1
            ;;
        ?)
            echo "错误：不支持的参数 -$OPTARG"
            print_usage
            exit 1
            ;;
    esac
done

# 移除已解析的flag参数，读取后续位置参数
shift $((OPTIND - 1))
if [[ $# -lt 1 ]]; then
    echo "错误：缺少必填的进程名参数"
    print_usage
    exit 1
fi
exe_name=$1

# 拼接top执行参数
top_cmd_args=()
top_cmd_args+=( -d "$timeout" )  # 刷新间隔默认1秒，永远生效
[[ -n $counts ]] && top_cmd_args+=( -n "$counts" )  # 用户传了-c才加次数参数
[[ -n $user_name ]] && top_cmd_args+=( -u "$user_name" )  # 用户传了-u才加用户过滤参数

# 执行命令，可自行添加grep过滤exe_name的逻辑
top "${top_cmd_args[@]}"

关键注意点 #

• 默认值必须在getopts解析逻辑之前定义，用户传入对应参数时会自动覆盖默认值
• shell中[ ]条件判断的所有操作符、变量前后必须留空格，比如判断变量非空要写[ -n "$var" ]，不能把等号和变量写在一起
• 对应命令的默认行为如果是「不传某个参数就生效」，不要硬给参数塞无意义的默认值，直接判断变量非空时再加参数即可
• 用数组拼接命令参数可以避免空格、特殊字符导致的命令执行错误，比字符串拼接更安全

内容的提问来源于stack exchange，提问作者nati malm