golang 反向代理 Gin框架反向代理详解

前言

想用gin框架做一个反向代理服务，搜索了一圈，全是只讲解些皮毛的帖子，今天我就总结一下gin做反向代理的详细操作和原理

正文

开始之前我们先了解一些前置知识

gin的通配符

gin的动态路由可以自行查阅，我这里指列出两个通配符：冒号( ：)、星号（ * )

冒号通配符

只能匹配到一层路由

// 路由请求
router.GET("shuiche/:name",  handle)

// 请求结果匹配
/shuiche/query     匹配
/shuiche/invoke    匹配
/shuiche/v1/find    不匹配
/shuiche/            不匹配

星号通配符

能匹配到多层路由

// 路由请求
router.GET("shuiche/:name",  handle)

// 请求结果匹配
/shuiche/query     匹配
/shuiche/invoke    匹配
/shuiche/v1/find   匹配
/shuiche/          匹配

反向代理

概念：

反向代理是什么？有个很棒的说法是流量转发。服务端获取到客户端来的请求，将它发往另一个服务器，从另外一个服务器那里获取到响应再回给原先的客户端。反向的意义简单来说在于这个代理自身决定了何时将流量发往何处。

最小案例

我们先看一下最简单的代理案例

// router路由
router.Any("/proxy/*name", proxyHandler)

// api函数
func proxyHandler(c *gin.Context) {
	var target = "http://127.0.0.1:8081"
	proxyUrl, _ := url.Parse(target)
	proxy := httputil.NewSingleHostReverseProxy(proxyUrl)

	proxy.ServeHTTP(c.Writer, c.Request)
}

http://127.0.0.1:8080/proxy/test

代理函数这里只做了一个事情：就是把 http://127.0.0.1:8080/proxy/test 原样发送给了 8081。

http://127.0.0.1:8081/proxy/test

截取路由转发

有时候，我们的项目接口比较多，有些本地api，也有代理api，这时候就会给代理专门加一个前缀，例如：/proxy, 服务端拿到 /proxy 的路由请求全部做转发，这时候上面的案例就不能满足需求了，因为「最小案例」在做转发时会将 /proxy 原样转发到被代理的服务器上，这样就会报 404 错误。这时我们就需要将 /proxy 前缀去掉，看案例

// router路由
router.Any("/proxy/*name", proxyHandler)

// api函数
func proxyHandler(c *gin.Context) {
	var target = "http://127.0.0.1:8081"
	proxyUrl, _ := url.Parse(target)

	c.Request.URL.Path = c.Param("name")  //  重点是这行代码

        proxy := httputil.NewSingleHostReverseProxy(proxyUrl)

	proxy.ServeHTTP(c.Writer, c.Request)
}

c.Request.URL.Path = c.Param("name")*name

修改请求信息

有时候，我们在做代理时，需要增加一些新的请求头，比如增加鉴权 token 等字段到http的header里面

// router路由
router.Any("/proxy/*name", proxyHandler)

// api函数
func proxyHandler(c *gin.Context) {
	var target = "http://127.0.0.1:8081"
	proxyUrl, _ := url.Parse(target)

	c.Request.URL.Path = c.Param("name") 
	c.Request.Header.Set("token", "token_123") //  重点是这行代码

        proxy := httputil.NewSingleHostReverseProxy(proxyUrl)

	proxy.ServeHTTP(c.Writer, c.Request)
}

依然是只直接修改本来的请求内容，修改后再用 NewSingleHostReverseProxy 做代理，即可

服务器获取被代理服务的返回内容

另外一种写法

httputil.NewSingleHostReverseProxyhttputil.ReverseProxy

// router路由
router.Any("/proxy/*name", proxyHandler)


// api函数
func proxyHandler(c *gin.Context) {
        var target = "http://127.0.0.1:8081"
	director := func(req *http.Request) {
		req.URL.Scheme = "http"
		req.URL.Host = target
		req.Host = target
		req.URL.Path = c.Param("proxyPath")
	}
	proxy := &httputil.ReverseProxy{Director: director}
	proxy.ServeHTTP(c.Writer, c.Request)
}

directordirector

后记

转发代理其实很简单，就好比：我让你帮我把一个物品给他这么简单，你就是代理服务器，物品在过你手时，你可以对物品做二次处理，当然，他在把物品还回来的时候，你也可以拦截做二次处理。

如果你还有什么疑问，可以留言一起讨论

博客中所涉及到的图片都有版权，请谨慎使用