我有以下 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 保留在内存中并仅在成功时设置状态代码来处理此问题。但这会创建不需要的副本并且不太好。
有没有更好的方法来处理这个问题?
最佳答案
稍微扩展一下我的评论:
没有缓冲就没有办法做到这一点。如果你想一想,即使有,又该如何实现呢?响应头需要在内容之前发送。如果代码依赖于对响应进行编码,则必须首先进行编码、检查结果、设置 header 并刷新编码缓冲区。
因此,即使 Go API 支持“延迟”状态,您基本上也会将缓冲问题向下一级推到 http 库:)
您应该进行调用 - 要么不惜一切代价获得正确的响应代码很重要并且您可以承受缓冲,或者您想要流式传输响应并且无法更改结果。
理论上,Go 可以创建一个编码验证器,以确保您尝试编码的对象在实际编码之前 100% 通过。
顺便说一句,这让我想到了另一件事——响应代码的语义。你返回 HTTP Created,对吧?这是正确的代码,因为该对象实际上已经创建了。但如果由于编码的原因返回500错误,是不是对象还没有创建呢?居然还有!所以创建的代码仍然有效,错误只是在编码阶段。那么也许更好的设计是不返回对象作为响应?
关于rest - 如何在 Go HTTP 服务器中有条件地设置 HTTP 状态代码?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52849382/