[DCAE] INFO.yaml update

[dcaegen2/collectors/hv-ves.git] / sources / hv-collector-utils / src / main / kotlin / org / onap / dcae / collectors / veshv / utils / server_handle.kt

diff --git a/sources/hv-collector-utils/src/main/kotlin/org/onap/dcae/collectors/veshv/utils/server_handle.kt b/sources/hv-collector-utils/src/main/kotlin/org/onap/dcae/collectors/veshv/utils/server_handle.kt
index 670ab4a..9c0a6e1 100644 (file)
--- a/sources/hv-collector-utils/src/main/kotlin/org/onap/dcae/collectors/veshv/utils/server_handle.kt
+++ b/sources/hv-collector-utils/src/main/kotlin/org/onap/dcae/collectors/veshv/utils/server_handle.kt
@@ -19,17 +19,17 @@
  */
 package org.onap.dcae.collectors.veshv.utils
 
-import arrow.effects.IO
 import org.onap.dcae.collectors.veshv.utils.logging.Logger
 import reactor.core.publisher.Mono
 import reactor.netty.DisposableServer
+import java.time.Duration
 
 /**
  * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
  * @since August 2018
  */
 abstract class ServerHandle(val host: String, val port: Int) : Closeable {
-    abstract fun await(): IO<Unit>
+    abstract fun await(): Mono<Void>
 }
 
 /**
@@ -49,20 +49,25 @@ class NettyServerHandle(private val ctx: DisposableServer,
                     .then(dispose())
 
     private fun dispose(): Mono<Void> =
-            Mono.create { callback ->
+            Mono.create<Int> { callback ->
                 logger.debug { "About to dispose NettyServer" }
                 ctx.dispose()
                 ctx.onDispose {
                     logger.debug { "Netty server disposed" }
-                    callback.success()
+                    callback.success(1)
                 }
             }
+                    .delayElement(boundPortReleaseLatency)
+                    .then()
 
-    override fun await() = IO<Unit> {
-        ctx.channel().closeFuture().sync()
+    override fun await(): Mono<Void> = Mono.create { callback ->
+        ctx.channel().closeFuture().addListener {
+            callback.success()
+        }
     }
 
     companion object {
         private val logger = Logger(NettyServerHandle::class)
+        private val boundPortReleaseLatency = Duration.ofSeconds(1)
     }
 }