在 Go 中将 interface{} 参数转换为 *http.Request

Question

我正在创建一个 util 函数，它将读取请求/响应的主体并返回它。这是我目前所做的：func GetBody(in interface{}) []byte {    var body io.Reader    var statusCode int    switch v := in.(type) {        case *http.Request, *http.Response:            body = v.Body            statusCode = v.StatusCode        default:            log.Fatal("Only http.Request and http.Response parameters can be accepted to parse body")    }    if statusCode != 200 {        log.Fatalf("Received status code [%d] instead of [200]", statusCode)    }    body, err := ioutil.ReadAll(body)    if err != nil {        log.Fatal(err)    }    return body}但是我收到一个编译器错误：v.Body undefined (type interface {} is interface with no methods)我是不是遗漏了什么，或者不可能制作一个通用函数来同时为*http.Request和*http.Response

Answer 1

这是因为双case.

v仍然是一个interface{}因为它可以是一个*http.Request或一个*http.Response

switch v := in.(type) {

    case *http.Request

        body = v.Body

        statusCode = v.StatusCode

    case *http.Response:

        body = v.Body

        statusCode = v.StatusCode

    default:

        log.Fatal("Only http.Request and http.Response parameters can be accepted to parse body")

}

这应该工作

Answer 2

除了直接解决您的类型开关问题的其他答案之外，我想指出一个替代解决方案。请注意，该interface{}解决方案非常好，并且很容易被认为比这更可取。这是为了启迪。

首先，稍微说一下，如果您感兴趣的是一个通用方法（例如 Write或Cookies）而不是一个通用字段（Body），那么通过自定义界面访问它会更容易也更好。

通过定义一个类型：

type cookier interface { // Should probably use a better name

    Cookies() []*http.Cookie

}  

func ShowCookies1(r cookier) {

    log.Println("Got cookies:", r.Cookies())

}

或者在函数定义中使用匿名类型：

func ShowCookies2(r interface {

    Cookies() []*http.Cookie

}) {

    log.Println("Got cookies:", r.Cookies())

}

这些函数可以接受任何有Cookies方法的东西，包括*http.Request和*http.Response。

不幸的是，在您的特定情况下，您希望访问公共字段而不是公共方法，因此您不能直接使用匹配的接口。

您可以创建一个添加GetBody方法的小型包装器类型（人们可能会争辩说这样的函数应该已在标准包中定义）。

type reqbody struct{ *http.Request }

type respbody struct{ *http.Response }

type getbody interface {

    GetBody() io.ReadCloser

}

func (r reqbody) GetBody() io.ReadCloser  { return r.Body }

func (r respbody) GetBody() io.ReadCloser { return r.Body }

func GetBody2(r getbody) ([]byte, error) {

    body := r.GetBody()

    defer body.Close()

    return ioutil.ReadAll(body)

}

调用者知道他们有什么类型并执行以下操作之一：

    buf, err = GetBody2(reqbody{req})

    buf, err = GetBody2(respbody{resp})

从某种意义上说，这比仅使用interface{}. 但它有一个，而不是有一个函数，它绝对是任何类型和恐慌/错误的好处在运行时，如果一个程序员错误的东西不能适当类型的话来说，这反而迫使调用者安全通过一些你知道的是编译时类型正确。

望着这进一步，你只是读一切从io.ReadCloser再关闭它，所以它可以进一步简化为以下（这可能是比你的interface{}解决方案）：

func GetReqBody(r *http.Request) io.ReadCloser   { return r.Body }

// Could add checking r.StatusCode to the following one as well:

func GetRespBody(r *http.Response) io.ReadCloser { return r.Body }

func ReadAndClose(rc io.ReadCloser) ([]byte, error) {

    defer rc.Close()

    return ioutil.ReadAll(rc)

}

同样，调用者知道他们拥有什么类型并执行以下操作之一：

    buf, err = ReadAndClose(GetReqBody(req))

    buf, err = ReadAndClose(GetRespBody(resp))

要不就：

    buf, err = ReadAndClose(req.Body)

    buf, err = ReadAndClose(resp.Body)

您可以在Go Playground上看到所有这些选项的示例。

最后，小心使用ioutil.ReadAll. 通常最好避免将整个文件或网络流预读到缓冲区中，而是在读取时将其作为流处理。特别是，用任意大的 body 发出 HTTP 请求作为拒绝服务攻击或浪费服务器资源是微不足道的（http.MaxBytesReader也可以提供帮助）。

Answer 3

你想要做的更像是这样的：

switch in.(type) {

    case *http.Request:

        body = v.(*http.Request).Body

    case *http.Response:

        body = v.(*http.Response).Body

    default:

        log.Fatal(...)

}

编辑：我删除了我的答案的错误部分，请参阅 HectorJ 的答案以了解更多语法，请执行此操作。