Check FOREVER in operators with Duration

Fix Kotlin#868

Author: fvasco

integration/kotlinx-coroutines-jdk8/src/time/Time.kt

@@ -4,30 +4,45 @@
 package kotlinx.coroutines.time
 
 import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.selects.SelectBuilder
 import java.time.Duration
-import java.util.concurrent.TimeUnit
+import java.time.temporal.ChronoUnit
+import kotlin.coroutines.suspendCoroutine
 
 /**
  * "java.time" adapter method for [kotlinx.coroutines.delay]
  */
-public suspend fun delay(duration: Duration) =
-    kotlinx.coroutines.delay(duration.toMillis())
+suspend fun delay(duration: Duration) {
+    when {
+        duration.seconds < Long.MAX_VALUE / 1_000L -> kotlinx.coroutines.delay(duration.toMillis())
+        duration == ChronoUnit.FOREVER.duration -> suspendCoroutine<Unit> { }
+        else -> {
+            kotlinx.coroutines.delay(duration.seconds * 1_000L)
+            kotlinx.coroutines.delay(duration.nano / 1_000_000L)
+        }
+    }
+}
 
 /**
  * "java.time" adapter method for [SelectBuilder.onTimeout]
  */
-public fun <R> SelectBuilder<R>.onTimeout(duration: Duration, block: suspend () -> R) =
-    onTimeout(duration.toMillis(), block)
+public fun <R> SelectBuilder<R>.onTimeout(duration: Duration, block: suspend () -> R) {
+    if (duration != ChronoUnit.FOREVER.duration) {
+        onTimeout(duration.toMillis(), block)
+    }
+}
 
 /**
  * "java.time" adapter method for [kotlinx.coroutines.withTimeout]
  */
 public suspend fun <T> withTimeout(duration: Duration, block: suspend CoroutineScope.() -> T): T =
-    kotlinx.coroutines.withTimeout(duration.toMillis(), block)
+        if (duration == ChronoUnit.FOREVER.duration) coroutineScope { block() }
+        else kotlinx.coroutines.withTimeout(duration.toMillis(), block)
 
 /**
  * "java.time" adapter method for [kotlinx.coroutines.withTimeoutOrNull]
  */
 public suspend fun <T> withTimeoutOrNull(duration: Duration, block: suspend CoroutineScope.() -> T): T? =
-    kotlinx.coroutines.withTimeoutOrNull(duration.toMillis(), block)
+        if (duration == ChronoUnit.FOREVER.duration) coroutineScope { block() }
+        else kotlinx.coroutines.withTimeoutOrNull(duration.toMillis(), block)