http - Golang根据struct字段设置http响应码

Question

在我正在编写的 api 中，我有一个错误结构，该结构编码为 json。当 api 出现错误时，它返回结构，我将 http 响应代码设置为适当的值。

type PodsError struct {
    ErrorCode       int     `json:"error_code"`
    CallingFunction string  `json:"calling_function"`
    Message         string  `json:"error_message"`
}

type PodsErrorWrapper struct {
    Error   PodsError   `json:"error"`
}

现在每次我写结构时我也会写一个头文件，但我不喜欢我看到的重复代码的数量。

error := PodsError{http.StatusNotFound, "Calling Func", "Message"}
response.WriteHeader(error.ErrorCode)
response.WriteEntity(PodsErrorWrapper{error})

是否可以将 WriteHeader 调用移动到每当我将错误传递给 WriteEntity() 时调用的对象？我认为必须有一个我可以为 PodsErrorWapper 实现的函数，我可以在其中将 http 状态设置为 ErrorCode 字段的任何值。

编辑:抱歉我忘了说，我正在使用 go-restful 包 (github.com/emicklei/go-restful)

Answer 1

您可以创建自己的函数:

func writeEntity(r *restful.Response, value interface{}) error {
    // if value is an error
    if perr, ok := value.(PodsError); ok {
        r.WriteHeader(perr.ErrorCode)
        // reassign value so it gets wrapped: `{"error": value}`
        value = struct {
            Error PodsError `json:"error"`
        }{perr}
    }
    return r.WriteEntity(value)
}

然后总是调用它而不是 response.WriteEntity:

writeEntity(response, PodsError{http.StatusNotFound, "Calling Func", "Message"})