1 Star 0 Fork 1

猿梦/ginmultitemplate

forked from Jalright/ginmultitemplate 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
templates.go 1.76 KB
一键复制 编辑 原始数据 按行查看 历史
Jalright 提交于 2020-02-17 23:25 . gin多级目录支持
package ginmultitemplate
import (
"fmt"
"html/template"
"io/ioutil"
"os"
"path"
"path/filepath"
"strings"
"github.com/gin-gonic/gin"
"github.com/gin-contrib/multitemplate"
)
func getFilelist(path string, stuffix string) (files []string) {
// 遍历目录
filepath.Walk(path, func(path string, f os.FileInfo, err error) error {
if f == nil {
return err
}
if f.IsDir() {
return nil
}
// 将模板后缀的文件放到列表
if strings.HasSuffix(path, stuffix) {
files = append(files, path)
}
return nil
})
return
}
// LoadTemplateFiles 加载模板
func LoadTemplateFiles(templateDir, stuffix string, funcMap template.FuncMap) multitemplate.Renderer {
if gin.IsDebugging() {
fmt.Println("========================")
fmt.Println("[debug] loading template files ")
fmt.Println("========================")
}
r := multitemplate.NewRenderer()
rd, _ := ioutil.ReadDir(templateDir)
for _, fi := range rd {
if fi.IsDir() {
// 如果是目录
for _, f := range getFilelist(path.Join(templateDir, fi.Name()), stuffix) {
// 添加到模板的时候,去掉跟路径
if len(funcMap) > 0 {
r.AddFromFilesFuncs(f[len(templateDir)+1:], funcMap, f)
} else {
r.AddFromFiles(f[len(templateDir)+1:], f)
}
if gin.IsDebugging() {
fmt.Println(f[len(templateDir)+1:])
}
}
} else {
if strings.HasSuffix(fi.Name(), stuffix) {
// 如果再根目录底下的文件直接添加
if len(funcMap) > 0 {
r.AddFromFilesFuncs(fi.Name(), funcMap, path.Join(templateDir, fi.Name()))
} else {
r.AddFromFiles(fi.Name(), path.Join(templateDir, fi.Name()))
}
if gin.IsDebugging() {
fmt.Println(fi.Name())
}
}
}
}
if gin.IsDebugging() {
fmt.Println("========================")
}
return r
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/polarrwl/ginmultitemplate.git
git@gitee.com:polarrwl/ginmultitemplate.git
polarrwl
ginmultitemplate
ginmultitemplate
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385