fix: actually make it suspendable

d1snin · Oct 16, 2024 · 62ee5a0 · 62ee5a0
1 parent 1c4e0a7
commit 62ee5a0
Showing 1 changed file with 10 additions and 13 deletions.
diff --git a/...-events-client/src/commonMain/kotlin/dev/d1s/ktor/events/client/WebSocketEventReceiver.kt b/...-events-client/src/commonMain/kotlin/dev/d1s/ktor/events/client/WebSocketEventReceiver.kt
@@ -19,8 +19,6 @@ package dev.d1s.ktor.events.client
 import io.ktor.client.plugins.websocket.*
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.Dispatchers
-import kotlinx.coroutines.Job
-import kotlinx.coroutines.launch
 import org.lighthousegames.logging.KmLog
 import org.lighthousegames.logging.logging
 
@@ -41,16 +39,15 @@ public suspend inline fun <reified T> DefaultClientWebSocketSession.receiveWebSo
  *
  * @see webSocketEvents
  */
-public inline fun <reified T> DefaultClientWebSocketSession.receiveWebSocketEvents(crossinline receiver: suspend (ClientWebSocketEvent<T>) -> Unit): Job =
-    EventReceivingScope.launch {
-        withRetries(continuous = true, onError = {
-            EventReceiverLog.w {
-                "Error receiving web socket events: ${it.message}"
+public suspend inline fun <reified T> DefaultClientWebSocketSession.receiveWebSocketEvents(crossinline receiver: suspend (ClientWebSocketEvent<T>) -> Unit) {
+    withRetries(continuous = true, onError = {
+        EventReceiverLog.w {
+            "Error receiving web socket events: ${it.message}"
 
-                it.printStackTrace()
-            }
-        }) {
-            val event = receiveWebSocketEvent<T>()
-            receiver(event)
+            it.printStackTrace()
         }
-    }
+    }) {
+        val event = receiveWebSocketEvent<T>()
+        receiver(event)
+    }
+}