commonJvmMain.com.ditchoom.socket.nio.util.SelectorExtensions.kt Maven / Gradle / Ivy

Go to download
@file:OptIn(ExperimentalCoroutinesApi::class)

package com.ditchoom.socket.nio.util

import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.ExperimentalCoroutinesApi
import kotlinx.coroutines.isActive
import kotlinx.coroutines.withContext
import kotlinx.coroutines.withTimeout
import java.nio.channels.AsynchronousCloseException
import java.nio.channels.Selector
import kotlin.coroutines.resume
import kotlin.coroutines.resumeWithException
import kotlin.coroutines.suspendCoroutine
import kotlin.time.Duration

suspend fun Selector.aSelect(timeout: Duration): Int {
    val selector = this
    return withTimeout(timeout) {
        withContext(Dispatchers.IO.limitedParallelism(1)) {
            while (isActive) {
                try {
                    val keys = selectNow()
                    if (keys > 0) {
                        return@withContext keys
                    }
                } catch (e: Throwable) {
                    if (e is AsynchronousCloseException) {
                        selector.close()
                        return@withContext 0
                    }
                }
            }
            0
        }
    }
}

suspend fun Selector.aClose() =
    suspendCoroutine {
        try {
            it.resume(close())
        } catch (e: Throwable) {
            it.resumeWithException(e)
        }
    }