diff --git a/tools/goctl/model/mongo/generate/template.go b/tools/goctl/model/mongo/generate/template.go index 03bbce18..a422ed71 100644 --- a/tools/goctl/model/mongo/generate/template.go +++ b/tools/goctl/model/mongo/generate/template.go @@ -19,18 +19,22 @@ var templates = map[string]string{ errTemplateFile: template.Error, } +// Category returns the mongo category. func Category() string { return category } +// Clean cleans the mongo templates. func Clean() error { return util.Clean(category) } +// Templates initializes the mongo templates. func Templates(_ *cli.Context) error { return util.InitTemplates(category, templates) } +// RevertTemplate reverts the given template. func RevertTemplate(name string) error { content, ok := templates[name] if !ok { @@ -40,6 +44,7 @@ func RevertTemplate(name string) error { return util.CreateTemplate(category, name, content) } +// Update cleans and updates the templates. func Update() error { err := Clean() if err != nil { diff --git a/tools/goctl/model/mongo/mongo.go b/tools/goctl/model/mongo/mongo.go index 39d96281..b22d1264 100644 --- a/tools/goctl/model/mongo/mongo.go +++ b/tools/goctl/model/mongo/mongo.go @@ -10,7 +10,7 @@ import ( "github.com/urfave/cli" ) -// Command provides the entry for goctl +// Action provides the entry for goctl mongo code generation. func Action(ctx *cli.Context) error { tp := ctx.StringSlice("type") c := ctx.Bool("cache") diff --git a/tools/goctl/model/mongo/template/template.go b/tools/goctl/model/mongo/template/template.go index 1862d029..416064fd 100644 --- a/tools/goctl/model/mongo/template/template.go +++ b/tools/goctl/model/mongo/template/template.go @@ -101,6 +101,7 @@ func (m *default{{.Type}}Model) Delete(ctx context.Context, id string) error { } ` +// Error provides the default template for error definition in mongo code generation. var Error = ` package model