add flushIfNeeded

rsocket · Dec 10, 2024 · c42960b · c42960b
1 parent f9a1cf1
commit c42960b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.../ktor-tcp/src/commonMain/kotlin/io/rsocket/kotlin/transport/ktor/tcp/KtorTcpConnection.kt b/.../ktor-tcp/src/commonMain/kotlin/io/rsocket/kotlin/transport/ktor/tcp/KtorTcpConnection.kt
@@ -92,9 +92,10 @@ private class KtorTcpConnection(
 }
 
 @OptIn(InternalAPI::class)
-private fun ByteWriteChannel.writeFrame(frame: Buffer) {
+private suspend fun ByteWriteChannel.writeFrame(frame: Buffer) {
     writeBuffer.writeInt24(frame.size.toInt())
     writeBuffer.transferFrom(frame)
+    flushIfNeeded()
 }
 
 @OptIn(InternalAPI::class)