-
Notifications
You must be signed in to change notification settings - Fork 0
/
gotsql.go
57 lines (47 loc) · 1.1 KB
/
gotsql.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
package gotsql
import (
"bytes"
"fmt"
"os"
"path/filepath"
"text/template"
)
type GoTSQL struct {
Templates map[string]*template.Template
}
func (g *GoTSQL) Load(filedir string) {
if g.Templates == nil {
g.Templates = make(map[string]*template.Template)
}
err := filepath.Walk(filedir, func(path string, info os.FileInfo, err error) error {
if info.IsDir() {
return nil
}
if filepath.Ext(path) != ".gotsql" {
return nil
}
t, er := template.ParseFiles(path)
if er != nil {
return fmt.Errorf("error parsing template %s", path, er)
}
g.Templates[path[:len(path)-len(filepath.Ext(path))]] = t
return nil
})
if err != nil {
panic(err)
}
}
func (g *GoTSQL) Get(commandName string, data interface{}) (string, error) {
namespace := filepath.Dir(commandName)
if t, ok := g.Templates[namespace]; ok {
var b bytes.Buffer
templateName := filepath.Base(commandName)
tmpl_err := t.ExecuteTemplate(&b, templateName, data)
if tmpl_err != nil {
return "", tmpl_err
}
return b.String(), nil
} else {
return "", fmt.Errorf("template %s not found or loaded", namespace)
}
}