-
Notifications
You must be signed in to change notification settings - Fork 0
/
template.go
83 lines (72 loc) · 1.64 KB
/
template.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package task
import (
"errors"
"fmt"
"io"
"text/template"
"github.com/rs/zerolog/log"
)
type TemplateRenderTask struct {
// name template name
name string
// data template data
data interface{}
in io.Reader
out io.Writer
}
func NewTemplateRenderTask(name string, data interface{}, in io.Reader, out io.Writer) *TemplateRenderTask {
return &TemplateRenderTask{
name: name,
data: data,
in: in,
out: out,
}
}
func (trt *TemplateRenderTask) Run() error {
if err := trt.validateTask(); err != nil {
return err
}
if err := trt.renderTemplate(); err != nil {
return err
}
return nil
}
func (trt *TemplateRenderTask) validateTask() error {
if trt.name == "" {
return errors.New("empty template name")
}
return nil
}
func (trt *TemplateRenderTask) renderTemplate() error {
b, err := io.ReadAll(trt.in)
if err != nil {
return fmt.Errorf("read template data: %w", err)
}
log.Debug().Str("template_name", trt.name).Interface("data", trt.data).Msg("render template")
err = RenderTextTemplate(&RenderRequest{
name: trt.name,
template: string(b),
data: trt.data,
}, trt.out)
if err != nil {
return fmt.Errorf("render template: %w", err)
}
return nil
}
type RenderRequest struct {
// name template name
name string
template string
data interface{}
}
func RenderTextTemplate(req *RenderRequest, out io.Writer) error {
t := template.New(req.name)
pt, err := t.Parse(req.template)
if err != nil {
return fmt.Errorf("parse template [name: %s]: %w", req.name, err)
}
if err := pt.Execute(out, req.data); err != nil {
return fmt.Errorf("execute template [name: %s]: %w", req.name, err)
}
return nil
}