Correctly close zlib stream when gateway disconnects (#849)

gateway/src/commonMain/kotlin/DefaultGateway.kt

@@ -195,6 +195,8 @@ public class DefaultGateway(private val data: DefaultGatewayData) : Gateway {
     }
 
     private suspend fun handleClose() {
+        inflater.close()
+
         val reason = withTimeoutOrNull(1500) {
             socket.closeReason.await()
         } ?: return

gateway/src/commonMain/kotlin/Inflater.kt

@@ -1,7 +1,8 @@
 package dev.kord.gateway
 
+import io.ktor.utils.io.core.*
 import io.ktor.websocket.*
 
-internal expect class Inflater() {
+internal expect class Inflater() : Closeable {
     fun Frame.inflateData(): String
 }

gateway/src/jsMain/kotlin/Inflater.kt

@@ -1,16 +1,19 @@
 package dev.kord.gateway
 
 import dev.kord.gateway.internal.Inflate
+import io.ktor.utils.io.core.*
 import io.ktor.websocket.*
 import node.buffer.Buffer
 import node.buffer.BufferEncoding
 
-internal actual class Inflater {
+internal actual class Inflater : Closeable  {
     private val inflate = Inflate()
 
     actual fun Frame.inflateData(): String {
         val buffer = Buffer.from(data)
 
         return inflate.process(buffer).toString(BufferEncoding.utf8)
     }
+
+    override fun close() = inflate.close()
 }

gateway/src/jsMain/kotlin/internal/JsInflater.kt

@@ -8,4 +8,6 @@ import node.buffer.Buffer
 
 internal external class Inflate {
     fun process(data: Uint8Array): Buffer
+
+    fun close()
 }

gateway/src/jvmMain/kotlin/Inflater.kt

@@ -1,10 +1,12 @@
 package dev.kord.gateway
 
+import io.ktor.utils.io.core.*
 import io.ktor.websocket.*
 import java.io.ByteArrayOutputStream
 import java.util.zip.InflaterOutputStream
+import kotlin.io.use
 
-internal actual class Inflater {
+internal actual class Inflater : Closeable {
     private val delegate = java.util.zip.Inflater()
 
     actual fun Frame.inflateData(): String {
@@ -15,4 +17,6 @@ internal actual class Inflater {
 
         return outputStream.use { it.toByteArray().decodeToString() }
     }
+
+    override fun close() = delegate.end()
 }