优化代码以避免多余的response.WriteHeader调用，实现返回500状态码

由于 `http.ResponseWriter.WriteHeader` 方法仅允许在 HTTP 响应中调用一次，并且 `http.ResponseWriter.Write` 方法可能会失败，因此存在一个问题：如果 `http.ResponseWriter.Write` 失败，如何将响应状态代码从 201 更改为 500？

我知道 http.responsewriter 的 writeheader 方法每个 http 响应只能调用一次，只能有一个响应状态代码，并且只能发送一次标头。这一切都很好。

问题是，如果 http.responsewriter.write 返回错误，我应该如何重构代码才能覆盖 201 并返回 500 ？正如你在下面看到的，我故意强制恐慌，看看 httprouter.router.panichandler 如何处理它。正如预期的那样，日志显示 http: 多余的响应。来自 ... 的 writeheader 调用，响应是 201，因为如上所述为时已晚。

package server

import (
    "github.com/julienschmidt/httprouter"
    "log"
    "net/http"
)

func serve() {
    rtr := httprouter.new()
    rtr.get("/", home.welcome)

    handle500(rtr)

    err := http.listenandserve(":8080", rtr)
    if err != nil {
        log.fatalf("server crash")
    }
}

func handle500(r *httprouter.router) {
    r.panichandler = func(res http.responsewriter, req *http.request, err interface{}) {
        res.writeheader(http.statusinternalservererror)
        // http: superfluous response.writeheader call from line above
    }
}

package home

import (
    "github.com/julienschmidt/httprouter"
    "net/http"
)

func Welcome(res http.ResponseWriter, _ *http.Request, _ httprouter.Params) {
    // doing a few bits and building the body

    res.Header().Set("Content-Type", "application/json")
    res.WriteHeader(201)

    _, err := res.Write("body goes here")
    if err == nil {  // I am doing this deliberately to test 500
        panic("assume that something has gone wrong with res.Write and an error occurred")
    }
}

解决方案

无法“覆盖”状态代码，因为它会立即发送到浏览器。

您正在检查 http.ResponseWriter.Write() 的返回值。我不确定这是一个好的策略。如果写入响应失败，那么写入更多内容也可能会失败。

记录失败似乎更合适，但我希望大多数失败是由于连接断开和其他不需要操作的错误。

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~