modelcontextprotocol/kotlin-sdk
Watch
1
0
Fork
You've already forked kotlin-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

Remove the double bang operator and replace it with a null check instead. (#49)

Browse source 
2. add try catch in the block for transport send in acatch block

Co-authored-by: Leonid Stashevsky <e5l@users.noreply.github.com>
This commit is contained in:
Adithya Kamath 2025-03-12 15:18:21 +05:30 • committed by GitHub
commit 7d2e993b96
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 19 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

32
src/commonMain/kotlin/io/modelcontextprotocol/kotlin/sdk/shared/Protocol.kt
View file

@ -209,7 +209,7 @@ public abstract class Protocol(
if (handler === null) {
LOGGER.trace { "No handler found for request: ${request.method}" }
try {
transport!!.send(
transport?.send(
JSONRPCResponse(
id = request.id,
error = JSONRPCError(
@ -229,24 +229,30 @@ public abstract class Protocol(
val result = handler(request, RequestHandlerExtra())
LOGGER.trace { "Request handled successfully: ${request.method} (id: ${request.id})" }
val response = JSONRPCResponse(
id = request.id,
result = result
transport?.send(
JSONRPCResponse(
id = request.id,
result = result
)
)
transport!!.send(response)
} catch (cause: Throwable) {
LOGGER.error(cause) { "Error handling request: ${request.method} (id: ${request.id})" }
transport!!.send(
JSONRPCResponse(
id = request.id,
error = JSONRPCError(
ErrorCode.Defined.InternalError,
message = cause.message ?: "Internal error",
try {
transport?.send(
JSONRPCResponse(
id = request.id,
error = JSONRPCError(
code = ErrorCode.Defined.InternalError,
message = cause.message ?: "Internal error"
)
)
)
)
} catch (sendError: Throwable) {
LOGGER.error(sendError) { "Failed to send error response for request: ${request.method} (id: ${request.id})" }
// Optionally implement fallback behavior here
}
}
}
@ -385,7 +391,7 @@ public abstract class Protocol(
try {
withTimeout(timeout) {
LOGGER.trace { "Sending request message with id: $messageId" }
this@Protocol.transport!!.send(message)
this@Protocol.transport?.send(message)
}
return result.await()
} catch (cause: TimeoutCancellationException) {