You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
3.1 KiB
Go
74 lines
3.1 KiB
Go
package rpc
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
"github.com/zeromicro/go-zero/tools/goctl/config"
|
|
"github.com/zeromicro/go-zero/tools/goctl/internal/cobrax"
|
|
"github.com/zeromicro/go-zero/tools/goctl/rpc/cli"
|
|
)
|
|
|
|
var (
|
|
// Cmd describes a rpc command.
|
|
Cmd = cobrax.NewCommand("rpc", cobrax.WithRunE(func(command *cobra.Command, strings []string) error {
|
|
return cli.RPCTemplate(true)
|
|
}))
|
|
templateCmd = cobrax.NewCommand("template", cobrax.WithRunE(func(command *cobra.Command, strings []string) error {
|
|
return cli.RPCTemplate(false)
|
|
}))
|
|
|
|
newCmd = cobrax.NewCommand("new", cobrax.WithRunE(cli.RPCNew), cobrax.WithArgs(cobra.MatchAll(cobra.ExactArgs(1), cobra.OnlyValidArgs)))
|
|
protocCmd = cobrax.NewCommand("protoc", cobrax.WithRunE(cli.ZRPC), cobrax.WithArgs(cobra.MatchAll(cobra.ExactArgs(1), cobra.OnlyValidArgs)))
|
|
)
|
|
|
|
func init() {
|
|
var (
|
|
rpcCmdFlags = Cmd.Flags()
|
|
newCmdFlags = newCmd.Flags()
|
|
protocCmdFlags = protocCmd.Flags()
|
|
templateCmdFlags = templateCmd.Flags()
|
|
)
|
|
|
|
rpcCmdFlags.StringVar(&cli.VarStringOutput, "o")
|
|
rpcCmdFlags.StringVar(&cli.VarStringHome, "home")
|
|
rpcCmdFlags.StringVar(&cli.VarStringRemote, "remote")
|
|
rpcCmdFlags.StringVar(&cli.VarStringBranch, "branch")
|
|
|
|
newCmdFlags.StringSliceVar(&cli.VarStringSliceGoOpt, "go_opt")
|
|
newCmdFlags.StringSliceVar(&cli.VarStringSliceGoGRPCOpt, "go-grpc_opt")
|
|
newCmdFlags.StringVarWithDefaultValue(&cli.VarStringStyle, "style", config.DefaultFormat)
|
|
newCmdFlags.BoolVar(&cli.VarBoolIdea, "idea")
|
|
newCmdFlags.StringVar(&cli.VarStringHome, "home")
|
|
newCmdFlags.StringVar(&cli.VarStringRemote, "remote")
|
|
newCmdFlags.StringVar(&cli.VarStringBranch, "branch")
|
|
newCmdFlags.BoolVarP(&cli.VarBoolVerbose, "verbose", "v")
|
|
newCmdFlags.MarkHidden("go_opt")
|
|
newCmdFlags.MarkHidden("go-grpc_opt")
|
|
|
|
protocCmdFlags.BoolVarP(&cli.VarBoolMultiple, "multiple", "m")
|
|
protocCmdFlags.StringSliceVar(&cli.VarStringSliceGoOut, "go_out")
|
|
protocCmdFlags.StringSliceVar(&cli.VarStringSliceGoGRPCOut, "go-grpc_out")
|
|
protocCmdFlags.StringSliceVar(&cli.VarStringSliceGoOpt, "go_opt")
|
|
protocCmdFlags.StringSliceVar(&cli.VarStringSliceGoGRPCOpt, "go-grpc_opt")
|
|
protocCmdFlags.StringSliceVar(&cli.VarStringSlicePlugin, "plugin")
|
|
protocCmdFlags.StringSliceVarP(&cli.VarStringSliceProtoPath, "proto_path", "I")
|
|
protocCmdFlags.StringVar(&cli.VarStringStyle, "style")
|
|
protocCmdFlags.StringVar(&cli.VarStringZRPCOut, "zrpc_out")
|
|
protocCmdFlags.StringVar(&cli.VarStringHome, "home")
|
|
protocCmdFlags.StringVar(&cli.VarStringRemote, "remote")
|
|
protocCmdFlags.StringVar(&cli.VarStringBranch, "branch")
|
|
protocCmdFlags.BoolVarP(&cli.VarBoolVerbose, "verbose", "v")
|
|
protocCmdFlags.MarkHidden("go_out")
|
|
protocCmdFlags.MarkHidden("go-grpc_out")
|
|
protocCmdFlags.MarkHidden("go_opt")
|
|
protocCmdFlags.MarkHidden("go-grpc_opt")
|
|
protocCmdFlags.MarkHidden("plugin")
|
|
protocCmdFlags.MarkHidden("proto_path")
|
|
|
|
templateCmdFlags.StringVar(&cli.VarStringOutput, "o")
|
|
templateCmdFlags.StringVar(&cli.VarStringHome, "home")
|
|
templateCmdFlags.StringVar(&cli.VarStringRemote, "remote")
|
|
templateCmdFlags.StringVar(&cli.VarStringBranch, "branch")
|
|
|
|
Cmd.AddCommand(newCmd, protocCmd, templateCmd)
|
|
}
|