Method onWebsocket

pkg/proxy/proxy_websocket.go:71–116

(c *proxyContext, addr string)

}

func (p *Proxy) onWebsocket(c *proxyContext, addr string) (*fasthttp.Response, error) {
	resp := fasthttp.AcquireResponse()

	var r http.Request
	r.Method = "GET"
	r.Proto = "HTTP/1.1"
	r.ProtoMajor = 1
	r.ProtoMinor = 1
	r.RequestURI = string(c.forwardReq.RequestURI())
	r.Host = string(c.forwardReq.Host())

	hdr := make(http.Header)
	c.forwardReq.Header.VisitAll(func(k, v []byte) {
		sk := string(k)
		sv := string(v)
		switch sk {
		case "Transfer-Encoding":
			r.TransferEncoding = append(r.TransferEncoding, sv)
		default:
			hdr.Set(sk, sv)
		}
	})
	r.Header = hdr
	r.URL, _ = url.ParseRequestURI(r.RequestURI)

	wp := &websocketproxy.WebsocketProxy{
		Upgrader: &websocket.Upgrader{
			ReadBufferSize:  c.result.httpOption().ReadBufferSize,
			WriteBufferSize: c.result.httpOption().WriteBufferSize,
			CheckOrigin: func(r *http.Request) bool {
				return true
			},
		},
		Director: func(incoming *http.Request, out http.Header) {
			out.Set("Origin", fmt.Sprintf("http://%s", addr))
			out.Set("Host", addr)
		},
		Backend: func(r *http.Request) *url.URL {
			u, _ := url.Parse(fmt.Sprintf("ws://%s%s", addr, r.RequestURI))
			return u
		},
	}

	wp.ServeHTTP(c.GetAttr(websocketRspKey).(http.ResponseWriter), &r)
	return resp, nil
}