+ 收藏我们

网站模板

网站模板搜索
404模板 营销型模板 外贸网站模板 单页模板 双语模板 标签大全
电话:18630701785
首页 > 站长学院 > 如何让每个 cobra 命令解析自己的标志?获取初始化循环(预期,但如何避免?) >

如何让每个 cobra 命令解析自己的标志?获取初始化循环(预期,但如何避免?)

时间:2024-04-06 14:45:44

本篇文章给大家分享《如何让每个 cobra 命令解析自己的标志?获取初始化循环(预期,但如何避免?)》,覆盖了golang的常见基础知识,其实一个语言的全部知识点一篇文章是不可能说完的,但希望通过这些问题,让读者对自己的掌握程度有一定的认识(B 数),从而弥补自己的不足,更好的掌握它。

问题内容
我正在遵循 https://GitHub.com/spf13/cobra#flags 上的指南,但我对那里的一些内容感到困惑。

我有一些服务(rest api、电子邮件服务、事件),我正在尝试做这样的事情:

Go 运行 *.go rest -env dev -p 3000

go 运行 *.go events -env dev -p 3001

我正在关注 github 页面,因此我已经这样定义了 rootcmd 和 restcmd:

var rootcmd = &cobra.command{
    use: "myappname",
}

var restcmd = &cobra.command{
    use:   "rest",
    short: "rest api",
    long:  "rest api",
    run:   runrest,
}
在 runrest 方法中,我应该做类似的事情

var env string
restCmd.Flags().StringVarP(&env, "env", "env", "", "environment")
请告诉我。

谢谢


解决方案

每个子命令都可以有自己的标志。您可以按以下方式执行此操作:

package main

import (
    "fmt"
    "log"

    "github.com/spf13/cobra"
)

var rootCmd = &cobra.Command{
    Use: "app",
}

func NewCmdRest() *cobra.Command {
    var env string
    var restCmd = &cobra.Command{
        Use: "rest",
        Run: func(cmd *cobra.Command, args []string) {
            fmt.Println("rest:", env)
        },
    }

    restCmd.Flags().StringVarP(&env, "env", "e", "", "environment")
    return restCmd
}

func NewCmdEvent() *cobra.Command {
    var env string
    var eventCmd = &cobra.Command{
        Use: "event",
        Run: func(cmd *cobra.Command, args []string) {
            fmt.Println("event:", env)
        },
    }

    eventCmd.Flags().StringVarP(&env, "env", "e", "", "environment")
    return eventCmd
}

func init() {
    rootCmd.AddCommand(NewCmdRest())
    rootCmd.AddCommand(NewCmdEvent())
}

func main() {
    if err := rootCmd.Execute(); err != nil {
        log.Fatal(err)
    }
}
好了,本文到此结束,带大家了解了《如何让每个 cobra 命令解析自己的标志?获取初始化循环(预期,但如何避免?)》,希望本文对你有所帮助!

有问题可以加入网站技术QQ群一起交流学习

本站会员学习、解决问题QQ群(691961965)

客服微信号:lpf010888

Title