Skip to content

Commit 449dd47

Browse files
committed
Fix: Memory leak from unclosed sessions in Server
1 parent 458a49a commit 449dd47

File tree

1 file changed

+6
-0
lines changed
  • kotlin-sdk-server/src/commonMain/kotlin/io/modelcontextprotocol/kotlin/sdk/server

1 file changed

+6
-0
lines changed

kotlin-sdk-server/src/commonMain/kotlin/io/modelcontextprotocol/kotlin/sdk/server/Server.kt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,12 @@ public open class Server(
152152
logger.debug { "Server session successfully connected to transport" }
153153
sessions.update { it.add(session) }
154154

155+
// Register cleanup handler to remove session from list when it closes
156+
session.onClose {
157+
logger.debug { "Removing closed session from active sessions list" }
158+
sessions.update { list -> list - session }
159+
}
160+
155161
_onConnect()
156162
return session
157163
}

0 commit comments

Comments
 (0)