如何在 Go HTTP 服务器中有条件地设置 HTTP 状态代码？答案

【问题标题】：How do I set HTTP status code conditionally in a Go HTTP server?如何在 Go HTTP 服务器中有条件地设置 HTTP 状态代码？
【发布时间】：2019-03-21 19:24:16
【问题描述】：

我有以下 HTTP 处理函数：

func (h *UptimeHttpHandler) CreateSchedule(w http.ResponseWriter, r *http.Request) {
    defer r.Body.Close()
    dec := json.NewDecoder(r.Body)

    var req ScheduleRequest
    if err := dec.Decode(&req); err != nil {
        // error handling omited
    }

    result, err := saveToDb(req)
    if err != nil {
        // error handling omited
    }

    // Responding with 201-Created instead of default 200-Ok
    w.WriteHeader(http.StatusCreated)

    enc := json.NewEncoder(w)
    if err := enc.Encode(result); err != nil {
        // this will have no effect as the status is already set before
        w.WriteHeader(http.StatusInternalServerError)
        fmt.Fprintf(w, "%v", err)
    }
}

上面的代码做了以下事情：

请求以 JSON 数据形式出现。解码为req
请求被持久化在数据库中。这将返回一个结果对象，该对象将作为响应

现在在这里，一旦数据库插入成功，我们将状态码设置为201。然后使用 JSON 编码器将 JSON 编码值直接流式传输到 ResponseWriter。

当encode返回错误时，我需要将状态码更改为500。但目前我不能这样做，因为 Go 只允许设置一次状态码。

我可以通过将编码后的 JSON 保存在内存中来处理这个问题，并仅在成功时设置状态码。但这会产生不需要的副本，而且不是很好。

有没有更好的方法来处理这个问题？

【问题讨论】：

仅供参考，您不需要关闭 server 请求的主体，仅当您使用 client 获取请求对象时。请参阅本文档的最后一段：golang.org/pkg/net/http/#Request.Body
没有办法延迟写入，但是编码器返回错误的情况非常有限并且有记录，如果您的对象很大并且不缓冲 json 很重要，那么只需在这种情况下记录错误并返回一些理智的东西。由你决定哪个更重要......

标签： rest http go

【解决方案1】：

稍微扩展一下我的评论：

没有缓冲就没有办法做到这一点。如果你想一想，即使有，那将如何实施？响应头需要在内容之前发送。如果代码依赖于编码响应，则必须先编码，检查结果，设置标头并刷新编码缓冲区。

因此，即使 Go API 支持“延迟”状态，您基本上也会将缓冲问题向下推到 http 库:)

您应该拨打电话 - 重要的是不惜一切代价获取正确的响应代码并且您可以负担得起缓冲，或者您想要流式传输响应并且您无法更改结果。

理论上，Go 可以创建一个编码验证器，以确保您尝试编码的对象在实际编码之前 100% 通过。

顺便说一句，这让我想到了另一件事——响应代码的语义。您返回 HTTP Created，对吗？这是正确的代码，因为实际上已经创建了对象。但是如果你因为编码返回500错误，是不是对象没有被创建？它仍然有！所以创建的代码仍然有效，错误只是在编码阶段。所以也许更好的设计是不返回对象作为响应？

【讨论】：

感谢您的详细解答。可能不返回对象是正确的事情。我能做到。我可以设置Location 标头。或者暂时坚持200 OK。

【解决方案2】：

如果您的某些类型（深度）无法序列化为 JSON，则 JSON 编组可能会失败。

所以你应该问自己：响应的编码可能会失败的情况是什么？

如果您知道答案，请解决这些问题。如果你不知道，这可能是那些案例根本不存在。

如果您不确定，捕获错误的唯一方法是缓冲编码。您必须在缓冲和不返回干净错误之间平衡“不好”。

无论如何，如果只是 JSON 编码失败，则返回 500 错误确实不好。至少，您应该恢复对象创建（saveToDb 的工作），因此您必须包装一个在任何失败情况下都将回滚的事务。

【讨论】：