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

Allow multiple client connections to the MCP server, by making a unique server tranpsort for each web app transport.

Browse source 
* In server/src/index.ts
  - add serverTransports map

  - remove backingServerTransport var

  - in /mcp POST handler, when a new connection is being made
    - create a server transport
    - map the server transport using the session id from the web app connection
    - pass serverTransport to the mcpProxy instead of backingServerTransport

  - in /stdio GET handler, when a new connection is being made
    - create a server transport
    - map the server transport using the session id from the web app connection
    - pass serverTransport to the mcpProxy instead of backingServerTransport

  - in /sse GET handler, when a new connection is being made
    - create a server transport
    - map the server transport using the session id from the web app connection
    - pass serverTransport to the mcpProxy instead of backingServerTransport
This commit is contained in:
cliffhall 2025-05-21 17:45:19 -04:00
commit f04470ace7
1 changed files with 27 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

57
server/src/index.ts
View file

@ -48,7 +48,8 @@ app.use((req, res, next) => {
next(); next();
}); });
const webAppTransports: Map<string, Transport> = new Map<string, Transport>(); // Transports by sessionId const webAppTransports: Map<string, Transport> = new Map<string, Transport>(); // Web app transports by web app sessionId
const serverTransports: Map<string, Transport> = new Map<string, Transport>(); // Server Transports by web app sessionId
const createTransport = async (req: express.Request): Promise<Transport> => { const createTransport = async (req: express.Request): Promise<Transport> => {
const query = req.query; const query = req.query;
@ -137,8 +138,6 @@ const createTransport = async (req: express.Request): Promise<Transport> => {
} }
}; };
let backingServerTransport: Transport | undefined;
app.get("/mcp", async (req, res) => { app.get("/mcp", async (req, res) => {
const sessionId = req.headers["mcp-session-id"] as string; const sessionId = req.headers["mcp-session-id"] as string;
console.log(`Received GET message for sessionId ${sessionId}`); console.log(`Received GET message for sessionId ${sessionId}`);
@ -161,12 +160,12 @@ app.get("/mcp", async (req, res) => {
app.post("/mcp", async (req, res) => { app.post("/mcp", async (req, res) => {
const sessionId = req.headers["mcp-session-id"] as string | undefined; const sessionId = req.headers["mcp-session-id"] as string | undefined;
console.log(`Received POST message for sessionId ${sessionId}`); console.log(`Received POST message for sessionId ${sessionId}`);
let serverTransport: Transport | undefined;
if (!sessionId) { if (!sessionId) {
try { try {
console.log("New streamable-http connection"); console.log("New streamable-http connection");
try { try {
await backingServerTransport?.close(); serverTransport = await createTransport(req);
backingServerTransport = await createTransport(req);
} catch (error) { } catch (error) {
if (error instanceof SseError && error.code === 401) { if (error instanceof SseError && error.code === 401) {
console.error( console.error(
@ -180,12 +179,13 @@ app.post("/mcp", async (req, res) => {
throw error; throw error;
} }
console.log("Connected MCP client to backing server transport"); console.log("Connected MCP client to server transport");
const webAppTransport = new StreamableHTTPServerTransport({ const webAppTransport = new StreamableHTTPServerTransport({
sessionIdGenerator: randomUUID, sessionIdGenerator: randomUUID,
onsessioninitialized: (sessionId) => { onsessioninitialized: (sessionId) => {
webAppTransports.set(sessionId, webAppTransport); webAppTransports.set(sessionId, webAppTransport);
serverTransports.set(sessionId, serverTransport!);
console.log("Created streamable web app transport " + sessionId); console.log("Created streamable web app transport " + sessionId);
}, },
}); });
@ -194,7 +194,7 @@ app.post("/mcp", async (req, res) => {
mcpProxy({ mcpProxy({
transportToClient: webAppTransport, transportToClient: webAppTransport,
transportToServer: backingServerTransport, transportToServer: serverTransport,
}); });
await (webAppTransport as StreamableHTTPServerTransport).handleRequest( await (webAppTransport as StreamableHTTPServerTransport).handleRequest(
@ -229,10 +229,9 @@ app.post("/mcp", async (req, res) => {
app.get("/stdio", async (req, res) => { app.get("/stdio", async (req, res) => {
try { try {
console.log("New connection"); console.log("New connection");
let serverTransport: Transport | undefined;
try { try {
await backingServerTransport?.close(); serverTransport = await createTransport(req);
backingServerTransport = await createTransport(req);
} catch (error) { } catch (error) {
if (error instanceof SseError && error.code === 401) { if (error instanceof SseError && error.code === 401) {
console.error( console.error(
@ -250,26 +249,24 @@ app.get("/stdio", async (req, res) => {
const webAppTransport = new SSEServerTransport("/message", res); const webAppTransport = new SSEServerTransport("/message", res);
webAppTransports.set(webAppTransport.sessionId, webAppTransport); webAppTransports.set(webAppTransport.sessionId, webAppTransport);
serverTransports.set(webAppTransport.sessionId, serverTransport);
console.log("Created web app transport"); console.log("Created client/server transports");
await webAppTransport.start(); await webAppTransport.start();
(backingServerTransport as StdioClientTransport).stderr!.on(
"data", (serverTransport as StdioClientTransport).stderr!.on("data", (chunk) => {
(chunk) => { webAppTransport.send({
webAppTransport.send({ jsonrpc: "2.0",
jsonrpc: "2.0", method: "notifications/stderr",
method: "notifications/stderr", params: {
params: { content: chunk.toString(),
content: chunk.toString(), },
}, });
}); });
},
);
mcpProxy({ mcpProxy({
transportToClient: webAppTransport, transportToClient: webAppTransport,
transportToServer: backingServerTransport, transportToServer: serverTransport,
}); });
console.log("Set up MCP proxy"); console.log("Set up MCP proxy");
@ -284,10 +281,9 @@ app.get("/sse", async (req, res) => {
console.log( console.log(
"New SSE connection. NOTE: The sse transport is deprecated and has been replaced by streamable-http", "New SSE connection. NOTE: The sse transport is deprecated and has been replaced by streamable-http",
); );
let serverTransport: SSEServerTransport | undefined;
try { try {
await backingServerTransport?.close(); serverTransport = (await createTransport(req)) as SSEServerTransport;
backingServerTransport = await createTransport(req);
} catch (error) { } catch (error) {
if (error instanceof SseError && error.code === 401) { if (error instanceof SseError && error.code === 401) {
console.error( console.error(
@ -305,13 +301,14 @@ app.get("/sse", async (req, res) => {
const webAppTransport = new SSEServerTransport("/message", res); const webAppTransport = new SSEServerTransport("/message", res);
webAppTransports.set(webAppTransport.sessionId, webAppTransport); webAppTransports.set(webAppTransport.sessionId, webAppTransport);
console.log("Created web app transport"); serverTransports.set(webAppTransport.sessionId, serverTransport);
console.log("Created client/server transports");
await webAppTransport.start(); await webAppTransport.start();
mcpProxy({ mcpProxy({
transportToClient: webAppTransport, transportToClient: webAppTransport,
transportToServer: backingServerTransport, transportToServer: serverTransport,
}); });
console.log("Set up MCP proxy"); console.log("Set up MCP proxy");