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

Question

我有以下 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)
    }
}

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

1. 请求以 JSON 数据的形式出现。它被解码为req
2. 请求保存在数据库中。这将返回一个结果对象，这将是响应

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

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

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

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

Answer 1

稍微扩展一下我的评论:

没有缓冲就无法做到这一点。如果您考虑一下，即使有，那将如何实现？响应头需要在内容之前发送。如果代码依赖于对响应进行编码，则必须先进行编码，检查结果，设置 header 并刷新编码缓冲区。

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

您应该进行调用 - 要么不惜一切代价获得正确的响应代码很重要并且您有能力缓冲，要么您想要流式传输响应并且您不能更改结果。

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

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