在 Go 中复用可以处理非 HTTP 协议的 HTTP 服务器的正确方法

学习知识要善于思考，思考，再思考！今天golang学习网小编就给大家带来《在 Go 中复用可以处理非 HTTP 协议的 HTTP 服务器的正确方法》，以下内容主要包含等知识点，如果你正在学习或准备学习Golang，就都不要错过本文啦~让我们一起来看看吧，能帮助到你就更好了！

我使用 go 构建了一个具有扩展日志记录功能的路由器。它适用于大多数用例。但是，当客户端在端口 80 上发送非标准 http 消息时，它会遇到问题。

到目前为止，我已经通过实现我自己的 servehttp() 版本解决了这个问题：

func (myproxy *myproxy) servehttp(w http.responsewtier, r *http.request) {

  // inspect headers
  // determine if it is a custom protocol (ie: websockets, connect requests)
  // implement handlers for each time
}

如果我确定请求是非标准 http 协议，则该请求将回放到原始目的地（通过 http.request.write()），每个人都很高兴。

至少，大多数时候。边缘情况会出现问题。 tivo 客户端不发送“host”标头，并且似乎不喜欢 go 所做的所有其他标准操作（例如大写标头名称）。可能的变化数量是无穷无尽的，所以我非常想做的是缓冲原始请求 - 与发送给我的请求完全一样，没有任何修改 - 并将其重播回原始目标服务器。

我可以通过重新实现 net.http.server 来解决这个问题，但这似乎是一种非常困难和脆弱的方法。我更愿意做的是以某种方式挂钩到 net/http/http/server.go 接收连接的点周围，然后将其包装在间谍包装器中，将请求记录到缓冲区。

func (srv *Server) Serve(l net.Listener) error {

        // Lots of listener code...

    c := srv.newConn(rw)
    c.setState(c.rwc, StateNew) // before Serve can return

        // I'd like to wrap c here and save the request for possible reply later.
    go c.serve(ctx)
}

https://golang.org/src/net/http/server.go?s=93229:93284#l2805

编辑：我查看了 httputil:dumprequest 但这稍微修改了原始请求（更改了标头的大小写和顺序）。如果它不这样做，这将是一个理想的解决方案。

https://godoc.org/net/http/httputil#dumprequest

有没有办法在 http.request 解析连接之前围绕这一点挂钩连接？

解决方案

为了帮助别人，我想回答我的问题。我上面建议的方法确实有效，并且是执行此操作的正确方法。总结一下：

• 实现 ListenAndServe()
• 将传入的 net.Conn 包装在 TeeReader 或其他多路复用连接包装器中
• 记录请求
• 拨打原始目的地并连接入站连接，必要时重放原始请求。

升级 Websockets 服务器的连接请求时需要类似的用例。可以在这里找到一篇不错的文章：

https://medium.freecodecamp.org/million-websockets-and-go-cc58418460bb

理论要掌握，实操不能落！以上关于《在 Go 中复用可以处理非 HTTP 协议的 HTTP 服务器的正确方法》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！