handle BrokenResourceError gracefully

lingster · lingster · commit a5b633e0a3de · 2025-03-01T22:45:59.000+01:00
diff --git a/src/mcp/server/stdio.py b/src/mcp/server/stdio.py
@@ -67,16 +67,27 @@ async def stdin_reader():
                         continue
 
                     await read_stream_writer.send(message)
+        except anyio.BrokenResourceError as ex:
+            print(f"BrokenResourceError(reader): {ex}")
+            pass
         except anyio.ClosedResourceError:
             await anyio.lowlevel.checkpoint()
+        except Exception as ex:
+            print(f"read error: {ex}")
 
     async def stdout_writer():
         try:
             async with write_stream_reader:
                 async for message in write_stream_reader:
-                    json = message.model_dump_json(by_alias=True, exclude_none=True)
-                    await stdout.write(json + "\n")
-                    await stdout.flush()
+                    try:
+                        json = message.model_dump_json(by_alias=True, exclude_none=True)
+                        await stdout.write(json + "\n")
+                        await stdout.flush()
+                    except Exception as ex:
+                        print(f"problem with writer: {ex}")
+        except anyio.BrokenResourceError as ex:
+            print(f"BrokenResourceError(writer): {ex}")
+            pass
         except anyio.ClosedResourceError:
             await anyio.lowlevel.checkpoint()
 
