Skip to content

Commit 482fca8

Browse files
committed
refactor(tpl): refine template initializing
1 parent 8e2959d commit 482fca8

File tree

2 files changed

+17
-17
lines changed

2 files changed

+17
-17
lines changed

src/tpl/page.html.go

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -68,33 +68,33 @@ const pageTplStr = `
6868
</html>
6969
`
7070

71-
var defaultPage *template.Template
71+
var defaultPageTpl *template.Template
7272

7373
func init() {
74-
tplObj := template.New("page")
75-
tplObj = addFuncMap(tplObj)
74+
pageTpl := template.New("page")
75+
pageTpl = addFuncMap(pageTpl)
7676

7777
var err error
78-
defaultPage, err = tplObj.Parse(pageTplStr)
78+
defaultPageTpl, err = pageTpl.Parse(pageTplStr)
7979
if serverErrHandler.CheckError(err) {
80-
defaultPage = template.Must(tplObj.Parse("Builtin Template Error"))
80+
defaultPageTpl = template.Must(pageTpl.Parse("Builtin Template Error"))
8181
}
8282
}
8383

8484
func LoadPage(tplPath string) (*template.Template, error) {
85-
var tplObj *template.Template
86-
var err error
87-
88-
if len(tplPath) > 0 {
89-
tplObj = template.New(path.Base(tplPath))
90-
tplObj = addFuncMap(tplObj)
91-
tplObj, err = tplObj.ParseFiles(tplPath)
85+
if len(tplPath) == 0 {
86+
return defaultPageTpl, nil
9287
}
93-
if err != nil || len(tplPath) == 0 {
94-
tplObj = defaultPage
88+
89+
var err error
90+
pageTpl := template.New(path.Base(tplPath))
91+
pageTpl = addFuncMap(pageTpl)
92+
pageTpl, err = pageTpl.ParseFiles(tplPath)
93+
if err != nil {
94+
pageTpl = defaultPageTpl
9595
}
9696

97-
return tplObj, err
97+
return pageTpl, err
9898
}
9999

100100
func addFuncMap(tpl *template.Template) *template.Template {

src/vhostMux/main.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ func NewServeMux(
4343
}
4444

4545
// template
46-
tplObj, err := tpl.LoadPage(p.Template)
46+
pageTpl, err := tpl.LoadPage(p.Template)
4747
errorHandler.LogError(err)
4848

4949
// register handlers
@@ -58,7 +58,7 @@ func NewServeMux(
5858
handlers := map[string]http.Handler{}
5959
for urlPath, fsPath := range aliases {
6060
emptyHandlerRoot := emptyRoot && urlPath == "/"
61-
handlers[urlPath] = serverHandler.NewHandler(fsPath, emptyHandlerRoot, urlPath, p, users, tplObj, logger, errorHandler)
61+
handlers[urlPath] = serverHandler.NewHandler(fsPath, emptyHandlerRoot, urlPath, p, users, pageTpl, logger, errorHandler)
6262
}
6363

6464
// create ServeMux

0 commit comments

Comments
 (0)