blob: a6ceb73ca41d018fec3077affd9fb0d044699c76 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
package app
import io.undertow.websockets.WebSocketConnectionCallback
import io.undertow.websockets.core.{AbstractReceiveListener, BufferedTextMessage, WebSocketChannel, WebSockets}
import io.undertow.websockets.spi.WebSocketHttpExchange
object Websockets extends cask.MainRoutes{
@cask.websocket("/connect/:userName")
def showUserProfile(userName: String): cask.WebsocketResult = {
if (userName != "haoyi") cask.Response("", statusCode = 403)
else new WebSocketConnectionCallback() {
override def onConnect(exchange: WebSocketHttpExchange, channel: WebSocketChannel): Unit = {
channel.getReceiveSetter.set(
new AbstractReceiveListener() {
override def onFullTextMessage(channel: WebSocketChannel, message: BufferedTextMessage) = {
message.getData match{
case "" => channel.close()
case data => WebSockets.sendTextBlocking(userName + " " + data, channel)
}
}
}
)
channel.resumeReceives()
}
}
}
initialize()
}
|