aboutsummaryrefslogtreecommitdiff
path: root/ws.go
diff options
context:
space:
mode:
Diffstat (limited to 'ws.go')
-rw-r--r--ws.go16
1 files changed, 8 insertions, 8 deletions
diff --git a/ws.go b/ws.go
index b1cb1a3..b07a17b 100644
--- a/ws.go
+++ b/ws.go
@@ -11,6 +11,14 @@ type WebSocketProxy struct {
Director func(*http.Request)
}
+func NewWebSocketProxy(target *url.URL) *WebSocketProxy {
+ director := func(req *http.Request) {
+ req.URL.Scheme = target.Scheme
+ req.URL.Host = target.Host
+ }
+ return &WebSocketProxy{Director: director}
+}
+
func (p *WebSocketProxy) ServeHTTP(w http.ResponseWriter, r *http.Request) {
p.Director(r)
d, err := net.Dial("tcp", r.URL.Host)
@@ -48,11 +56,3 @@ func (p *WebSocketProxy) ServeHTTP(w http.ResponseWriter, r *http.Request) {
go cp(conn, d)
<-errc
}
-
-func NewWebSocketProxy(target *url.URL) *WebSocketProxy {
- director := func(req *http.Request) {
- req.URL.Scheme = target.Scheme
- req.URL.Host = target.Host
- }
- return &WebSocketProxy{Director: director}
-}