Go cobra庫(kù)使用教程

更新時(shí)間：2022年12月14日 15:20:23 作者：kina100

cobra既是一個(gè)用于創(chuàng)建強(qiáng)大現(xiàn)代CLI應(yīng)用程序的庫(kù)，也是一個(gè)生成應(yīng)用程序和命令文件的程序。cobra被用在很多go語(yǔ)言的項(xiàng)目中，比如 Kubernetes、Docker、Istio、ETCD、Hugo、Github CLI等等

一，安裝

$ go get -u github.com/spf13/cobra

二，簡(jiǎn)單的cobra項(xiàng)目

先構(gòu)建一個(gè)簡(jiǎn)單的cobra命令，以此來(lái)熟悉cobra是怎么用的

創(chuàng)建一個(gè)cmd目錄，在目錄里創(chuàng)建一個(gè)myOrder.go，實(shí)際上這個(gè)名字可以隨便命名，一般來(lái)說(shuō)，你創(chuàng)建的啥命令就寫啥名字，這不是為了好找么。

package cmd
import (
	"fmt"
	"github.com/spf13/cobra"
)
var rootCmd = cobra.Command{
	// 命令的名稱
	Use: "myOrder",
	// 剪短介紹
	Short: "這是一個(gè)我的私有命令",
	// 詳細(xì)介紹
	Long: "自定義私有命令，主要是想測(cè)試下這玩意能不能用",
	// 普通運(yùn)行，如果需要返回error要使用RUNE
	Run: func(cmd *cobra.Command, args []string) {
		fmt.Println("這是一條myOrder命令的反饋，反正啥也沒處理")
	},
}
func Execute() {
	if err := rootCmd.Execute(); err != nil {
		fmt.Println("這里有錯(cuò)誤", err)
	}
}

然后在main.go 入口里運(yùn)行Execute即可

main.go ↓

package main
import "cobraTest/cmd"
func main() {
	cmd.Execute()
}

cobraTest是這個(gè)項(xiàng)目的名字，cmd是剛才創(chuàng)建的那個(gè)包（目錄）。

此時(shí)是不能直接像往常一樣go run的，要先構(gòu)建為可執(zhí)行文件，在windows上就是exe

go build

go build之后會(huì)發(fā)現(xiàn)目錄里多出了一個(gè)可執(zhí)行文件，

這時(shí)候我們可以進(jìn)行簡(jiǎn)單的測(cè)試

三，測(cè)試

執(zhí)行可執(zhí)行文件，并且?guī)衔覀儎偛诺哪莻€(gè)命令

.\cobraTest.exe myOrder

這就是剛才寫的那個(gè)RUN 里面的運(yùn)行結(jié)果

四，增加flag參數(shù)

package cmd
import (
	"fmt"
	"github.com/spf13/cobra"
)
var rootCmd = cobra.Command{
	// 命令的名稱
	Use: "myOrder",
	// 剪短介紹
	Short: "這是一個(gè)我的私有命令",
	// 詳細(xì)介紹
	Long: "自定義私有命令，主要是想測(cè)試下這玩意能不能用",
	// 普通運(yùn)行，如果需要返回error要使用RUNE
	Run: func(cmd *cobra.Command, args []string) {
		fmt.Println("這是一條myOrder命令的反饋，反正啥也沒處理")
        // 讀取第一個(gè)flag 參數(shù)
		getString, _ := cmd.Flags().GetString("hahaha")
		fmt.Println(getString)
        // 讀取第二個(gè)flag 參數(shù)
		getString, _ = cmd.Flags().GetString("heiheihei")
		fmt.Println(getString)
	},
}
func Execute() {
    // 增加一個(gè)flag 參數(shù)
    // flag參數(shù)的名稱，示例值，flag參數(shù)解釋
	rootCmd.PersistentFlags().String("hahaha", "", "隨便寫點(diǎn)啥吧")
    // 再增加一個(gè)flag 參數(shù)
	rootCmd.Flags().String("heiheihei", "", "輸入一些東西")
	if err := rootCmd.Execute(); err != nil {
		fmt.Println("這里有錯(cuò)誤", err)
	}
}

在設(shè)定完rootCmd后，就可以直接給rootCmd結(jié)構(gòu)體增加flag參數(shù)，并在run里面讀取出來(lái)，修改代碼后要重新go build,之后執(zhí)行命令的時(shí)候，加上flag參數(shù)

.\cobraTest.exe myOrder --hahaha 哈哈哈 --heiheihei 嘿嘿嘿

五，額外命令

有的文章說(shuō)這是子命令，其實(shí)是不恰當(dāng)?shù)模@是為了讓一個(gè)工程可以有多條命令，而無(wú)需生成多個(gè)可執(zhí)行文件。

構(gòu)建兩個(gè)命令，并在main.go里面寫一個(gè)根命令

firstOrder.go ↓

package cmd
import (
	"fmt"
	"github.com/spf13/cobra"
)
// 大寫可用于外部訪問
var FirstCmd = cobra.Command{
	// 命令的名稱
	Use: "firstOrder",
	// 剪短介紹
	Short: "1號(hào)命令",
	// 普通運(yùn)行，如果需要返回error要使用RUNE
	Run: func(cmd *cobra.Command, args []string) {
		fmt.Println("1號(hào)命令可以執(zhí)行")
	},
}

secondOrder.go ↓

package cmd
import (
	"fmt"
	"github.com/spf13/cobra"
)
var SecondCmd = cobra.Command{
	// 命令的名稱
	Use: "secondOrder",
	// 剪短介紹
	Short: "2號(hào)命令",
	// 普通運(yùn)行，如果需要返回error要使用RUNE
	Run: func(cmd *cobra.Command, args []string) {
		fmt.Println("2號(hào)命令可以執(zhí)行")
	},
}

main.go ↓

package main
import (
	"cobraTest/cmd"
	"fmt"
	"github.com/spf13/cobra"
)
func main() {
	rootCmd := cobra.Command{
		Use:              "rootOrder",
		Short:            "根命令",
		TraverseChildren: true,
	}
	// 把這兩條命令加入到根命令里面
	rootCmd.AddCommand(&cmd.FirstCmd)
	rootCmd.AddCommand(&cmd.SecondCmd)
	if err := rootCmd.Execute(); err != nil {
		fmt.Println("Could not run command")
	}
}

執(zhí)行g(shù)o build