com.fireflysource.net.websocket.client.impl.AsyncWebSocketClientConnectionBuilder.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.net.websocket.client.WebSocketClientConnectionBuilder
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.WebSocketMessageHandler
import com.fireflysource.net.websocket.common.model.WebSocketBehavior
import com.fireflysource.net.websocket.common.model.WebSocketPolicy
import java.util.concurrent.CompletableFuture

/**
 * @author Pengtao Qiu
 */
class AsyncWebSocketClientConnectionBuilder(
    private val connectionManager: WebSocketClientConnectionManager
) : WebSocketClientConnectionBuilder {

    private val request = WebSocketClientRequest()

    override fun url(url: String): WebSocketClientConnectionBuilder {
        request.url = url
        return this
    }

    override fun policy(policy: WebSocketPolicy): WebSocketClientConnectionBuilder {
        request.policy = policy
        return this
    }

    override fun extensions(extensions: List): WebSocketClientConnectionBuilder {
        request.extensions = extensions
        return this
    }

    override fun subProtocols(subProtocols: List): WebSocketClientConnectionBuilder {
        request.subProtocols = subProtocols
        return this
    }

    override fun onMessage(handler: WebSocketMessageHandler): WebSocketClientConnectionBuilder {
        request.handler = handler
        return this
    }

    override fun connect(): CompletableFuture {
        if (request.policy == null) {
            request.policy = WebSocketPolicy(WebSocketBehavior.CLIENT)
        }
        if (request.subProtocols == null) {
            request.subProtocols = listOf()
        }
        if (request.extensions == null) {
            request.extensions = listOf()
        }
        return connectionManager.connect(request)
    }

}