aboutsummaryrefslogtreecommitdiff
path: root/ws.go
diff options
context:
space:
mode:
authorDimitri Sokolyuk <demon@dim13.org>2016-04-04 14:47:27 +0200
committerDimitri Sokolyuk <demon@dim13.org>2016-04-04 14:47:27 +0200
commita09a47d1f8812e837080d5af9461113109555b23 (patch)
treeee6c98582c657a895ee7b0b9bb5ddbc40083a0e2 /ws.go
parent1ef50ddd7f6ae07c68fec7c87c7052e4004179ea (diff)
Split files
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}
-}