如何在 Revel 控制器中获取 http.ResponseWriter 和 http.Request

大家好，今天本人给大家带来文章《如何在 Revel 控制器中获取 http.ResponseWriter 和 http.Request》，文中内容主要涉及到，如果你对Golang方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

我正在尝试实现一个 oauth 服务器，我使用的包需要完整的 http.responsewriter 和 http.request 类型。

c.response 不包含 http.responsewriter 包含的所有方法，并且 c.request 给出错误类型不兼容。

如何在 revel 控制器中获取 http.responsewriter 和 http.request？

type client struct {
    ClientId string
    ClientSecret string
}
type App struct {
    *revel.Controller
}

func (c App) TokenRequest() {

    r := c.Request
    w := c.Response

    body, err := ioutil.ReadAll(r.Body)
    if err != nil {
        panic(err)
    }
    log.Println(string(body))

    var cli client
    err = json.Unmarshal(body, &cli)

    if err != nil {
        panic(err)
    }
    log.Println(cli.ClientId)

    err = OauthSrv.HandleTokenRequest(w, r)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
    }
}

解决方案

警告

我通常不喜欢像 go 中的 revel 这样的框架，因为我希望在此页面上展示自己的原因。我的第一个建议是，你仔细检查你实际上从 revel 中得到了什么，值得使用如此重的抽象层，如果它真的那么有价值，你可能想问另一个方向的问题，例如如何人们可以让 oauthsrv 在 revel 的定制生态系统中工作。

使用 revel 控制器作为 responsewriter

要成为 http.responsewriter，只需具有 these methods 即可。

标题

您需要一个名为 header() 的方法，该方法返回 http.header，您可以使用任何 map[string][]string 构建该方法。 revel 提供了类似的功能，但通过多个抽象层。您需要解开它们：

1. c.response 是 *response，因此它有一个名为 out 的字段，其中包含 outresponse。
2. outresponse 具有 header() 方法，但它不返回 http.header。相反，它返回一个 *revelheader。
3. *revelheader 具有 getall(key string) []string 方法，该方法与内置 map 类型已提供的 api 非常相似，但并不完全相同。因此，每次调用 header() 时，您都需要将返回值复制到新的映射中，以完全满足函数签名要求。
4. 此外，getall() 要求您知道您感兴趣的密钥名称，而 *revelheader 本身并不提供查找可用密钥的方法。 现在我们可以依赖这样一个事实：the current implementation 只有一个字段，即 serverheader，它确实提供了 getkeys() []string 方法。

将所有这些放在一起，我们可以构建我们的 header 方法：

func (rrw revelresponsewrapper) header() http.header {
  revelheader := rrw.response.out.header()
  keys := revelheader.server.getkeys()
  headermap := make(map[string][]string)

  for _, key := range keys {
    headermap[key] = revelheader.getall(key)
  }
  return http.header(headermap)
}

写入和写入标题

您可以使用类似的反模式来公开 rrw.write([]byte) (int, error)，以便它调用 c.response.out.write(data []byte) (int, error) 和 rrw.writeheader(int) 错误 以便它调用 c.response.writeheader(int, string)。根据认为适合框架的内容，要么因错误而恐慌，要么默默失败，因为他们的 api 不希望 writeheader 错误是可处理的。

从 revel 获取 http.request

不幸的是，http.request类型是一个结构体，所以你不能仅仅模拟它。您基本上有两个选择：使用 net/http 包从您能够访问的所有属性中重建它，或者希望您拥有的 *revel.request 是秘密的 http.request 。在后一种情况下，您可以使用类型断言：

revelReq, ok := c.Request.In.(*revel.GoRequest)
if !ok {
  // handle this somehow
}
r := revelReq.Original

以上就是《如何在 Revel 控制器中获取 http.ResponseWriter 和 http.Request》的详细内容，更多关于的资料请关注golang学习网公众号！