com.fireflysource.net.websocket.client.impl.AsyncWebSocketClientConnectionManager.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.client.impl

import com.fireflysource.common.`object`.Assert
import com.fireflysource.common.lifecycle.AbstractLifeCycle
import com.fireflysource.net.http.client.impl.Http1ClientConnection
import com.fireflysource.net.http.common.HttpConfig
import com.fireflysource.net.http.common.model.HttpURI
import com.fireflysource.net.tcp.TcpClientConnectionFactory
import com.fireflysource.net.tcp.aio.ApplicationProtocol.HTTP1
import com.fireflysource.net.websocket.client.WebSocketClientConnectionManager
import com.fireflysource.net.websocket.client.WebSocketClientRequest
import com.fireflysource.net.websocket.common.WebSocketConnection
import com.fireflysource.net.websocket.common.exception.WebSocketException
import com.fireflysource.net.websocket.common.model.WebSocketBehavior
import java.net.InetSocketAddress
import java.util.concurrent.CompletableFuture

/**
 * @author Pengtao Qiu
 */
class AsyncWebSocketClientConnectionManager(
    private val config: HttpConfig,
    private val connectionFactory: TcpClientConnectionFactory
) : WebSocketClientConnectionManager, AbstractLifeCycle() {

    init {
        start()
    }

    override fun connect(request: WebSocketClientRequest): CompletableFuture {
        Assert.hasText(request.url, "The websocket url must be not blank")
        Assert.notNull(request.policy, "The websocket policy must be not null")
        Assert.notNull(request.handler, "The websocket message handler must be not null")
        Assert.isTrue(request.policy.behavior == WebSocketBehavior.CLIENT, "The websocket behavior must be client")

        val uri = HttpURI(request.url)
        val inetSocketAddress = InetSocketAddress(uri.host, uri.port)
        val tcpConnection = when (uri.scheme) {
            "ws" -> connectionFactory.connect(inetSocketAddress, false)
            "wss" -> connectionFactory.connect(inetSocketAddress, true, listOf(HTTP1.value))
            else -> throw WebSocketException("The websocket scheme error. scheme: ${uri.scheme}")
        }

        return tcpConnection
            .thenCompose { connection -> connection.beginHandshake().thenApply { connection } }
            .thenApply { Http1ClientConnection(config, it) }
            .thenCompose { it.upgradeWebSocket(request) }
    }

    override fun init() {
        connectionFactory.start()
    }

    override fun destroy() {
        connectionFactory.stop()
    }

}