diff --git a/utils/template_renderer.go b/utils/template_renderer.go index bf16f0e..6c82d14 100644 --- a/utils/template_renderer.go +++ b/utils/template_renderer.go @@ -8,8 +8,15 @@ import ( "github.com/labstack/echo/v4" ) +var isDevMode bool + +func init() { + isDevMode = os.Getenv("MODE") == "development" +} + // TemplateRenderer holds cached templates for rendering type TemplateRenderer struct { + baseFile string baseTemplate *template.Template templateMap map[string]*template.Template } @@ -17,6 +24,7 @@ type TemplateRenderer struct { // NewTemplateRenderer constructs a template renderer with a base file func NewTemplateRenderer(baseFile string) *TemplateRenderer { return &TemplateRenderer{ + baseFile: baseFile, baseTemplate: template.Must(template.ParseFiles("./views/" + baseFile)), templateMap: make(map[string]*template.Template), } @@ -24,9 +32,16 @@ func NewTemplateRenderer(baseFile string) *TemplateRenderer { // Render the template func (t *TemplateRenderer) Render(w io.Writer, name string, data interface{}, c echo.Context) error { + tmpl := t.templateMap[name] - if os.Getenv("MODE") == "development" || tmpl == nil { - tmpl = template.Must(t.baseTemplate.Clone()) + if isDevMode || tmpl == nil { + + if isDevMode { + tmpl = template.Must(template.ParseFiles("./views/" + t.baseFile)) + } else { + tmpl = template.Must(t.baseTemplate.Clone()) + } + tmpl.ParseFiles("./views/" + name) t.templateMap[name] = tmpl }