快捷導(dǎo)航

Go語言中使用urfave/cli命令行框架

更新時間：2022年07月19日 08:54:59 作者：奮斗的大橙子

這篇文章介紹了Go語言中使用urfave/cli命令行框架的方法，文中通過示例代碼介紹的非常詳細。對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下

命令行參數(shù)處理以及urfave/cli使用

1.通過Os.Args獲取cli的參數(shù)

package main
 
import (
   "fmt"
   "os"
)
 
func main() {
 
   if(len(os.Args) > 1) {
      for _,value := range os.Args{
         fmt.Printf( "%v\n",value)
      }
      return
   }
   fmt.Printf( "%s","沒輸入?yún)?shù)")
}

2. 增加命令行選項

我們經(jīng)常能看到使用一些命令行會有很多選項。例如 ls -l 等

可以使用flag包來獲取選項，例如下面的代碼：

package main
 
import (
    "flag"
    "fmt"
)
 
var recusive bool
var test string
var level int
 
func init() {
    flag.BoolVar(&recusive, "r", false, "Recusive xxxx")
    flag.StringVar(&test, "t", "Default String", "string option")
    flag.IntVar(&level, "l", 1, "level of xxxx")
    flag.Parse()
}
 
func main() {
    fmt.Println("recusive:", recusive)
    fmt.Println("test:", test)
    fmt.Println("level:", level)
}

init函數(shù)內(nèi)部使用了flag包中的BoolVar、StringVar以及IntVar等方法，標(biāo)記了命令的選項。

// StringVar defines a string flag with specified name, default value, and usage string.
// The argument p points to a string variable in which to store the value of the flag.
func StringVar(p *string, name string, value string, usage string) {
    CommandLine.Var(newStringValue(value, p), name, usage)
}

像源碼中描述那樣，第一個參數(shù)用來接收輸入的參數(shù)值，第二個用來定義參數(shù)名稱（-l -r 等），第三個是默認參數(shù)、第四個是使用方法。

于是像上面的代碼我們就可以這樣使用：這里BoolVar的默認值是false， -r后面不增加其他參數(shù)，不用 -r true 這樣。

3.urfave/cli的簡單使用

urfave/cli是一個命令行的框架。舉例說明：

package main
 
import (
    "fmt"
    "os"
 
    "github.com/urfave/cli" //必須使用這個包
)
 
func main() {
 
    //定義兩個變量用于接收控制臺輸入的值
    var stringValue string
    var boolValue bool
 
    //new一個app出來，就是我們的命令行程序
    app := cli.NewApp()
    app.Name = "TestCliApp" //起個名稱
    app.Usage = "Test"      //描述一下用途
    app.Version = "2.0.0"   //設(shè)置一下版本號
    //重點可以設(shè)置一些選項操作
    //第一個是一個字符串的選項，第二個是一個布爾的選項
    app.Flags = []cli.Flag{
        cli.StringFlag{
            Name:        "StringOption,s",
            Value:       "DefaultValue",
            Usage:       "Display a string value",
            Destination: &stringValue,
        },
        cli.BoolFlag{
            Name:        "BoolOption,b",
            Usage:       "Display a bool value",
            Destination: &boolValue,
        },
    }
     
    //定義我們命令行程序主要的工作
    app.Action = func(c *cli.Context) error {
 
        if c.NArg() > 0 {
            cmds := c.Args()
            for index, v := range cmds {
                fmt.Printf("args[%d]=%v\t", index, v)
            }
        } else {
            fmt.Println("No Args")
        }
 
        fmt.Println("stringOption", stringValue)
        fmt.Println("boolOption", boolValue)
        return nil
    }
 
    //執(zhí)行程序
    app.Run(os.Args)
}

有幾個點：

1.定義Flag的時候，Name可以用【,】分割，前面長的可以用 --XXXX 來增加參數(shù)，后面則是短名 -x 來表示
2.默認會有help和version兩個選項
3.我在使用的時候發(fā)現(xiàn)，當(dāng)輸入正常的 -b -s Hello 這樣的選項的時候，C.NArg()返回的值是0，而輸出錯誤的參數(shù)列表的時候，則是有值存在的。

我們可以用 -h （--help）查看一下生成的文檔