Skip to content

Commit 44a35b2

Browse files
committed
Remove transport-specific code from server
1 parent 00fb685 commit 44a35b2

File tree

3 files changed

+6
-13
lines changed

3 files changed

+6
-13
lines changed

examples/http_server.rb

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,8 @@ def template(args, server_context:)
9494
end
9595

9696
# Create the Streamable HTTP transport
97-
server.transport = MCP::Transports::HTTP.new(server)
97+
transport = MCP::Transports::HTTP.new(server)
98+
server.transport = transport
9899

99100
# Create a logger for MCP-specific logging
100101
mcp_logger = Logger.new($stdout)
@@ -121,7 +122,7 @@ def template(args, server_context:)
121122

122123
# Handle the request
123124
puts "request: #{request.inspect}"
124-
response = server.handle_rack_request(request)
125+
response = transport.handle_request(request)
125126

126127
# Log the MCP response details
127128
_, _, body = response

examples/sse_test_server.rb

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,8 @@ def call(message:, delay: 0)
6464
end
6565

6666
# Create the Streamable HTTP transport
67-
server.transport = MCP::Transports::HTTP.new(server)
67+
transport = MCP::Transports::HTTP.new(server)
68+
server.transport = transport
6869

6970
# Create a logger for MCP request/response logging
7071
mcp_logger = Logger.new($stdout)
@@ -98,7 +99,7 @@ def call(message:, delay: 0)
9899
end
99100

100101
# Handle the request
101-
response = server.handle_rack_request(request)
102+
response = transport.handle_request(request)
102103

103104
# Log response details
104105
status, headers, body = response

lib/mcp/server.rb

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
require "json_rpc_handler"
44
require_relative "instrumentation"
55
require_relative "methods"
6-
require_relative "transports/http"
76

87
module MCP
98
class Server
@@ -94,14 +93,6 @@ def define_prompt(name: nil, description: nil, arguments: [], &block)
9493
@prompts[prompt.name_value] = prompt
9594
end
9695

97-
def handle_rack_request(request)
98-
@transport.handle_request(request)
99-
end
100-
101-
def close
102-
@transport.close
103-
end
104-
10596
def resources_list_handler(&block)
10697
@handlers[Methods::RESOURCES_LIST] = block
10798
end

0 commit comments

Comments
 (0)