Fixed CommunicationServer.register.

outr · Oct 11, 2016 · 167e0df · 167e0df
1 parent 4321a57
commit 167e0df
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/core/jvm/src/main/scala/org/hyperscala/communication/server/CommunicationServer.scala b/core/jvm/src/main/scala/org/hyperscala/communication/server/CommunicationServer.scala
@@ -6,7 +6,7 @@ import io.undertow.websockets.WebSocketConnectionCallback
 import io.undertow.websockets.core.{AbstractReceiveListener, BufferedTextMessage, StreamSourceFrameChannel, WebSocketChannel, WebSockets}
 import io.undertow.websockets.spi.WebSocketHttpExchange
 import org.hyperscala.communication.Communication
-import org.hyperscala.Server
+import org.hyperscala.{Handler, Server}
 import pl.metastack.metarx.Channel
 
 class CommunicationServer(path: String,
@@ -53,6 +53,10 @@ class CommunicationServer(path: String,
     }
   }
 
+  def register(server: Server): Unit = {
+    Handler.pathMatch(path).withHandler(Handlers.websocket(this)).register(server)
+  }
+
   override def onConnect(exchange: WebSocketHttpExchange, channel: WebSocketChannel): Unit = synchronized {
     val authorized = authorization(exchange) match {
       case Some(auth) => exchange.getRequestHeader(Headers.AUTHORIZATION_STRING) == auth