com.fireflysource.net.websocket.server.impl.AsyncWebSocketManager.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of firefly-net Show documentation

There is a newer version: 5.0.2

package com.fireflysource.net.websocket.server.impl

import com.fireflysource.common.`object`.Assert
import com.fireflysource.net.websocket.common.model.WebSocketBehavior
import com.fireflysource.net.websocket.server.WebSocketManager
import com.fireflysource.net.websocket.server.WebSocketServerConnectionHandler

/**
 * @author Pengtao Qiu
 */
class AsyncWebSocketManager : WebSocketManager {

    private val webSocketHandlers: MutableMap = HashMap()

    override fun register(connectionHandler: WebSocketServerConnectionHandler) {
        Assert.notNull(connectionHandler.url, "The websocket url must be not null")
        Assert.notNull(connectionHandler.extensionSelector, "The websocket extension selector must be not null")
        Assert.notNull(connectionHandler.subProtocolSelector, "The websocket sub protocol selector must be not null")
        Assert.notNull(connectionHandler.policy, "The websocket policy must be not null")
        Assert.notNull(connectionHandler.connectionListener, "The websocket connection listener must be not null")
        Assert.notNull(connectionHandler.messageHandler, "The websocket message handler must be not null")
        Assert.isTrue(
            connectionHandler.policy.behavior == WebSocketBehavior.SERVER,
            "The websocket behavior must be server"
        )

        webSocketHandlers[connectionHandler.url] = connectionHandler
    }

    override fun findWebSocketHandler(path: String): WebSocketServerConnectionHandler? {
        return webSocketHandlers[path]
    }
}