// os_args.go
package main

import (
    "fmt"
    "os"
    "strings"
)

func main() {
    who := "Alice "
    if len(os.Args) > 1 {
        who += strings.Join(os.Args[1:], " ")
    }
    fmt.Println("Good Morning", who)
}

我們在 IDE 或編輯器中直接運行這個程序輸出：Good Morning Alice

我們在命令行運行 os_args or ./os_args 會得到同樣的結(jié)果。

但是我們在命令行加入?yún)?shù)，像這樣：os_args John Bill Marc Luke，將得到這樣的輸出：Good Morning Alice John Bill Marc Luke

這個命令行參數(shù)會放置在切片 os.Args[] 中（以空格分隔），從索引1開始（os.Args[0] 放的是程序本身的名字，在本例中是 os_args）。函數(shù) strings.Join 以空格為間隔連接這些參數(shù)。

練習(xí) 12.5：hello_who.go

寫一個"Hello World"的變種程序：把人的名字作為程序命令行執(zhí)行的一個參數(shù)，比如： hello_who Evan Michael Laura 那么會輸出Hello Evan Michael Laura!

12.4.2 flag 包

flag 包有一個擴展功能用來解析命令行選項。但是通常被用來替換基本常量，例如，在某些情況下我們希望在命令行給常量一些不一樣的值。（參看 19 章的項目）

在 flag 包中有一個 Flag 被定義成一個含有如下字段的結(jié)構(gòu)體：

type Flag struct {
    Name     string // name as it appears on command line
    Usage    string // help message
    Value    Value  // value as set
    DefValue string // default value (as text); for usage message
}

下面的程序 echo.go 模擬了 Unix 的 echo 功能：

package main

import (
    "flag" // command line option parser
    "os"
)

var NewLine = flag.Bool("n", false, "print newline") // echo -n flag, of type *bool

const (
    Space   = " "
    Newline = "\n"
)

func main() {
    flag.PrintDefaults()
    flag.Parse() // Scans the arg list and sets up flags
    var s string = ""
    for i := 0; i < flag.NArg(); i++ {
        if i > 0 {
            s += " "
            if *NewLine { // -n is parsed, flag becomes true
                s += Newline
            }
        }
        s += flag.Arg(i)
    }
    os.Stdout.WriteString(s)
}

flag.Parse() 掃描參數(shù)列表（或者常量列表）并設(shè)置 flag, flag.Arg(i) 表示第i個參數(shù)。Parse() 之后 flag.Arg(i) 全部可用，flag.Arg(0) 就是第一個真實的 flag，而不是像 os.Args(0) 放置程序的名字。

flag.Narg() 返回參數(shù)的數(shù)量。解析后 flag 或常量就可用了。 flag.Bool() 定義了一個默認(rèn)值是 false 的 flag：當(dāng)在命令行出現(xiàn)了第一個參數(shù)（這里是 "n"），flag 被設(shè)置成 true（NewLine 是 *bool 類型）。flag 被解引用到 *NewLine，所以當(dāng)值是 true 時將添加一個 Newline（"\n"）。

flag.PrintDefaults() 打印 flag 的使用幫助信息，本例中打印的是：

-n=false: print newline

flag.VisitAll(fn func(*Flag)) 是另一個有用的功能：按照字典順序遍歷 flag，并且對每個標(biāo)簽調(diào)用 fn （參考 15.8 章的例子）

當(dāng)在命令行（Windows）中執(zhí)行：echo.exe A B C，將輸出：A B C；執(zhí)行 echo.exe -n A B C，將輸出：

A
B
C

每個字符的輸出都新起一行，每次都在輸出的數(shù)據(jù)前面打印使用幫助信息：-n=false: print newline。

對于 flag.Bool 你可以設(shè)置布爾型 flag 來測試你的代碼，例如定義一個 flag processedFlag:

var processedFlag = flag.Bool("proc", false, "nothing processed yet")

在后面用如下代碼來測試：

if *processedFlag { // found flag -proc
    r = process()
}

要給 flag 定義其它類型，可以使用 flag.Int()，flag.Float64()，flag.String()。

在第 15.8 章你將找到一個具體的例子。

鏈接

上一篇：?# 11.8 第二個例子：讀和寫下一篇：?# 11 接口（Interfaces）與反射（reflection）