GO命令行詳細(xì)價紹

這篇文章為大家?guī)碛嘘P(guān)GO命令行的詳細(xì)介紹。大部分命令行知識點(diǎn)都是大家經(jīng)常用到的，為此分享給大家做個參考。一起跟隨小編過來看看吧。

目錄

? 一. os.Args
? 二. flag
? 三. 結(jié)合os.Args與flag實現(xiàn)子命令
? 附. 參考文檔

一. os.Args

os.Args用于獲取通過命令行傳入的參數(shù)

? os.Args[0]：程序執(zhí)行路徑
? os.Args[1]：第1個參數(shù)
? os.Args[2]：第2個參數(shù)
? len(os.Args)：參數(shù)數(shù)量

示例如下

package main

import (
    "fmt"
    "os"
)

func main() {
    fmt.Println("Program:", os.Args[0])
    for i, v := range os.Args[1:] {
        fmt.Printf("Arg[%d]: %v\n", i, v)
    }
}

測試效果

# 編譯執(zhí)行
go build arg.go
./arg foo "hello world" bar

# 輸出
Program: ./arg
Arg[0]: foo
Arg[1]: hello world
Arg[2]: bar

二. flag

golang內(nèi)置的flag模塊可以實現(xiàn)命令行flag解析。

什么是flag，舉例說明（加粗部分就是flag）：

? wc -l
? ./configure --prefix=/usr/local/nginx

示例如下（摘取自gobyexample）

package main

import (
    "flag"
    "fmt"
)

func main() {
    // flag.String返回的是指針
    // word表示參數(shù)名，foo表示默認(rèn)值，a string表示參數(shù)釋義（在-h或解析異常時候會看到）
    // flag.Int、flag.Bool與flag.String同理，不再贅述
    wordPtr := flag.String("word", "foo", "a string")
    numbPtr := flag.Int("numb", 42, "an int")
    boolPtr := flag.Bool("fork", false, "a bool")

    // flag.StringVar返回的是非指針
    var svar string
    flag.StringVar(&svar, "svar", "bar", "a string var")

    // 進(jìn)行flag解析
    flag.Parse()

    fmt.Println("word:", *wordPtr)
    fmt.Println("numb:", *numbPtr)
    fmt.Println("fork:", *boolPtr)
    fmt.Println("svar:", svar)
    fmt.Println("tail:", flag.Args())
}

測試效果

# 編譯
go build flag.go

./flag -word=opt -numb=7 -fork -svar=flag
# 輸出
word: opt
numb: 7
fork: true
svar: flag
tail: []

./flag -word=opt
# 輸出
word: opt
numb: 42
fork: false
svar: bar
tail: []

./flag -word=opt a1 a2 a3
# 輸出
word: opt
numb: 42
fork: false
svar: bar
tail: [a1 a2 a3]

./flag -word=opt a1 a2 a3 -numb=7
# 輸出
word: opt
numb: 42
fork: false
svar: bar
tail: [a1 a2 a3 -numb=7]

./flag -h
# 輸出
Usage of ./flag:
  -fork
        a bool
  -numb int
        an int (default 42)
  -svar string
        a string var (default "bar")
  -word string
        a string (default "foo")

./flag -wat
# 輸出
flag provided but not defined: -wat
Usage of ./flag:
...

該例子，在測試時候只使用了“單橫線+參數(shù)名”，實際有4種方式，效果是相同的

-word opt
-word=opt
--word opt
--word=opt

三. 結(jié)合os.Args與flag實現(xiàn)子命令

如果想實現(xiàn)子命令效果，例如：

? docker ps --last 10
? docker images --help

這需要將os.Args和flag結(jié)合使用?；驹恚簞?chuàng)建多個flag集合（即*flag.FlagSet），每個flag集合對應(yīng)不同的os.Args

示例如下（摘取自gobyexample）

package main

import (
    "flag"
    "fmt"
    "os"
)

func main() {
    fooCmd := flag.NewFlagSet("foo", flag.ExitOnError)
    fooEnable := fooCmd.Bool("enable", false, "enable")
    fooName := fooCmd.String("name", "", "name")

    barCmd := flag.NewFlagSet("bar", flag.ExitOnError)
    barLevel := barCmd.Int("level", 0, "level")

    if len(os.Args) < 2 {
        fmt.Println("expected 'foo' or 'bar' subcommands")
        os.Exit(1)
    }

    switch os.Args[1] {
    case "foo":
        fooCmd.Parse(os.Args[2:])
        fmt.Println("subcommand 'foo'")
        fmt.Println(" enable:", *fooEnable)
        fmt.Println(" name:", *fooName)
        fmt.Println(" tail:", fooCmd.Args())
    case "bar":
        barCmd.Parse(os.Args[2:])
        fmt.Println("subcommand 'bar'")
        fmt.Println(" level:", *barLevel)
        fmt.Println(" tail:", barCmd.Args())
    default:
        fmt.Println("expected 'foo' or 'bar' subcommands")
        os.Exit(1)
    }
}

測試效果

# 編譯
go build subflag.go

./subflag foo -enable -name=joe a1 a2
# 輸出
subcommand 'foo'
  enable: true
  name: joe
  tail: [a1 a2]

./subflag bar -level 8 a1
# 輸出
subcommand 'bar'
  level: 8
  tail: [a1]

./subflag bar -enable a1
# 輸出
flag provided but not defined: -enable
Usage of bar:
  -level int
        level