hzhang/inspector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

handle multiple connections

Browse Source
This commit is contained in:
Ashwin Bhat
2024-10-10 13:20:57 -07:00
parent 1497dbda29
commit 02bfd16309
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
server/src/index.ts
View File

@@ -14,7 +14,7 @@ import { SSEClientTransport } from "mcp-typescript/client/sse.js";
const app = express(); const app = express();
app.use(cors()); app.use(cors());
let transports: SSEServerTransport[] = []; let webAppTransports: SSEServerTransport[] = [];
app.get("/sse", async (req, res) => { app.get("/sse", async (req, res) => {
console.log("New SSE connection"); console.log("New SSE connection");
@@ -43,7 +43,8 @@ app.get("/sse", async (req, res) => {
} }
const webAppTransport = new SSEServerTransport("/message"); const webAppTransport = new SSEServerTransport("/message");
transports.push(webAppTransport); webAppTransports.push(webAppTransport);
console.log("Created web app transport");
await webAppTransport.connectSSE(req, res); await webAppTransport.connectSSE(req, res);
@@ -58,9 +59,10 @@ app.get("/sse", async (req, res) => {
}); });
app.post("/message", async (req, res) => { app.post("/message", async (req, res) => {
console.log("Received message"); const sessionId = req.query.sessionId;
console.log(`Received message for sessionId ${sessionId}`);
const transport = transports.find((t) => true); const transport = webAppTransports.find((t) => t.sessionId === sessionId);
if (!transport) { if (!transport) {
res.status(404).send("Session not found"); res.status(404).send("Session not found");
return; return;