如何访问接口的属性

Question

我打算在两个响应结构的标头和正文中都使用 HTTP 状态代码。但是，没有将状态码设置为函数参数两次，并再次为结构体设置以避免冗余。

JSON() 的参数response 是一个允许两个结构都被接受的接口。编译器抛出以下异常：

response.Status undefined (type interface {} has no field or method Status)

因为响应字段不能有状态属性。有没有另一种方法可以避免两次设置状态码？

type Response struct {
    Status int         `json:"status"`
    Data   interface{} `json:"data"`
}

type ErrorResponse struct {
    Status int      `json:"status"`
    Errors []string `json:"errors"`
}

func JSON(rw http.ResponseWriter, response interface{}) {
    payload, _ := json.MarshalIndent(response, "", "    ")
    rw.WriteHeader(response.Status)
    ...
}

Answer 1

rw.WriteHeader(response.Status) 中的类型response 是interface{}。在 Go 中，您需要显式声明底层结构的类型，然后访问该字段：

func JSON(rw http.ResponseWriter, response interface{}) {
    payload, _ := json.MarshalIndent(response, "", "    ")
    switch r := response.(type) {
    case ErrorResponse:
        rw.WriteHeader(r.Status)
    case Response:
        rw.WriteHeader(r.Status) 
    }
    ...
}

然而，更好的首选方法是为您的响应定义一个通用接口，该接口具有获取响应状态的方法：

type Statuser interface {
    Status() int
}

// You need to rename the fields to avoid name collision.
func (r Response) Status() int { return r.ResStatus }
func (r ErrorResponse) Status() int { return r.ResStatus }

func JSON(rw http.ResponseWriter, response Statuser) {
    payload, _ := json.MarshalIndent(response, "", "    ")
    rw.WriteHeader(response.Status())
    ...
}

最好将 Response 重命名为 DataResponse 和 ResponseInterface 重命名为 Response，IMO。

Answer 2

接口没有属性，所以需要从接口中提取结构体。为此，您使用type assertion

if response, ok := response.(ErrorResponse); ok {
    rw.WriteHeader(response.Status)
    ...