Go GetoptGo 解析命令行参数工具-技术圈

Go GetOpt，让你在 go 里解析命令行参数~~无聊地~~跟写 shell 脚本一样。

为了不引起混淆，以下说明将使用

go getopt 表示本代码仓库

shell getopt、getopt 命令 表示 util-linux 中的 getopt 二进制程序

getopt（或 C getopt）表示 libc 中的 getopt 方法

但在某个上下文（如标题说明了该段是 shell getopt）中可能有时会直接使用 getopt 指代。请各位注意区分。

怎么用



go get gitee.com/go-getopt/go-getopt



package main



import (

    "fmt"

    "os"



    // 这里为了方便，直接使用 . 进行 import。

    // 这样可以直接调用 GetOpt、Get 和 Shift 方法。

    . "gitee.com/go-getopt/go-getopt"

)



func main() {

    // 传入 os.Args、options 和 longOptions 字符串参数即可

    err := GetOpt(os.Args, "ab:c::", "a-long,b-long:,c-long::")

    if err != nil {

        fmt.Fprintln(os.Stderr, err)

        os.Exit(1)

    }



    // 解析后的参数列表存储在全局变量 Args 中

    fmt.Println("Arguments:", Args)

    fmt.Println("Program name:", Args[0])



    // 接下来的步骤就和 shell 差不多了

    for loop := true; loop; {

        switch Get(1) {

        case "-a", "--a-long":

            fmt.Println("Option a")

            Shift(1)

        case "-b", "--b-long":

            fmt.Println("Option b, argument '" + Get(2) + "'")

            Shift(2)

        case "-c", "--c-long":

            if Get(2) == "" {

                fmt.Println("Option c, no argument")

            } else {

                fmt.Println("Option c with arg '" + Get(2) + "'")

            }

            Shift(2)

        case "--":

            Shift(1)

            loop = false

        default:

            fmt.Fprintln(os.Stderr, "Error: wrong argument '"+arg1+"'")

            os.Exit(1)

        }

    }

    fmt.Println("Remains:", Args[1:])

}

对比一下 shell getopt 解析命令行的脚本：



# 检查 getopt 命令是否正常运行

getopt --test > /dev/null



[ $? -ne 4 ] &&

    echo "Error: command 'getopt --test' failed in this environment." &&

    exit 1



# 设定 options 和 longOptions，调用 getopt 命令

options=ab:c::

longOptions=a-long,b-long:,c-long::



parsed=$(getopt --options=$options --longoptions=$longOptions --name "$0" -- "$@")



[ $? -ne 0 ] &&

    echo "Error: failed to parse cmd arguments" &&

    exit 1



eval "set -- $parsed"



# 循环判断是哪个 flag，处理完后 shift 掉

while true; do

    case "$1" in

        -a|--a-long)

            echo 'Option a'

            shift

        ;;

        -b|--b-long)

            echo "Option b, argument '$2'"

            shift 2

        ;;

        -c|--c-long)

            [ -n "$2" ] && \\

            echo "Option c, argument '$2'" || \\

            echo 'Option c, no argument'

            shift 2

        --)

            shift

            break

        *)

            echo "Error: wrong argument '$1'"

            exit 1

            ;;

    esac

done

echo "Remains: $@"