com.github.psxpaul.util.PortUtils.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gradle-execfork-plugin Show documentation

A gradle plugin to execute applications in the background

There is a newer version: 0.1.4

Show newest version

package com.github.psxpaul.util

import org.gradle.api.GradleException
import java.net.*
import java.util.concurrent.TimeUnit

fun findOpenPort():Int {
    ServerSocket(0).use { return it.localPort }
}

private fun isPortOpen(port:Int):Boolean {
    Socket().use {
        val inetAddress: InetAddress = InetAddress.getByName("127.0.0.1")
        val socketAddress: InetSocketAddress = InetSocketAddress(inetAddress, port)
        try {
            it.connect(socketAddress)
            return true;
        } catch (e: ConnectException) {
            return false
        }
    }
}

fun waitForPortOpen(port:Int, timeout:Long, unit: TimeUnit, process:Process) {
    val millisToWait:Long = unit.toMillis(timeout)
    val waitUntil:Long = System.currentTimeMillis() + millisToWait

    while (System.currentTimeMillis() < waitUntil) {
        Thread.sleep(100)
        if (!process.isAlive) throw GradleException("Process died before port $port was opened")
        if (isPortOpen(port)) return
    }

    throw GradleException("Timed out waiting for port $port to be opened")
}