com.hexagonkt.http.server.netty.NettyWebSocketHandler.kt Maven / Gradle / Ivy

Go to download
package com.hexagonkt.http.server.netty

import com.hexagonkt.http.model.ws.CloseStatus
import com.hexagonkt.http.model.ws.WsCloseStatus
import com.hexagonkt.http.server.model.ws.WsServerSession
import io.netty.channel.ChannelHandlerContext
import io.netty.channel.ChannelInboundHandlerAdapter
import io.netty.handler.codec.http.websocketx.*

internal class NettyWebSocketHandler(
    private val session: WsServerSession,
    private val onBinary: WsServerSession.(data: ByteArray) -> Unit = {},
    private val onText: WsServerSession.(text: String) -> Unit = {},
    private val onPing: WsServerSession.(data: ByteArray) -> Unit = {},
    private val onPong: WsServerSession.(data: ByteArray) -> Unit = {},
    private val onClose: WsServerSession.(status: WsCloseStatus, reason: String) -> Unit = { _, _ -> },
) : ChannelInboundHandlerAdapter() {

    override fun channelRead(context: ChannelHandlerContext, message: Any) {
        if (message !is WebSocketFrame)
            return

        val content = message.content()
        when (message) {
            is BinaryWebSocketFrame -> session.onBinary(content.retain().array())
            is TextWebSocketFrame -> session.onText(message.text())
            is PingWebSocketFrame -> session.onPing(content.retain().array())
            is PongWebSocketFrame -> session.onPong(content.retain().array())
            is CloseWebSocketFrame -> {
                // TODO Close the channel?
                session.onClose(CloseStatus.valueOf(message.statusCode()), message.reasonText())
            }
            else -> error("Unsupported WebSocketFrame")
        }
    }
}