From 7ef66c57dc12b45ba3e0e8edcecc060cfd2c126f Mon Sep 17 00:00:00 2001 From: Shelikhoo Date: Sat, 13 Aug 2016 21:43:46 +0800 Subject: [PATCH] Support Listen WebSocket Connection --- transport/internet/tcp_hub.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/transport/internet/tcp_hub.go b/transport/internet/tcp_hub.go index d0642de7e..82daa82dc 100644 --- a/transport/internet/tcp_hub.go +++ b/transport/internet/tcp_hub.go @@ -17,6 +17,7 @@ var ( KCPListenFunc ListenFunc TCPListenFunc ListenFunc RawTCPListenFunc ListenFunc + WSListenFunc ListenFunc ) type ListenFunc func(address v2net.Address, port v2net.Port) (Listener, error) @@ -42,6 +43,8 @@ func ListenTCP(address v2net.Address, port v2net.Port, callback ConnectionHandle listener, err = TCPListenFunc(address, port) case settings.IsCapableOf(StreamConnectionTypeKCP): listener, err = KCPListenFunc(address, port) + case settings.IsCapableOf(StreamConnectionTypeWebSocket): + listener, err = WSListenFunc(address, port) case settings.IsCapableOf(StreamConnectionTypeRawTCP): listener, err = RawTCPListenFunc(address, port) default: