雖然現(xiàn)在我們使用的大多數(shù)軟件都是可視化的，很容易上手，但是這并不代表 CLI（命令行）應(yīng)用就沒有用武之地了，特別是對于開發(fā)人員來說，還是會經(jīng)常和 CLI 應(yīng)用打交道。而 Golang 就非常適合用來構(gòu)建 CLI 應(yīng)用，下面我們就將來介紹如何在 Golang 中構(gòu)建一個 CLI 應(yīng)用。

對于開發(fā)人員來說平時可能就需要使用到很多 CLI 工具，比如 npm、node、go、python、docker、kubectl 等等，因?yàn)檫@些工具非常小巧、沒有依賴性、非常適合系統(tǒng)管理或者一些自動化任務(wù)等等。

我們這里選擇使用 Golang 里面非常有名的Cobra庫來進(jìn)行 CLI 工具的開發(fā)。Cobra 是一個功能強(qiáng)大的現(xiàn)代化 CLI 應(yīng)用程序庫，有很多知名的 Go 項目使用 Cobra 進(jìn)行構(gòu)建，比如：Kubernetes、Docker、Hugo 等等

概念

Cobra 是構(gòu)建在命令、參數(shù)和標(biāo)識符之上的：

• Commands表示執(zhí)行動作
• Args就是執(zhí)行參數(shù)
• Flags是這些動作的標(biāo)識符

基本的執(zhí)行命令如下所示：

$ APPNAME Command Args --Flags
# 或者
$ APPNAME Command --Flags Args

比如我們平時使用的一些命令行工具：

git clone URL -bare
go get -u URL
npm install package –save
kubectl get pods -n kube-system -l app=cobra

示例

下面我們來看下 Cobra 的使用，這里我們使用的 go1.13.3 版本，使用 Go Modules 來進(jìn)行包管理，如果對這部分知識點(diǎn)不熟悉的，可以查看前面我們的文章Go Modules 基本使用（視頻）了解。

新建一個名為my-calc的目錄作為項目目錄，然后初始化 modules：

$ mkdir my-calc && cd my-calc
# 如果 go modules 默認(rèn)沒有開啟，需要執(zhí)行 export GO111MODULE=on 開啟
$ go mod init my-calc
go: creating new go.mod: module my-calc

初始化完成后可以看到項目根目錄下面多了一個go.mod的文件，現(xiàn)在我們還沒有安裝cobra庫，執(zhí)行下面的命令進(jìn)行安裝：

# 強(qiáng)烈推薦配置該環(huán)境變量
$ export GOPROXY=https://goproxy.cn
$ go get -u github.com/spf13/cobra/cobra

安裝成功后，現(xiàn)在我們可以使用cobra init命令來初始化 CLI 應(yīng)用的腳手架：

$ cobra init --pkg-name my-calc
Your Cobra applicaton is ready at
/Users/ych/devs/workspace/youdianzhishi/course/my-calc

需要注意的是新版本的 cobra 庫需要提供一個--pkg-name參數(shù)來進(jìn)行初始化，也就是指定上面我們初始化的模塊名稱即可。上面的 init 命令就會創(chuàng)建出一個最基本的 CLI 應(yīng)用項目：

$ tree .
.
├── LICENSE
├── cmd
│   └── root.go
├── go.mod
├── go.sum
└── main.go

1 directory, 5 files

其中main.go是 CLI 應(yīng)用的入口，在main.go里面調(diào)用好了cmd/root.go下面的Execute函數(shù)：

// main.go
package main

import "my-calc/cmd"

func main() {
	cmd.Execute()
}

然后我們再來看下cmd/root.go文件。

rootCmd

root（根）命令是 CLI 工具的最基本的命令，比如對于我們前面使用的go get URL，其中go就是 root 命令，而get就是go這個根命令的子命令，而在root.go中就直接使用了 cobra 命令來初始化rootCmd結(jié)構(gòu)，CLI 中的其他所有命令都將是rootCmd這個根命令的子命令了。

這里我們將cmd/root.go里面的rootCmd變量內(nèi)部的注釋去掉，并在Run函數(shù)里面加上一句fmt.Println("Hello Cobra CLI")：

var rootCmd = &cobra.Command{
	Use:   "my-calc",
	Short: "A brief description of your application",
	Long: `A longer description that spans multiple lines and likely contains
examples and usage of using your application. For example:

Cobra is a CLI library for Go that empowers applications.
This application is a tool to generate the needed files
to quickly create a Cobra application.`,
	Run: func(cmd *cobra.Command, args []string) {
        fmt.Println("Hello Cobra CLI")
    },
}

這個時候我們在項目根目錄下面執(zhí)行如下命令進(jìn)行構(gòu)建：

$ go build -o my-calc

該命令會在項目根目錄下生成一個名為my-calc的二進(jìn)制文件，直接執(zhí)行這個二進(jìn)制文件可以看到如下所示的輸出信息：

$ ./my-calc
Hello Cobra CLI

init

我們知道init函數(shù)是 Golang 中初始化包的時候第一個調(diào)用的函數(shù)。在cmd/root.go中我們可以看到init函數(shù)中調(diào)用了cobra.OnInitialize(initConfig)，也就是每當(dāng)執(zhí)行或者調(diào)用命令的時候，它都會先執(zhí)行init函數(shù)中的所有函數(shù)，然后再執(zhí)行execute方法。該初始化可用于加載配置文件或用于構(gòu)造函數(shù)等等，這完全依賴于我們應(yīng)用的實(shí)際情況。

在初始化函數(shù)里面cobra.OnInitialize(initConfig)調(diào)用了initConfig這個函數(shù)，所有，當(dāng)rootCmd的執(zhí)行方法RUN: func運(yùn)行的時候，rootCmd根命令就會首先運(yùn)行initConfig函數(shù)，當(dāng)所有的初始化函數(shù)執(zhí)行完成后，才會執(zhí)行rootCmd的RUN: func執(zhí)行函數(shù)。

我們可以在initConfig函數(shù)里面添加一些 Debug 信息：

func initConfig() {
    fmt.Println("I'm inside initConfig function in cmd/root.go")
    ...
}

然后同樣重新構(gòu)建一次再執(zhí)行：

$ go build -o my-calc
$ ./my-calc
I'm inside initConfig function in cmd/root.go
Hello Cobra CLI

可以看到是首先運(yùn)行的是initConfig函數(shù)里面的信息，然后才是真正的執(zhí)行函數(shù)里面的內(nèi)容。

為了搞清楚整個 CLI 執(zhí)行的流程，我們在main.go里面也添加一些 Debug 信息：

// cmd/root.go
func init() {
    fmt.Println("I'm inside init function in cmd/root.go")
    cobra.OnInitialize(initConfig)
    ...
}

func initConfig() {
    fmt.Println("I'm inside initConfig function in cmd/root.go")
    ...
}

// main.go
func main() {
     fmt.Println("I'm inside main function in main.go")
     cmd.Execute()
}

然后同樣重新構(gòu)建一次再執(zhí)行：

$ go build -o my-calc
$ ./my-calc
I'm inside init function in cmd/root.go
I'm inside main function in main.go
I'm inside initConfig function in cmd/root.go
Hello Cobra CLI

根據(jù)上面的日志信息我們就可以了解到 CLI 命令的流程了。

init函數(shù)最后處理的就是flags了，Flags就類似于命令的標(biāo)識符，我們可以把他們看成是某種條件操作，在 Cobra 中提供了兩種類型的標(biāo)識符：Persistent Flags和Local Flags。

• Persistent Flags: 該標(biāo)志可用于為其分配的命令以及該命令的所有子命令。
• Local Flags: 該標(biāo)志只能用于分配給它的命令。

initConfig

該函數(shù)主要用于在 home 目錄下面設(shè)置一個名為.my-calc的配置文件，如果該文件存在則會使用這個配置文件。

// cmd/root.go
// initConfig 讀取配置文件和環(huán)境變量
func initConfig() {
	if cfgFile != "" {
        // 使用 flag 標(biāo)志中傳遞的配置文件
		viper.SetConfigFile(cfgFile)
	} else {
		// 獲取 Home 目錄
		home, err := homedir.Dir()
		if err != nil {
			fmt.Println(err)
			os.Exit(1)
		}
		// 在 Home 目錄下面查找名為 ".my-calc" 的配置文件
		viper.AddConfigPath(home)
		viper.SetConfigName(".my-calc")
	}
    // 讀取匹配的環(huán)境變量
	viper.AutomaticEnv()
	// 如果有配置文件，則讀取它
	if err := viper.ReadInConfig(); err == nil {
		fmt.Println("Using config file:", viper.ConfigFileUsed())
	}
}

viper是一個非常優(yōu)秀的用于解決配置文件的 Golang 庫，它可以從 JSON、TOML、YAML、HCL、envfile 以及 Java properties 配置文件中讀取信息，功能非常強(qiáng)大，而且不僅僅是讀取配置這么簡單，了解更多相關(guān)信息可以查看 Git 倉庫相關(guān)介紹：https://github.com/spf13/viper。

現(xiàn)在我們可以去掉前面我們添加的一些打印語句，我們已經(jīng)創(chuàng)建了一個my-calc命令作為rootCmd命令，執(zhí)行該根命令會打印Hello Cobra CLI信息，接下來為我們的 CLI 應(yīng)用添加一些其他的命令。

添加數(shù)據(jù)

在項目根目錄下面創(chuàng)建一個名為add的命令，Cobra添加一個新的命令的方式為：cobra add <commandName>，所以我們這里直接這樣執(zhí)行：

$ cobra add add
add created at /Users/ych/devs/workspace/youdianzhishi/course/my-calc
$ tree .
.
├── LICENSE
├── cmd
│   ├── add.go
│   └── root.go
├── go.mod
├── go.sum
├── main.go
└── my-calc

1 directory, 7 files

現(xiàn)在我們可以看到cmd/root.go文件中新增了一個add.go的文件，我們仔細(xì)觀察可以發(fā)現(xiàn)該文件和cmd/root.go比較類似。首先是聲明了一個名為addCmd的結(jié)構(gòu)體變量，類型為*cobra.Command指針類型，*cobra.Command有一個RUN函數(shù)，帶有*cobra.Command指針和一個字符串切片參數(shù)。

然后在init函數(shù)中進(jìn)行初始化，初始化后，將其添加到rootCmd根命令中rootCmd.AddCommand(addCmd)，所以我們可以把addCmd看成是rootCmd的子命令。

同樣現(xiàn)在重新構(gòu)建應(yīng)用再執(zhí)行:

$ go build -o my-calc
$ ./my-calc
Hello Cobra CLI
$ ./my-calc add
add called

可以看到add命令可以正常運(yùn)行了，接下來我們來讓改命令支持添加一些數(shù)字，我們知道在RUN函數(shù)中是用戶字符串 slice 來作為參數(shù)的，所以要支持添加數(shù)字，我們首先需要將字符串轉(zhuǎn)換為 int 類型，返回返回計算結(jié)果。在cmd/add.go文件中添加一個名為intAdd的函數(shù)，定義如下所示：

// cmd/add.go
func intAdd(args []string) {
	var sum int
	// 循環(huán) args 參數(shù)，循環(huán)的第一個值為 args 的索引，這里我們不需要，所以用 _ 忽略掉
	for _, ival := range args {
		// 將 string 轉(zhuǎn)換成 int 類型
		temp, err := strconv.Atoi(ival)
		if err != nil {
			panic(err)
		}
		sum = sum + temp
	}
	fmt.Printf("Addition of numbers %s is %d\n", args, sum)
}

然后在addCmd變量中，更新RUN函數(shù)，移除默認(rèn)的打印信息，調(diào)用上面聲明的addInt函數(shù)：

// addCmd
Run: func(cmd *cobra.Command, args []string) {
    intAdd(args)
},

然后重新構(gòu)建應(yīng)用執(zhí)行如下所示的命令：

$ go build -o my-calc
$ ./my-calc
Hello Cobra CLI
# 注意參數(shù)之間的空格
$ ./my-calc add 1 2 3
Addition of numbers [1 2 3] is 6

由于RUN函數(shù)中的args參數(shù)是一個字符串切片，所以我們可以傳遞任意數(shù)量的參數(shù)，但是確有一個缺陷，就是只能進(jìn)行整數(shù)計算，不能計算小數(shù)，比如我們執(zhí)行如下的計算就會直接 panic 了:

$ ./my-calc add 1 2 3.5
panic: strconv.Atoi: parsing "3.5": invalid syntax

goroutine 1 [running]:
my-calc/cmd.intAdd(0xc0000a5890, 0x3, 0x3)
......

因?yàn)樵?code>intAdd函數(shù)里面，我們只是將字符串轉(zhuǎn)換成了 int，而不是 float32/64 類型，所以我們可以為addCmd命令添加一個flag標(biāo)識符，通過該標(biāo)識符來幫助 CLI 確定它是 int 計算還是 float 計算。

在cmd/add.go文件的init函數(shù)內(nèi)部，我們創(chuàng)建一個 Bool 類型的本地標(biāo)識符，命名成float，簡寫成f，默認(rèn)值為 false。這個默認(rèn)值是非常重要的，意思就是即使沒有在命令行中調(diào)用 flag 標(biāo)識符，該標(biāo)識符的值就將為 false。

// cmd/add.go
func init() {
	rootCmd.AddCommand(addCmd)
	addCmd.Flags().BoolP("float", "f", false, "Add Floating Numbers")
}

然后創(chuàng)建一個floatAdd的函數(shù)：

func floatAdd(args []string) {
	var sum float64
	for _, fval := range args {
		// 將字符串轉(zhuǎn)換成 float64 類型
		temp, err := strconv.ParseFloat(fval, 64)
		if err != nil {
			panic(err)
		}
		sum = sum + temp
	}
	fmt.Printf("Sum of floating numbers %s is %f\n", args, sum)
}

該函數(shù)和上面的intAdd函數(shù)幾乎是相同的，除了是將字符串轉(zhuǎn)換成 float64 類型。然后在addCmd的RUN函數(shù)中，我們根據(jù)傳入的標(biāo)識符來判斷到底應(yīng)該是調(diào)用intAdd還是floatAdd，如果傳遞了--float或者-f標(biāo)志，就將會調(diào)用floatAdd函數(shù)。

// cmd/add.go
// addCmd
Run: func(cmd *cobra.Command, args []string) {
    // 獲取 float 標(biāo)識符的值，默認(rèn)為 false
    fstatus, _ := cmd.Flags().GetBool("float")
    if fstatus { // 如果為 true，則調(diào)用 floatAdd 函數(shù)
        floatAdd(args)
    } else {
        intAdd(args)
    }
},

現(xiàn)在重新編譯構(gòu)建 CLI 應(yīng)用，按照如下方式執(zhí)行：

$ go build -o my-calc
$ ./my-calc add 1 2 3
Addition of numbers [1 2 3] is 6
$ ./my-calc add 1 2 3.5 -f
Sum of floating numbers [1 2 3.5] is 6.500000
$./my-calc add 1 2 3.5 --float
Sum of floating numbers [1 2 3.5] is 6.500000

然后接下來我們在給addCmd添加一些子命令來擴(kuò)展它。

添加偶數(shù)

同樣在項目根目錄下執(zhí)行如下命令添加一個名為even的命令：

$ cobra add even
even created at /Users/ych/devs/workspace/youdianzhishi/course/my-calc

和上面一樣會在root目錄下面新增一個名為even.go的文件，修改該文件中的init函數(shù)，將rootCmd修改為addCmd，因?yàn)槲覀兪菫?code>addCmd添加子命令:

// cmd/even.go
func init() {
	addCmd.AddCommand(evenCmd)
}

然后更新evenCmd結(jié)構(gòu)體參數(shù)的RUN函數(shù):

// cmd/even.go
Run: func(cmd *cobra.Command, args []string) {
    var evenSum int
    for _, ival := range args {
        temp, _ := strconv.Atoi(ival)
        if temp%2 == 0 {
            evenSum = evenSum + temp
        }
    }
    fmt.Printf("The even addition of %s is %d\n", args, evenSum)
},

首先將字符串轉(zhuǎn)換成整數(shù)，然后判斷如果是偶數(shù)才進(jìn)行累加。然后重新編譯構(gòu)建應(yīng)用:

$ go build -o my-calc
$ ./my-calc add even 1 2 3 4 5 6
The even addition of [1 2 3 4 5 6] is 12

my-calc是我們的根命令，add是rootCmd的子命令，even優(yōu)勢addCmd的子命令，所以按照上面的方式調(diào)用。可以用同樣的方式再去添加一個奇數(shù)相加的子命令。

到這里我們就在 Golang 里面使用Cobra創(chuàng)建了一個簡單的 CLI 應(yīng)用。本文的內(nèi)容雖然比較簡單，但是是我們了解學(xué)習(xí)Cobra基礎(chǔ)的一個很好的入門方式，后續(xù)我們也可以嘗試添加一些更加復(fù)雜的使用案例。

Reference:

https://www.qikqiak.com/post/create-cli-app-with-cobra在 Golang 中使用 Cobra 創(chuàng)建 CLI 應(yīng)用

https://cobra.dev cobra docs

到此這篇關(guān)于在 Golang 中使用 Cobra 創(chuàng)建 CLI 應(yīng)用的文章就介紹到這了,更多相關(guān)Golang創(chuàng)建 CLI 應(yīng)用內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論