当前位置： > > > > 在 Go 中复用可以处理非 HTTP 协议的 HTTP 服务器的正确方法
来源：stackoverflow
2024-04-28 13:06:36
0浏览
收藏
学习知识要善于思考，思考，再思考！今天小编就给大家带来《在 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 服务器的连接请求时需要类似的用例。可以在这里找到一篇不错的文章：
理论要掌握，实操不能落！以上关于《在 Go 中复用可以处理非 HTTP 协议的 HTTP 服务器的正确方法》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注公众号吧！
