Go语言HTTP HEAD请求与模板渲染的冲突解析及处理

package main

import (
    "html/template"
    "log"
    "net/http"
)

var (
    templates *template.Template
)

// fooHandler 示例：看似成功，实则忽略了错误
func fooHandler(w http.ResponseWriter, req *http.Request) {
    // 实际上，如果req.Method是HEAD，w.Write会返回http.ErrBodyNotAllowed错误
    // 但此处的代码忽略了该错误，导致外部看起来没有问题
    _, err := w.Write([]byte("fooHandler"))
    if err != nil && err != http.ErrBodyNotAllowed { // 明确处理ErrBodyNotAllowed
        log.Printf("Error writing to response for fooHandler: %v", err)
    }
}

// homeHandler 示例：使用模板渲染，直接触发错误
func homeHandler(w http.ResponseWriter, req *http.Request) {
    // 当req.Method是HEAD时，ExecuteTemplate尝试写入响应体，导致错误
    err := templates.ExecuteTemplate(w, "main.html", nil)
    if err != nil {
        // 对于HEAD请求，这里的错误通常是 "http: request method or response status code does not allow body"
        log.Printf("Error executing template for homeHandler: %v", err)
        // 避免在生产环境中直接log.Fatal，通常会返回一个错误状态码
        http.Error(w, "Internal Server Error", http.StatusInternalServerError)
    }
}

func main() {
    var err error
    templates, err = template.ParseGlob("templates/*.html")
    if err != nil {
        log.Fatalf("Loading template: %v", err)
    }
    http.HandleFunc("/", homeHandler)
    http.HandleFunc("/foo", fooHandler)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

// 假设 templates/main.html 文件内容为: homeHandler

// 改进后的 homeHandler
func homeHandler(w http.ResponseWriter, req *http.Request) {
    if req.Method == http.MethodHead {
        // 对于HEAD请求，我们只设置必要的响应头
        // 假设我们知道main.html渲染后内容是 "homeHandler"，长度是10
        w.Header().Set("Content-Type", "text/html; charset=utf-8")
        w.Header().Set("Content-Length", "10") // 根据实际渲染内容计算
        w.WriteHeader(http.StatusOK) // 发送状态码
        return // 不写入任何响应体
    }

    // 对于GET或其他请求，正常执行模板渲染
    err := templates.ExecuteTemplate(w, "main.html", nil)
    if err != nil {
        log.Printf("Error executing template for homeHandler: %v", err)
        http.Error(w, "Internal Server Error", http.StatusInternalServerError)
    }
}