workaround connection closure for now

Author: whyoleg

rsocket-core/src/commonMain/kotlin/io/rsocket/kotlin/connection/ConnectionEstablishmentHandler.kt

@@ -101,7 +101,15 @@ internal abstract class ConnectionEstablishmentHandler(
         }
     }
 
-    override suspend fun RSocketConnection.initialize() = coroutineScope {
-        handleConnection(wrapConnection(this@initialize, coroutineContext.supervisorContext()))
+    override suspend fun initialize(connection: RSocketConnection) {
+        try {
+            coroutineScope {
+                handleConnection(wrapConnection(connection, coroutineContext.supervisorContext()))
+            }
+            connection.cancel()
+        } catch (cause: Throwable) {
+            connection.cancel("Connection closed", cause)
+            throw cause
+        }
     }
 }

rsocket-core/src/commonMain/kotlin/io/rsocket/kotlin/connection/RSocketConnectionInitializer.kt

@@ -21,13 +21,13 @@ import kotlinx.coroutines.*
 
 @RSocketTransportApi
 internal interface RSocketConnectionInitializer<T> {
-    public suspend fun RSocketConnection.initialize(): T
+    suspend fun initialize(connection: RSocketConnection): T
 }
 
 @RSocketTransportApi
 internal suspend fun <T> RSocketConnectionInitializer<T>.runInitializer(connection: RSocketConnection): T {
     val result = connection.async {
-        connection.initialize()
+        initialize(connection)
     }
     try {
         result.join()
@@ -41,6 +41,6 @@ internal suspend fun <T> RSocketConnectionInitializer<T>.runInitializer(connecti
 @RSocketTransportApi
 internal fun RSocketConnectionInitializer<Unit>.launchInitializer(connection: RSocketConnection): Job {
     return connection.launch {
-        connection.initialize()
+        initialize(connection)
     }
 }