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

Question

我使用 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 协议(protocol)，请求将回放到原始目的地(通过 http.Request.Write())，大家都很高兴。

至少，大多数时候。问题出现在边缘情况下。 Tivo 客户端不发送“主机” header ，并且似乎不喜欢 Go 执行的所有其他标准操作(例如大写 header 名称)。可能的变化数量是无穷无尽的，所以我非常想做的是缓冲原始请求——完全按照它未经任何修改发送给我的请求——并将其重播回原始目标服务器。

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

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 解析之前围绕这一点 Hook 连接？

Answer 1

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

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

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

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