如何实现 wss 反向代理作为 gin 路由？

知识点掌握了，还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战，手把手教大家学习《如何实现 wss 反向代理作为 gin 路由？》，在实现功能的过程中也带大家重新温习相关知识点，温故而知新，回头看看说不定又有不一样的感悟！

我最近才开始用 go 编码，因此还不太熟练。

我有一个用例，我想将反向代理实现为 gin 路由。 所以我的路线如下：

server.router.post("/console", server.proxyconsoleurl)

我的处理程序功能如下：

func (server *server) proxyconsoleurl(ctx *gin.context) {
    director := func(req *http.request) {
        r := ctx.request
        // this is not working, scheme wss is not supported
        req.url.scheme = "wss"
        req.url.host = "192.168.******:8006"
        // the path which gets proxied should be overriden
        req.url.rawpath = "/api2/json/nodes/something/qemu/123/vncwebsocket?port=5900&vncticket=something"
        req.header["my-header"] = []string{r.header.get("my-header")}
        // golang camelcases headers
        delete(req.header, "my-header")
        // this header has to be added to every request which gets proxied
        req.header["authorization"] = []string{"mycustomheader"}
    }
    proxy := &httputil.reverseproxy{director: director, transport: &http.transport{
        proxy: http.proxyfromenvironment,
        dial: (&net.dialer{
            timeout:   30 * time.second,
            keepalive: 30 * time.second,
        }).dial,
        tlshandshaketimeout: 10 * time.second,
        tlsclientconfig:     &tls.config{insecureskipverify: true},
    }}
    proxy.servehttp(ctx.writer, ctx.request)
}

所以我的第一个问题是，据我所知并在运行代码时注意到，httputil.reverseproxy 不支持 web 套接字：

httputil: unsupported protocol scheme "wss"

第二个问题是，我想覆盖后端 url 并添加由代理添加的自定义标头。

也许有人有一个想法来实现这个，如果可能的话。 -提前致谢

正确答案

go 版本 1.12 中的 httputil.reverseproxy 添加了 websocket 支持。

使用 url.parse("https://192.168.******:8006/api2/json/nodes/something/qemu/123/vncwebsocket?port=5900&vncticket=something") 的结果设置目标 url。这修复了以下问题：

• websocket 协议在线上使用“http”或“https”，而不是“wss”。
• 当 rawpath 不是有效的路径转义时，rawpath 将被忽略。详情见EscapedPath。由于问题中的 rawpath 包含查询字符串，因此它永远不会是 path 的有效转义。客户端的路径始终按原样使用。客户端的查询字符串也是如此。

创建一次代理并重复使用它。重要的一点是根据 Transport 文档创建重复使用单个传输。重用代理可以实现该目标。

func createProxy() *httputil.ReverseProxy {
    target, _ := url.Parse("https://192.168.******:8006/api2/json/nodes/something/qemu/123/vncwebsocket?port=5900&vncticket=something")
    director := func(req *http.Request) {
        req.URL = target
        req.Header["my-header"] = []string{req.Header.Get("my-header")}
        delete(req.Header, "My-Header")
        req.Header["Authorization"] = []string{"MYCUSTOMHEADER"}
    }
    return &httputil.ReverseProxy{Director: director, Transport: &http.Transport{
        Proxy: http.ProxyFromEnvironment,
        Dial: (&net.Dialer{
            Timeout:   30 * time.Second,
            KeepAlive: 30 * time.Second,
        }).Dial,
        TLSHandshakeTimeout: 10 * time.Second,
        TLSClientConfig:     &tls.Config{InsecureSkipVerify: true},
    }}
}

var proxy = createProxy()

func (server *Server) proxyConsoleUrl(ctx *gin.Context) {
    proxy.ServeHTTP(ctx.Writer, ctx.Request)
}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《如何实现 wss 反向代理作为 gin 路由？》文章吧，也可关注golang学习网公众号了解相关技术文章。