Go語言 命令行解析（一）

命令行啓動服務的方式，在後端使用很是普遍，若是有寫過C語言的同窗相信不難理解這一點！在C語言中，咱們能夠根據argc和argv來獲取和解析命令行的參數，從而經過不一樣的參數調取不一樣的方法，同時也能夠用Usage來打印幫助信息了。

那麼開始今天的話題以前，咱們回顧一下在C語言中是如何解析傳遞的參數的。

示例代碼：

#include <stdio.h> #include <stdlib.h> // argc 爲int型 // argv 爲char指針數組，元素個數是argc，存放的是指向每個參數的指針 int main(int argc, char *argv[]) { printf("命令行參數個數: %d\n", argc); printf("執行程序的名稱：%s\n", argv[0]); int i = 1; // 從下標1開始獲取，由於0表明的是程序名稱 while(i < argc) { // 循環打印每一個命令行的參數 printf("%s\n", argv[i]); i++; } return 0; }

編譯運行：

#gcc c_cli.c -o c_cli ./c_cli /usr/local/service/config /usr/local/service/log/service.log 命令行參數個數: 3 執行程序的名稱：./c_cli /usr/local/service/config /usr/local/service/log/service.log

看完上面的例子，其實咱們能夠發現，上例中是C語言自帶的參數解析，對於簡單的參數構成仍是可使用的。下面咱們再看一下Go語言os標準庫的實現。

示例代碼：

package main import ( "fmt" "os" ) func main() { var args = os.Args fmt.Println(args) return } 

編譯執行：

#go build go_flag.go #./go_flag /usr/local/service/config /usr/local/service/log/service.log [./go_flag /usr/local/service/config /usr/local/service/log/service.log]

上例中，咱們能夠看到os.Args返回一個數組，數組裏面是咱們命令行執行時，所傳遞的參數和程序名稱。os自帶的參數獲取，對於簡單的參數來講還能使用，若是參數複雜的狀況下，那麼解析起來就比較費勁的。這個時候，咱們能夠選擇Go語言的flag標準庫來幫咱們處理命令行解析工做。

Flag包：https://golang.org/pkg/flag/

是Go語言提供的一個標準庫，可以較爲方便和靈活的解析命令行傳遞的參數。

flag有兩種方式：

一、flag.Type，其中Type能夠是：int、string、bool，float等類型，返回指針類型。

var port = flag.Int("port", 0, "相關描述")

參數1：flag的名稱

參數2：flag的值，上例中默認值是0

參數3：flag的描述

二、flag.TypeVar，將類型綁定到一個變量上。

var port int flag.IntVar(&port, "port", 0, "相關描述")

參數1：flag的值

參數2：flag的名稱

參數3：flag的值，上例中默認值是0

參數4：flag的描述

示例代碼：

package main import ( "flag" "fmt" "os" ) var ( // 定義一個tcp端口號 tcp_port int // flag.Type（Name爲Flag的名字，Value是Flag的值，Usage是Flag的提示信息） port = flag.Int("port", 0, "服務端口設置參數爲：-port=80") config = flag.String("config", "", "配置文件配置參數爲：-config=/usr/local/service/config") logFile = flag.String("logfile", "", "日誌文件配置參數爲：-logfile=/usr/local/service/log/service.log") // flag.Type返回的是指針類型，因此獲取值爲"*變量" ) func init() { flag.IntVar(&tcp_port, "tcp_port", 0, "TCP服務端口描述: -tcp_port=2001") flag.Usage = func() { _, _ = fmt.Fprint(os.Stderr, "cli : go_flag -port=8080 -config=/usr/local/service/config "+ "-logfile=/usr/local/service/log/service.log -tcp_port=2001\n") flag.PrintDefaults() } flag.Parse() } func main() { if *port <= 0 { flag.PrintDefaults() os.Exit(1) } if tcp_port <= 0 { flag.PrintDefaults() os.Exit(1) } if *config == "" { flag.PrintDefaults() os.Exit(2) } if *logFile == "" { flag.PrintDefaults() os.Exit(3) } fmt.Printf("service tcp port : %d \n", tcp_port) fmt.Printf("service port : %d \n", *port) fmt.Printf("service config : %s \n", *config) fmt.Printf("service logfile : %s \n", *logFile) return } 

運行結果：

#go build go_flag.go #./go_flag -config string 配置文件配置參數爲：-config=/usr/local/service/config -logfile string 日誌文件配置參數爲：-logfile=/usr/local/service/log/service.log -port int 服務端口設置參數爲：-port=80 -tcp_port int TCP服務端口描述: -tcp_port=2001 #./go_flag -port 8090 -config=/usr/local/service/config -logfile= /usr/local/service/log/service.log -tcp_port=2001 service tcp port : 2001 service port : 8090 service config : /usr/local/service/config service logfile : /usr/local/service/log/service.log 

查看幫助：

#./go_flag -h cli : go_flag -port=8080 -config=/usr/local/service/config -logfile=/usr/local/service/log/service.log -config string 配置文件配置參數爲：-config=/usr/local/service/config -logfile string 日誌文件配置參數爲：-logfile=/usr/local/service/log/service.log -port int 服務端口設置參數爲：-port=80 -tcp_port int TCP服務端口描述: -tcp_port=2001

Flag語法：

• -flag 只支持bool參數
• -flag=p 
• -flag p bool類型不能使用，當p爲false時會引發歧義

Flag解析：

parseOne()函數來處理-flag=value的，若是處理成功後，會將flag存儲到FlagSet.actucal map[string]*Flag中。

flag.Parse()函數來解析命令行中的參數中定義的flag，該方法遇到第一個非flag的命令方法就停止。

一、non-flag的時候，會終止解析工做。

​

二、連續兩個「-」的時候，會終止解析工做。
​

而後，再看咱們傳遞錯誤的參數的返回狀況。

#./go_flag -a=b flag provided but not defined: -a cli : go_flag -port=8080 -config=/usr/local/service/config -logfile=/usr/local/service/log/service.log -config string 配置文件配置參數爲：-config=/usr/local/service/config -logfile string 日誌文件配置參數爲：-logfile=/usr/local/service/log/service.log -port int 服務端口設置參數爲：-port=80 -tcp_port int TCP服務端口描述: -tcp_port=2001

跟蹤到代碼中能夠看到如圖：

​

從上例可見，其實flag並無很是強大，下一章節會介紹一個更強大的命令行解析工具。

總結：

1. os標準庫提供的解析方法，可以解析簡單的命令行參數。
2. flag可以解析約定好的常規按照-flag傳遞的命令行參數，有幫助信息。
3. os和flag還不可以解析複雜結構的啓動參數。