fix: memory leak

StarlaneStudios · Jun 30, 2023 · 0f7873e · 0f7873e
1 parent 5af1930
commit 0f7873e
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/src/main/kotlin/org/starlane/graphqlwskt/client/ClientHandler.kt b/src/main/kotlin/org/starlane/graphqlwskt/client/ClientHandler.kt
@@ -56,8 +56,6 @@ internal class ClientHandler(
 			return
 		}
 
-//		println("<- ${MessageParser.toJson(msg)}")
-
 		when(msg) {
 			is ConnectionAck -> {
 				acknowledged = true
@@ -93,10 +91,12 @@ internal class ClientHandler(
 
 	override fun onClose(code: Int, reason: String?, remote: Boolean) {
 		scope.cancel()
+		subscriptions.clear()
 	}
 
 	override fun onError(ex: Exception?) {
 		scope.cancel()
+		subscriptions.clear()
 	}
 
 	fun <T : Any> subscribe(request: OperationRequest, type: KClass<T>): Flow<T> {
@@ -109,21 +109,16 @@ internal class ClientHandler(
 				},
 				onComplete = {
 					close()
+					subscriptions.remove(id)
 				},
 				onError = {
 					close(it)
+					subscriptions.remove(id)
 				}
 			)
 
 			send(Subscribe(id, request))
 			awaitClose()
 		}
 	}
-
-//	override fun send(text: String) {
-//		println("-> $text")
-//
-//		super.send(text)
-//	}
-
 }