com.firefly.kotlin.ext.websocket.WebSocketClientExtension.kt Maven / Gradle / Ivy

Go to download

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

There is a newer version: 5.0.0-dev6

package com.firefly.kotlin.ext.websocket

import com.firefly.client.websocket.SimpleWebSocketClient
import com.firefly.codec.websocket.stream.WebSocketConnection
import kotlinx.coroutines.future.await
import kotlinx.coroutines.withTimeout
import java.util.concurrent.TimeUnit

/**
 * @author Pengtao Qiu
 */
suspend fun SimpleWebSocketClient.HandshakeBuilder.asyncConnect(
    time: Long = 10 * 1000L,
    unit: TimeUnit = TimeUnit.MILLISECONDS
                                                               ): WebSocketConnection =
    withTimeout(unit.toMillis(time)) { connect().await() }