我正在构建一个CLI来为自制的API框架生成代码(现在生成控制器部分)。为此,我使用模板,但是当我在模板中使用诸如package
或func
之类的单词时,我看到模板没有生成任何内容(空文件)。
我想构建以下模板:
package controllers
{{- range .Methods }}
{{ if eq .Name "Create" }}
func ({{ firstChar $.ModelName }}c {{ title $.ModelName }}Controller) Get{{ title $.ModelName }}(c *gin.Context) {
{{ $.ModelName }}, err := store.Find{{ title $.ModelName }}ById(c, c.Param("id"))
if err != nil {
c.AbortWithError(http.StatusNotFound, helpers.ErrorWithCode("{{ $.ModelName }}_not_found", "The {{ $.ModelName }} does not exist", err))
return
}
c.JSON(http.StatusOK, {{ $.ModelName }})
}
{{ else if eq .Name "Get" }}
{{ else if eq .Name "GetAll" }}
{{ else if eq .Name "Update" }}
{{ else if eq .Name "Delete" }}
{{ end }}
{{- end }}
你知道如何使模板工作吗?
它现在正在工作,我只是在template.Must
之前添加template.New
,现在它就像一个魅力。
path := filepath.Join("templates", "controller.tmpl")
body, _ := ioutil.ReadFile(path)
tmpl := template.Must(template.New("model").Funcs(funcMap).Parse(string(body)))
var buf bytes.Buffer
err := tmpl.Execute(&buf, selectedMethods)
utils.Check(err)
src, _ := format.Source(buf.Bytes())