Merge pull request #49 from koic/suppress_interrupt_exception_for_stdio_server
Suppress `Interrupt` exception for stdio server
This commit is contained in:
commit
2ee3d39814
1 changed files with 6 additions and 0 deletions
|
|
@ -7,6 +7,8 @@ module MCP
|
||||||
class Server
|
class Server
|
||||||
module Transports
|
module Transports
|
||||||
class StdioTransport < Transport
|
class StdioTransport < Transport
|
||||||
|
STATUS_INTERRUPTED = Signal.list["INT"] + 128
|
||||||
|
|
||||||
def initialize(server)
|
def initialize(server)
|
||||||
@server = server
|
@server = server
|
||||||
@open = false
|
@open = false
|
||||||
|
|
@ -20,6 +22,10 @@ module MCP
|
||||||
while @open && (line = $stdin.gets)
|
while @open && (line = $stdin.gets)
|
||||||
handle_json_request(line.strip)
|
handle_json_request(line.strip)
|
||||||
end
|
end
|
||||||
|
rescue Interrupt
|
||||||
|
warn("\nExiting...")
|
||||||
|
|
||||||
|
exit(STATUS_INTERRUPTED)
|
||||||
end
|
end
|
||||||
|
|
||||||
def close
|
def close
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue