Skip to content

Commit 8fbb2e8

Browse files
committed
Remove transport-specific code from server
1 parent 08be669 commit 8fbb2e8

File tree

3 files changed

+6
-12
lines changed

3 files changed

+6
-12
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 & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -98,14 +98,6 @@ def define_prompt(name: nil, description: nil, arguments: [], &block)
9898
@prompts[prompt.name_value] = prompt
9999
end
100100

101-
def handle_rack_request(request)
102-
@transport.handle_request(request)
103-
end
104-
105-
def close
106-
@transport.close
107-
end
108-
109101
def resources_list_handler(&block)
110102
@capabilities.support_resources
111103
@handlers[Methods::RESOURCES_LIST] = block

0 commit comments

Comments
 (0)