aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--server.go4
1 files changed, 2 insertions, 2 deletions
diff --git a/server.go b/server.go
index 437df4d..c0449ee 100644
--- a/server.go
+++ b/server.go
@@ -78,9 +78,9 @@ func (s *Server) Restore() error {
}
switch up.Scheme {
case "ws":
- mux.Handle(v.ServerName, httputil.NewSingleHostReverseProxy(up))
+ mux.Handle(v.ServerName, NewWebSocketProxy(up))
default:
- mux.Handle(v.ServerName, NewReverseProxy(up))
+ mux.Handle(v.ServerName, httputil.NewSingleHostReverseProxy(up))
}
}
s.Server.Handler = mux