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.
70 lines
2.0 KiB
Go
70 lines
2.0 KiB
Go
package gen
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
const (
|
|
category = "rpc"
|
|
callTemplateFile = "call.tpl"
|
|
callTypesTemplateFile = "call-types.tpl"
|
|
callInterfaceFunctionTemplateFile = "call-interface-func.tpl"
|
|
callFunctionTemplateFile = "call-func.tpl"
|
|
configTemplateFileFile = "config.tpl"
|
|
etcTemplateFileFile = "etc.tpl"
|
|
logicTemplateFileFile = "logic.tpl"
|
|
logicFuncTemplateFileFile = "logic-func.tpl"
|
|
mainTemplateFile = "main.tpl"
|
|
serverTemplateFile = "server.tpl"
|
|
serverFuncTemplateFile = "server-func.tpl"
|
|
svcTemplateFile = "svc.tpl"
|
|
rpcTemplateFile = "template.tpl"
|
|
)
|
|
|
|
var templates = map[string]string{
|
|
callTemplateFile: callTemplateText,
|
|
callTypesTemplateFile: callTemplateTypes,
|
|
callInterfaceFunctionTemplateFile: callInterfaceFunctionTemplate,
|
|
callFunctionTemplateFile: callFunctionTemplate,
|
|
configTemplateFileFile: configTemplate,
|
|
etcTemplateFileFile: etcTemplate,
|
|
logicTemplateFileFile: logicTemplate,
|
|
logicFuncTemplateFileFile: logicFunctionTemplate,
|
|
mainTemplateFile: mainTemplate,
|
|
serverTemplateFile: serverTemplate,
|
|
serverFuncTemplateFile: functionTemplate,
|
|
svcTemplateFile: svcTemplate,
|
|
rpcTemplateFile: rpcTemplateText,
|
|
}
|
|
|
|
func GenTemplates(_ *cli.Context) error {
|
|
return util.InitTemplates(category, templates)
|
|
}
|
|
|
|
func RevertTemplate(name string) error {
|
|
content, ok := templates[name]
|
|
if !ok {
|
|
return fmt.Errorf("%s: no such file name", name)
|
|
}
|
|
return util.CreateTemplate(category, name, content)
|
|
}
|
|
|
|
func Clean() error {
|
|
return util.Clean(category)
|
|
}
|
|
|
|
func Update(category string) error {
|
|
err := Clean()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return util.InitTemplates(category, templates)
|
|
}
|
|
|
|
func GetCategory() string {
|
|
return category
|
|
}
|