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

最近发现不少小伙伴都对Golang很感兴趣，所以今天继续给大家介绍Golang相关的知识，本文《如何在 Go HTTP 服务器中有条件地设置 HTTP 状态代码？》主要内容涉及到等等知识点，希望能帮到你！当然如果阅读本文时存在不同想法，可以在评论中表达，但是请勿使用过激的措辞~

我有以下 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 保留在内存中并仅在成功时设置状态代码来处理此问题。但这会创建不需要的副本并且不太好。

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

解决方案

稍微扩展一下我的评论：

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

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

您应该进行调用 - 要么不惜一切代价获得正确的响应代码很重要并且您可以承受缓冲，或者您想要流式传输响应并且无法更改结果。

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

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

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

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

如果您知道答案，只需修复这些情况即可。如果您不知道，这些案例可能根本不存在。

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

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

本篇关于《如何在 Go HTTP 服务器中有条件地设置 HTTP 状态代码？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！