前言
對于一名初學者來說，想要盡快熟悉 Go 語言特性，所以以操作式的學習方法為主，比如編寫一個簡單的數學計算器，讀取命令行參數，進行數學運算。
本文講述使用三種方式講述 Go 語言如何接受命令行參數，并完成一個簡單的數學計算，為演示方便，最后的命令行結果大概是這樣的：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
# input ./calc add 1 2# output3# input./calc sub 1 2# out-1# input./calc mul 10 20# out200 |
使用的三種方式是：
- 內置 os 包讀取命令參數
- 內置 flag 包讀取命令參數
- cli 框架讀取命令參數
0. 已有歷史經驗
如果你熟悉 Python 、Shell 腳本，你可以比較下：
Python
|
1
2
3
4
5
6
7
|
import sysargs = sys.argv# args 是一個列表# 第一個值表示的是 文件名# 除第一個之外，其他的值是接受的參數 |
Shell
|
1
2
3
4
5
6
7
8
9
10
11
12
|
if [ $# -ne 2 ]; then echo "Usage: $0 param1 pram2" exit 1finame=$1age=$2echo $nameecho $age# `$0` 表示文件名# `$1` 表示第一個參數# `$2` 表示第二個參數 |
能看出一些共性，接收參數，一般解析出來都是一個數組（列表、切片）， 第一個元素表示的是文件名，剩余的參數表示接收的參數。
好，那么為了實現 “簡單數學計算” 這個功能，讀取命令行參數：比如 ./calc add 1 2
除文件名之外的第一個元素：解析為 進行數學運算的 操作，比如： add、sub、mul、sqrt
其余參數表示：進行操作的數值
注意：命令行讀取的參數一般為字符串，進行數值計算需要進行數據類型轉換
大概思路就是這樣。
1. OS 獲取命令行參數
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
os.Args# 為接受的參數，是一個切片strconv.Atoi # 將字符串數值轉換為整型strconv.Itoa# 將整型轉換為字符串strconv.ParseFloat# 將字符串數值轉換為浮點型 |
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
var help = func () { fmt.Println("Usage for calc tool.") fmt.Println("====================================================") fmt.Println("add 1 2, return 3") fmt.Println("sub 1 2, return -1") fmt.Println("mul 1 2, return 2") fmt.Println("sqrt 2, return 1.4142135623730951")}func CalcByOs() error { args := os.Args if len(args) < 3 || args == nil { help() return nil } operate := args[1] switch operate { case "add":{ rt := 0 number_one, err1 := strconv.Atoi(args[2]) number_two, err2 := strconv.Atoi(args[3]) if err1 == nil && err2 == nil { rt = number_one + number_two fmt.Println("Result ", rt) } } case "sub": { rt := 0 number_one, err1 := strconv.Atoi(args[2]) number_two, err2 := strconv.Atoi(args[3]) if err1 == nil && err2 == nil { rt += number_one - number_two fmt.Println("Result ", rt) } } case "mul": { rt := 1 number_one, err1 := strconv.Atoi(args[2]) number_two, err2 := strconv.Atoi(args[3]) if err1 == nil && err2 == nil { rt = number_one * number_two fmt.Println("Result ", rt) } } case "sqrt": { rt := float64(0) if len(args) != 3 { fmt.Println("Usage: sqrt 2, return 1.4142135623730951") return nil } number_one, err := strconv.ParseFloat(args[2], 64) if err == nil { rt = math.Sqrt(number_one) fmt.Println("Result ", rt) } } default: help() } return nil} |
最后的效果大概是：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
./calc add 1 2Result 3====================./calc sub 1 2Result -1====================./calc mul 10 20Result 200===================./calc sqrt 2Result 1.4142135623730951 |
2. flag 獲取命令行參數
flag 包比 os 讀取參數更方便。可以自定義傳入的參數的類型：比如字符串，整型，浮點型，默認參數設置等
基本的使用方法如下：
|
1
2
3
|
var operate stringflag.StringVar(&operate,"o", "add", "operation for calc") |
# 解釋
綁定 operate 變量， name="o", value="add" , usage="operation for calc"
也可以這樣定義為指針變量
|
1
|
var operate := flag.String("o", "add", "operation for calc") |
同時還可以自定義 flag 類型
所有變量注冊之后，調用 flag.Parse() 來解析命令行參數， 如果是綁定變量的方式，直接使用變量進行操作，
如果使用指針變量型，需要 *operate 這樣使用。
flag.Args() 表示接收的所有命令行參數集， 也是一個切片
|
1
2
3
|
for index, value := range flag.Args { fmt.Println(index, value)} |
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
func CalcByFlag() error { var operation string var numberone float64 var numbertwo float64 flag.StringVar(&operation, "o", "add", "operation for this tool") flag.Float64Var(&numberone, "n1", 0, "The first number") flag.Float64Var(&numbertwo, "n2", 0, "The second number") flag.Parse() fmt.Println(numberone, numbertwo) if operation == "add" { rt := numberone + numbertwo fmt.Println("Result ", rt) } else if operation == "sub" { rt := numberone - numbertwo fmt.Println("Result ", rt) } else if operation == "mul" { rt := numberone * numbertwo fmt.Println("Result ", rt) } else if operation == "sqrt" { rt := math.Sqrt(numberone) fmt.Println("Result ", rt) } else { help() } return nil} |
最后的結果效果如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
./calc -o add -n1 1 -n2 2Result 3=============================./calc -o sub -n1 2 -n2 3Result -1============================./calc -o mul -n1 10 -n2 20Result 200===========================./calc -o sqrt -n1 2Result 1.4142135623730951 |
3. CLI 框架
cli 是一款業界比較流行的命令行框架。
所以你首先需要安裝：
|
1
|
go get github.com/urfave/cli |
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
# 一個簡單的示例如下：package mainimport ( "fmt" "os" "github.com/urfave/cli")func main() { app := cli.NewApp() app.Name = "boom" app.Usage = "make an explosive entrance" app.Action = func(c *cli.Context) error { fmt.Println("boom! I say!") return nil } app.Run(os.Args)} |
好，為實現 “簡單數學計算” 的功能，我們應該怎么實現呢？
主要是 使用 框架中的 Flag 功能，對參數進行設置
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
app.Flags = []cli.Flag { cli.StringFlag{ Name: "operation, o", Value: "add", Usage: "calc operation", }, cli.Float64Flag{ Name: "numberone, n1", Value: 0, Usage: "number one for operation", }, cli.Float64Flag{ Name: "numbertwo, n2", Value: 0, Usage: "number two for operation", },} |
能看出，我們使用了三個參數：operation、numberone、numbertwo
同時定義了參數的類型，默認值，以及別名（縮寫）
那么在這個框架中如何實現參數的操作呢：主要是重寫app.Action 方法
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
app.Action = func(c *cli.Context) error { operation := c.String("operation") numberone := c.Float64("numberone") numbertwo := c.Float64("numbertwo") //fmt.Println(operation, numberone, numbertwo) if operation == "add" { rt := numberone + numbertwo fmt.Println("Result ", rt) } else if operation == "sub" { rt := numberone - numbertwo fmt.Println("Result ", rt) } else if operation == "mul" { rt := numberone * numbertwo fmt.Println("Result ", rt) } else if operation == "sqrt" { rt := math.Sqrt(numberone) fmt.Println("Result ", rt) } else { help() } return nil}# 對 operation 參數進行判斷，執行的是那種運算，然后編寫相應的運算操作 |
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
func CalcByCli(){ app := cli.NewApp() app.Name = "calc with go" app.Usage = "calc tool operate by go" app.Version = "0.1.0" app.Flags = [] cli.Flag { cli.StringFlag{ Name: "operation, o", Value: "add", Usage: "calc operation", }, cli.Float64Flag{ Name: "numberone, n1", Value: 0, Usage: "number one for operation", }, cli.Float64Flag{ Name: "numbertwo, n2", Value: 0, Usage: "number two for operation", }, } app.Action = func(c *cli.Context) error { operation := c.String("operation") numberone := c.Float64("numberone") numbertwo := c.Float64("numbertwo") //fmt.Println(operation, numberone, numbertwo) if operation == "add" { rt := numberone + numbertwo fmt.Println("Result ", rt) } else if operation == "sub" { rt := numberone - numbertwo fmt.Println("Result ", rt) } else if operation == "mul" { rt := numberone * numbertwo fmt.Println("Result ", rt) } else if operation == "sqrt" { rt := math.Sqrt(numberone) fmt.Println("Result ", rt) } else { help() } return nil } app.Run(os.Args)} |
調用這個函數的最終效果如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
./calc -o add --n1 12 --n2 12Result 24===================================./calc -o sub --n1 100 --n2 200Result -100===================================./calc -o mul --n1 10 --n2 20Result 200===================================./calc -o sqrt --n1 2Result 1.4142135623730951 |
4 其他
知道如何讀取命令行參數，就可以實現一些更有意思的事。
比如網上有許多免費的 API 接口，比如查詢天氣，查詢農歷的API 接口。
還有一些查詢接口，比如有道云翻譯接口，你可以實現翻譯的功能。
或者扇貝的接口，實現查詢單詞的功能。
再比如一些音樂接口，實現音樂信息查詢。
不一一列了。
下面實現一個調用免費的查詢天氣的接口實現命令行查詢天氣。
GO 如何進行 HTTP 訪問？內置的 net/http 可以實現
一個簡易的GET 操作如下：
|
1
2
3
4
5
6
7
8
9
|
func Requests(url string) (string, error) { response, err := http.Get(url) if err != nil { return "", err } defer response.Body.Close() body, _ := ioutil.ReadAll(response.Body) return string(body), nil} |
免費的 API URL 如下：
|
1
|
http://www.sojson.com/open/api/weather/json.shtml?city=北京 |
返回的結果是一個Json 格式的數據
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
{ "status": 200, "data": { "wendu": "29", "ganmao": "各項氣象條件適宜，發生感冒機率較低。但請避免長期處于空調房間中，以防感冒。", "forecast": [ { "fengxiang": "南風", "fengli": "3-4級", "high": "高溫 32℃", "type": "多云", "low": "低溫 17℃", "date": "16日星期二" }, { "fengxiang": "南風", "fengli": "微風級", "high": "高溫 34℃", "type": "晴", "low": "低溫 19℃", "date": "17日星期三" }, { "fengxiang": "南風", "fengli": "微風級", "high": "高溫 35℃", "type": "晴", "low": "低溫 22℃", "date": "18日星期四" }, { "fengxiang": "南風", "fengli": "微風級", "high": "高溫 35℃", "type": "多云", "low": "低溫 22℃", "date": "19日星期五" }, { "fengxiang": "南風", "fengli": "3-4級", "high": "高溫 34℃", "type": "晴", "low": "低溫 21℃", "date": "20日星期六" } ], "yesterday": { "fl": "微風", "fx": "南風", "high": "高溫 28℃", "type": "晴", "low": "低溫 15℃", "date": "15日星期一" }, "aqi": "72", "city": "北京" }, "message": "OK"} |
所以我們的任務就是傳入 “城市” 的名稱，再對返回的 Json 數據解析。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
package mainimport ( "fmt" "os" "encoding/json" "github.com/urfave/cli" "net/http" "io/ioutil" //"github.com/modood/table")type Response struct { Status int `json:"status"` CityName string `json:"city"` Data Data `json:"data"` Date string `json:"date"` Message string `json:"message"` Count int `json:"count"`}type Data struct { ShiDu string `json:"shidu"` Quality string `json:"quality"` Ganmao string `json:"ganmao"` Yesterday Day `json:"yesterday"` Forecast []Day `json:"forecast"`}type Day struct { Date string `json:"date"` Sunrise string `json:"sunrise"` High string `json:"high"` Low string `json:"low"` Sunset string `json:"sunset"` Aqi float32 `json:"aqi"` Fx string `json:"fx"` Fl string `json:"fl"` Type string `json:"type"` Notice string `json:"notice"`}func main() { const apiURL = "http://www.sojson.com/open/api/weather/json.shtml?city=" app := cli.NewApp() app.Name = "weather-cli" app.Usage = "天氣預報小程序" app.Flags = []cli.Flag{ cli.StringFlag{ Name: "city, c", Value: "上海", Usage: "城市中文名", }, cli.StringFlag{ Name: "day, d", Value: "今天", Usage: "可選: 今天, 昨天, 預測", }, cli.StringFlag{ Name: "Author, r", Value: "xiewei", Usage: "Author name", }, } app.Action = func(c *cli.Context) error { city := c.String("city") day := c.String("day") var body, err = Requests(apiURL + city) if err != nil { fmt.Printf("err was %v", err) return nil } var r Response err = json.Unmarshal([]byte(body), &r) if err != nil { fmt.Printf("\nError message: %v", err) return nil } if r.Status != 200 { fmt.Printf("獲取天氣API出現錯誤, %s", r.Message) return nil } Print(day, r) return nil } app.Run(os.Args)}func Print(day string, r Response) { fmt.Println("城市:", r.CityName) if day == "今天" { fmt.Println("濕度:", r.Data.ShiDu) fmt.Println("空氣質量:", r.Data.Quality) fmt.Println("溫馨提示:", r.Data.Ganmao) } else if day == "昨天" { fmt.Println("日期:", r.Data.Yesterday.Date) fmt.Println("溫度:", r.Data.Yesterday.Low, r.Data.Yesterday.High) fmt.Println("風量:", r.Data.Yesterday.Fx, r.Data.Yesterday.Fl) fmt.Println("天氣:", r.Data.Yesterday.Type) fmt.Println("溫馨提示:", r.Data.Yesterday.Notice) } else if day == "預測" { fmt.Println("====================================") for _, item := range r.Data.Forecast { fmt.Println("日期:", item.Date) fmt.Println("溫度:", item.Low, item.High) fmt.Println("風量:", item.Fx, item.Fl) fmt.Println("天氣:", item.Type) fmt.Println("溫馨提示:", item.Notice) fmt.Println("====================================") } } else { fmt.Println("...") }}func Requests(url string) (string, error) { response, err := http.Get(url) if err != nil { return "", err } defer response.Body.Close() body, _ := ioutil.ReadAll(response.Body) return string(body), nil} |
最終的效果大概如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
./weather -c 上海城市: 上海濕度: 80%空氣質量: 輕度污染溫馨提示: 兒童、老年人及心臟、呼吸系統疾病患者人群應減少長時間或高強度戶外鍛煉================================./weaather -c 上海 -d 昨天城市: 上海日期: 28日星期二溫度: 低溫 12.0℃ 高溫 19.0℃風量: 西南風 <3級天氣: 小雨溫馨提示: 霧蒙蒙的雨天，最喜歡一個人聽音樂 |
總結
以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對服務器之家的支持。
原文鏈接：http://www.jianshu.com/p/548adff0d10d
