优化router、main等文件。

This commit is contained in:
2026-08-14 17:59:20 +08:00
parent 0f0850357d
commit f1780de632
6 changed files with 303 additions and 180 deletions
+30 -29
View File
@@ -67,12 +67,14 @@ func (r *htmlRender) WriteContentType(w http.ResponseWriter) {
}
// LoadTemplates 加载指定主题的模板
func LoadTemplates(theme string) *TemplateRenderer {
func LoadTemplates(theme string) (*TemplateRenderer, error) {
r := &TemplateRenderer{
templates: make(map[string]*template.Template),
}
r.reload(theme)
return r
if err := r.reload(theme); err != nil {
return nil, err
}
return r, nil
}
// Reload 重新加载主题模板(线程安全)
@@ -83,34 +85,35 @@ func (r *TemplateRenderer) Reload(theme string) {
}
// reload 内部重载方法(需要持有锁)
func (r *TemplateRenderer) reload(theme string) {
func (r *TemplateRenderer) reload(theme string) error {
templateDir := "templates/" + theme
// 需要独立渲染的页面模板
pages := []string{"index", "post", "page"}
// 收集主题目录下的公共局部模板(例如 sidebar/header/footer),
// 这些模板通过 {{template "xxx" .}} 被 base.html 引用,
// 必须与页面模板一起解析,否则渲染时会因模板未定义而失败(白屏)。
// 收集主题目录下的公共局部模板
partials := []string{}
if matches, err := filepath.Glob(templateDir + "/*.html"); err == nil {
for _, f := range matches {
base := filepath.Base(f)
if base == "base.html" {
continue
}
name := base[:len(base)-len(".html")]
isPage := false
for _, p := range pages {
if p == name {
isPage = true
break
}
}
if !isPage {
partials = append(partials, f)
matches, err := filepath.Glob(templateDir + "/*.html")
if err != nil {
return fmt.Errorf("扫描主题目录失败: %w", err)
}
for _, f := range matches {
base := filepath.Base(f)
if base == "base.html" {
continue
}
name := base[:len(base)-len(".html")]
isPage := false
for _, p := range pages {
if p == name {
isPage = true
break
}
}
if !isPage {
partials = append(partials, f)
}
}
// 为每个页面创建独立的模板组合
@@ -125,19 +128,17 @@ func (r *TemplateRenderer) reload(theme string) {
)
allFiles = append(allFiles, partials...)
// 解析模板文件
// 解析模板文件,遇到错误立即返回
tmpl, err := template.New("").Funcs(templateFuncs()).ParseFiles(allFiles...)
if err != nil {
// 如果解析失败,记录错误但不中断
fmt.Printf("警告: 加载主题 %s 的 %s 页面模板失败: %v\n", theme, pageName, err)
continue
return fmt.Errorf("解析页面模板 %s 失败: %w", pageName, err)
}
newTemplates[pageName] = tmpl
}
// 更新模板映射
// 所有页面模板解析成功后才更新映射
r.templates = newTemplates
fmt.Printf("主题 %s 加载完成,共加载 %d 个页面模板\n", theme, len(newTemplates))
return nil
}
// 全局模板渲染器实例