http覆盖golang中的http头代码，而json编码有错误

Question

考虑一下这种情况！

http请求执行成功后，如果执行json编码出错，如何覆盖header代码

func writeResp(w http.ResponseWriter, code int, data interface{}) {
    w.Header().Set("Content-Type", "application/json")

    //Here I set the status to 201 StatusCreated
    w.WriteHeader(code) 
    s := success{Data: data}

    //what if there is an error here and want to override the status to 5xx error
    //how to handle error here, panic?, http.Error() is not an option because as we already wrote header to 201, it just prints `http: multiple response.WriteHeader calls`
    if err := json.NewEncoder(w).Encode(s); err != nil {
        w.Header().Set("Content-Type", "application/json")

        //it throws http: multiple response.WriteHeader calls here as we already wrote header above to 201
        w.WriteHeader(code)
        e := errorResponse{
            Code:        code,
            Error:       error,
            Description: msg,
        }
        if err := json.NewEncoder(w).Encode(e); err != nil {
         //same how to handle here
        }
    }
}

我在这里有多个选项，如果我们只是进行致命日志记录，用户将不会确切知道发生了什么，即使我使用w.Write([]byte(msg)) 编写字符串，状态仍然显示201 created，如何以错误代码 5xx 响应

非常感谢任何帮助

Answer 1

首先，编码时出错的可能性似乎不大。

查看此问题了解Marshal 失败的原因：

What input will cause golang's json.Marshal to return an error?

另一个潜在的错误原因是实际将数据写入响应流时出现问题，但在这种情况下，您也无法写入自定义错误。

回到你的问题，如果你担心编码你的对象可能会失败，你可以先编组你的数据（检查错误），然后如果编组成功，只写 201 状态代码（和编码的数据）。

稍微修改一下你的例子：

s := success{Data: data}
jsonData, err := json.Marshal(s)
if err != nil {
    // write your error to w, then return
}
w.WriteHeader(code)
w.Header().Set("Content-Type", "application/json")
w.Write(jsonData)

现在，最后一个 write 也可能引发错误。

但如果发生这种情况，在编写自定义错误时也会失败，因此在这种情况下，您最好将其记录在服务器端（或将该错误发送到 New Relic 等跟踪器）。