Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > Golang urfave/cli库使用

Golang urfave/cli库简单应用示例详解

作者：EricLee

这篇文章主要为大家介绍了Golang urfave/cli库简单应用示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪

引言

通过应用cli库可以在程序中像执行cmd命令一样运行可执行文件，同时也可以通过--help执行的帮助说明。

最简单的应用

package main
import (
    "fmt"
    "log"
    "os"
    "github.com/urfave/cli/v2"
)
func main() {
    app := &cli.App{
        Name:  "greet",
        Usage: "fight the loneliness!",
        Action: func(*cli.Context) error {
            fmt.Println("Hello friend!")
            return nil
        },
    }
    if err := app.Run(os.Args); err != nil {
        log.Fatal(err)
    }
}

上面代码编译后，会生成对应的可执行文件（Name与App的参数Name并没有强制关联，但最好保持一致）。在无参数执行可执行文件时，会执行对应app的Action。上述代码就会打印出Hello friend

cli最主要的几个参数是：Arguments、Flag、Command。这三个就像linux命令（或cmd命令）一样，给同一个应用程序通过传入不同的参数来执行不同的逻辑。

Args

app := &cli.App{
        Action: func(cCtx *cli.Context) error {
            fmt.Printf("Hello %q", cCtx.Args().Get(0))
            return nil
        },
    }

通过cCTX.Args().Get(参数索引)获取具体的参数。

Flag

package main
import (
    "fmt"
    "log"
    "os"
    "github.com/urfave/cli/v2"
)
func main() {
    app := &cli.App{
        Flags: []cli.Flag{
            &cli.StringFlag{
                Name:  "lang",
                Value: "english",
                Usage: "language for the greeting",
            },
        },
        Action: func(cCtx *cli.Context) error {
            name := "Nefertiti"
            if cCtx.NArg() > 0 {
                name = cCtx.Args().Get(0)
            }
            if cCtx.String("lang") == "spanish" {
                fmt.Println("Hola", name)
            } else {
                fmt.Println("Hello", name)
            }
            return nil
        },
    }
    if err := app.Run(os.Args); err != nil {
        log.Fatal(err)
    }
}

对于上述代码执行编译和安装（之后的代码修改后都需要重新执行，后面省略此步骤，"cliTest/greet"为项目目录，细节自查go install命令的使用）

go install cliTest/greet

上述文件执行完成之后会生成可执行文件（windows注意文件所在目录加入到环境变量path中）

执行 greet --help

显示如下：

NAME:                                                             
   greet - A new cli application                                  
                                                                  
USAGE:                                                            
   greet [global options] command [command options] [arguments...]
                                                                  
COMMANDS:                                                         
   help, h  Shows a list of commands or help for one command

GLOBAL OPTIONS:
   --lang value  language for the greeting (default: "english")
   --help, -h    show help

可以看到内容主要包含：

NAME:应用名字及说明

USAGE:应用的用法及格式

COMMANDS：当前应用支持的命令（目前只有help）

GLOBAL OPTIONS: 当前应用支持的可选参数

执行greet --lang spanish eric ，输出Hola eric，因为通过--lang选项我们设置了语言为spanish，在后面的执行中同样根据lang选项执行了不同的逻辑。

执行greet --lang english eric 和greet eric，都输出了Hello eric，这是因为在lang定义的时候设置了默认值Value:"english"。

Flag中各个参数的含义

常用的Flag类型

cli.StringFlag：用于接收字符串值的选项。例如：

goCopy code
cli.StringFlag{
    Name:  "config, c",
    Usage: "Load configuration from `FILE`",
}

cli.IntFlag：用于接收整数值的选项。例如：

goCopy code
cli.IntFlag{
    Name:  "port, p",
    Usage: "Listen on `PORT`",
}

cli.BoolFlag：用于接收布尔值（true 或 false）的选项。通常用于标识性选项。例如：

goCopy code
cli.BoolFlag{
    Name:  "verbose, v",
    Usage: "Enable verbose mode",
}

cli.Float64Flag：用于接收浮点数值的选项。例如：

goCopy code
cli.Float64Flag{
    Name:  "threshold, t",
    Usage: "Set the threshold value",
}

cli.StringSliceFlag：用于接收多个字符串值的选项，返回一个字符串切片。例如：

goCopy code
cli.StringSliceFlag{
    Name:  "tags",
    Usage: "Add one or more tags",
}

cli.GenericFlag：用于自定义类型的选项，需要实现 cli.Generic 接口。可以用于处理非标准选项类型。例如：

goCopy code
cli.GenericFlag{
    Name:  "myflag",
    Value: &MyCustomType{}, // MyCustomType 需要实现 cli.Generic 接口
    Usage: "Custom flag",
}

这些是一些常见的 cli.Flag 类型，您可以根据需要选择适当的类型来定义您的命令行选项。不同的类型允许您接受不同类型的值，并提供了相应的方法来解析和处理这些值。您还可以创建自定义的 cli.Flag 类型来处理特定需求。Flag的类型有很多，具体的参考cli官方文档

Command

像上面的代码，我们本身只存在一个App，同时执行此app会执行器Action方法，如果希望一个App中可以定义多个Action方法，可以引入Command。同时Command还支持定义SubCommand，可以对功能进行更细化的区分。

app := &cli.App{
        Commands: []*cli.Command{
            {
                Name:    "add",
                Aliases: []string{"a"},
                Usage:   "add a task to the list",
                Action: func(cCtx *cli.Context) error {
                    fmt.Println("added task: ", cCtx.Args().First())
                    return nil
                },
            },
            {
                Name:    "complete",
                Aliases: []string{"c"},
                Usage:   "complete a task on the list",
                Action: func(cCtx *cli.Context) error {
                    fmt.Println("completed task: ", cCtx.Args().First())
                    return nil
                },
            },
            {
                Name:    "template",
                Aliases: []string{"t"},
                Usage:   "options for task templates",
                Subcommands: []*cli.Command{
                    {
                        Name:  "add",
                        Usage: "add a new template",
                        Action: func(cCtx *cli.Context) error {
                            fmt.Println("new task template: ", cCtx.Args().First())
                            return nil
                        },
                    },
                    {
                        Name:  "remove",
                        Usage: "remove an existing template",
                        Action: func(cCtx *cli.Context) error {
                            fmt.Println("removed task template: ", cCtx.Args().First())
                            return nil
                        },
                    },
                },
            },
        },
    }
    if err := app.Run(os.Args); err != nil {
        log.Fatal(err)
    }

greet --help显示：

COMMANDS:
   add, a       add a task to the list
   complete, c  complete a task on the list
   template, t  options for task templates
   help, h      Shows a list of commands or help for one command

可以看到当前包含的所有命令。但并没有显示template命令的子命令，我们可以通过greet template --help查看template命令的帮助文档

NAME:
   greet template - options for task templates

USAGE:
   greet template command [command options] [arguments...]

COMMANDS:
   add      add a new template
   remove   remove an existing template
   help, h  Shows a list of commands or help for one command

命令的排序

通过在代码中app.Run之前调用sort.Sort(cli.CommandsByName(app.Commands))可以对命令按名字排序。

命令的分类

同级的命令中可以通过增加Category对命令进行分类（同类命令在展示时会再一个类下显示）。

例如：

Commands: []*cli.Command{
            {
                Name:    "add",
                Aliases: []string{"a"},
                Category: "C1",
                Usage:   "add a task to the list",
                Action: func(cCtx *cli.Context) error {
                    fmt.Println("added task: ", cCtx.Args().First())
                    return nil
                },
            },
            {
                Name:    "complete",
                Aliases: []string{"c"},
                Category: "C1",
                Usage:   "complete a task on the list",
                Action: func(cCtx *cli.Context) error {
                    fmt.Println("completed task: ", cCtx.Args().First())
                    return nil
                },
            },

--help显示

COMMANDS:
   template, t  options for task templates
   help, h      Shows a list of commands or help for one command
   C1:
     add, a       add a task to the list
     complete, c  complete a task on the list

Comand中各个参数的作用

以上就是Golang urfave/cli库简单应用示例详解的详细内容，更多关于Golang urfave/cli库使用的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:
阅读全文