如果❤️我的文章有帮助,欢迎点赞、关注。这是对我继续技术创作最大的鼓励。更多系列文章在我博客[1]
目录
初探 Golang ReverseProxy 源码
Golang 版本 1.16
使用 ReverseProxy 实现反向代理的例子
♀️不废话, 直接举例
测试 代理服务器
$ curl :2002/sda?sda=111 # 2003 设置了 打印请求地址代码: upath := fmt.Sprintf("http://%s%s\n", r.Addr, req.URL.Path) http://127.0.0.1:2003/base/sda RemoteAddr=127.0.0.1:51738,X-Forwarded-For=127.0.0.1,X-Real-Ip= headers =map[Accept:[*/*] Accept-Encoding:[gzip] User-Agent:[curl/7.69.1] X-Forwarded-For:[127.0.0.1]]查看 httputil.NewSingleHostReverseProxy() 函数, 便可以找到 ReverseProxy源码。 位于该路径下文件go/src/net/http/httputil/reverseproxy.go
核心源码
ReverseProxy 结构体
ReverseProxy 处理请求的方法 ServeHTTP()
ReverseProxy 实现了ServeHTTP() 方法, 最终会在请求到达代理服务器, 由监听方法ListenAndServe()调起。调用链路为:
http.ListenAndServe(addr string, handler Handler) -> Server.ListenAndServeTLS(certFile, keyFile) -> Server.ServeTLS(ln, certFile, keyFile) -> Server.Serve(l net.Listener) -> go c.serve(connCtx) -> serverHandler{c.server}.ServeHTTP(w, w.req)所以 ReverseProxy 结构体也实现了 ServeHTTP 方法 , 方法实现功能有:
1.拷贝上游请求的 context 到下游请求2.使用 指定 director(对请求进行修改的函数)修改请求(例如协议、参数、url 等)3.根据请求 Header["Connection"] 判断是否需要升级协议(Upgrade)4.删除上游请求中的 hop-by-hop Header, 维持上游持久(相对)连接, 不需要透传到下游5.设置 X-Forward-For Header,追加当前节点 IP6.使用连接池,向下游发起请求7.处理 httpcode 101 协议升级:(WebSocket、h2c 等)8.删除请求中的 hop-by-hop Header, 不要返回给上游9.根据结构体 ReverseProxy.ModifyResponse(函数)判断是否修改响应体内容10.拷贝下游响应头部到上游响应请求11.返回 下游请求 HTTP 状态码12.拷贝 下游响应内容 到 上游响应请求13.刷新内容到 response
其他源码
NewSingleHostReverseProxy 函数
NewSingleHostReverseProxy 中 URL 拼接方法
References
[1] 更多系列文章在我博客: