@douyinfe/semi-mcp 1.0.12 → 1.0.14

package/dist/http.js

@@ -1326,6 +1326,8 @@ function createMCPServer() {
     });
     return server;
 }
+const sessions = new Map();
+const SESSION_TIMEOUT = 1800000;
 function parseArgs() {
     const args = process.argv.slice(2);
     let port = 3000;
@@ -1383,27 +1385,40 @@ Endpoints:
         timeout
     };
 }
+function cleanupSessions() {
+    const now = Date.now();
+    for (const [sessionId, info] of sessions)if (now - info.lastActivity > SESSION_TIMEOUT) sessions.delete(sessionId);
+}
+setInterval(cleanupSessions, 60000);
+async function processSessionRequest(sessionId, handler) {
+    const session = sessions.get(sessionId);
+    if (!session) throw new Error(`Session not found: ${sessionId}`);
+    const requestPromise = handler();
+    session.requestQueue.push(()=>requestPromise);
+    if (!session.isProcessing) {
+        session.isProcessing = true;
+        try {
+            while(session.requestQueue.length > 0){
+                const nextHandler = session.requestQueue.shift();
+                await nextHandler();
+            }
+        } finally{
+            session.isProcessing = false;
+        }
+    }
+    return requestPromise;
+}
 async function main() {
     const { port, hosts, stateless, timeout } = parseArgs();
     const version = getPackageVersion();
     let processedHosts = hosts;
     const hasIPv4All = hosts.includes('0.0.0.0');
     const hasIPv6All = hosts.includes('::');
-    if (hasIPv4All && hasIPv6All) {
-        processedHosts = hosts.filter((h)=>'0.0.0.0' !== h);
-        console.log(`[${new Date().toISOString()}] 检测到同时监听 IPv4 和 IPv6，使用 :: (IPv6) 统一监听`);
-    }
-    if (0 === processedHosts.length) {
-        processedHosts = [
-            '::'
-        ];
-        console.log(`[${new Date().toISOString()}] 使用默认配置: 监听 IPv6 (::)`);
-    }
+    if (hasIPv4All && hasIPv6All) processedHosts = hosts.filter((h)=>'0.0.0.0' !== h);
+    if (0 === processedHosts.length) processedHosts = [
+        '::'
+    ];
     const server = createMCPServer();
-    const transport = new StreamableHTTPServerTransport({
-        sessionIdGenerator: stateless ? void 0 : ()=>crypto.randomUUID()
-    });
-    await server.connect(transport);
     console.log(`[${new Date().toISOString()}] MCP 服务器已启动`);
     console.log(`[${new Date().toISOString()}] 模式: ${stateless ? '无状态 (Stateless)' : '有状态 (Stateful)'}`);
     const httpServer = createServer(async (req, res)=>{
@@ -1427,11 +1442,13 @@ async function main() {
                 version,
                 transport: 'streamable-http',
                 stateless,
-                sessionTimeout: `${timeout} minutes`
+                sessionTimeout: `${timeout} minutes`,
+                activeSessions: sessions.size
             }));
             return;
         }
         if ('/mcp' === url.pathname) {
+            const sessionId = req.headers['mcp-session-id'];
             let body = '';
             req.on('data', (chunk)=>{
                 body += chunk.toString();
@@ -1439,12 +1456,45 @@ async function main() {
             await new Promise((resolve)=>{
                 req.on('end', async ()=>{
                     try {
-                        const parsedBody = body ? JSON.parse(body) : void 0;
-                        await transport.handleRequest(req, res, parsedBody);
-                        console.log(`[${new Date().toISOString()}] ${req.method} ${url.pathname} - ${res.statusCode}`);
+                        let transport;
+                        if (stateless) {
+                            transport = new StreamableHTTPServerTransport({
+                                sessionIdGenerator: void 0
+                            });
+                            await server.connect(transport);
+                        } else if (sessionId && sessions.has(sessionId)) {
+                            transport = sessions.get(sessionId).transport;
+                            sessions.get(sessionId).lastActivity = Date.now();
+                        } else {
+                            const newSessionId = sessionId || crypto.randomUUID();
+                            transport = new StreamableHTTPServerTransport({
+                                sessionIdGenerator: ()=>newSessionId
+                            });
+                            await server.connect(transport);
+                            sessions.set(newSessionId, {
+                                transport,
+                                lastActivity: Date.now(),
+                                requestQueue: [],
+                                isProcessing: false
+                            });
+                        }
+                        if ('GET' === req.method) {
+                            transport.handleRequest(req, res).catch(()=>{});
+                            resolve();
+                            return;
+                        }
+                        let parsedBody;
+                        if (body.trim()) try {
+                            parsedBody = JSON.parse(body);
+                        } catch  {
+                            parsedBody = void 0;
+                        }
+                        if (sessionId && sessions.has(sessionId)) await processSessionRequest(sessionId, async ()=>{
+                            await transport.handleRequest(req, res, parsedBody);
+                        });
+                        else await transport.handleRequest(req, res, parsedBody);
                     } catch (error) {
                         const errorMessage = error instanceof Error ? error.message : String(error);
-                        console.error(`[${new Date().toISOString()}] 请求处理错误:`, errorMessage);
                         if (!res.headersSent) {
                             res.writeHead(500, {
                                 'Content-Type': 'application/json'
@@ -1477,18 +1527,10 @@ async function main() {
                 sessionTimeout: `${timeout} minutes`,
                 endpoints: {
                     mcp: {
-                        POST: '/mcp - 发送 MCP 请求',
-                        GET: '/mcp - SSE 流 (需要 Mcp-Session-Id 头)'
+                        POST: '/mcp',
+                        GET: '/mcp (SSE)'
                     },
-                    health: '/health - 健康检查'
-                },
-                headers: {
-                    'Mcp-Session-Id': '会话 ID (初始化响应后获取，后续请求需携带)'
-                },
-                usage: {
-                    step1: '客户端发送 initialize 请求',
-                    step2: '服务器返回响应并包含 Mcp-Session-Id header',
-                    step3: '后续请求携带 Mcp-Session-Id header'
+                    health: '/health'
                 }
             }, null, 2));
             return;
@@ -1497,15 +1539,15 @@ async function main() {
             'Content-Type': 'application/json'
         });
         res.end(JSON.stringify({
-            error: '未知的端点'
+            error: 'Unknown endpoint'
         }));
     });
     const servers = [];
     let startedCount = 0;
     console.log(`
-╔══════════════════════════════════════════════════════════════╗
+╔══════════════════════════════════════════════════════════════════════╗
 ║        Semi MCP Server (Streamable HTTP) v${version.padEnd(10)}        ║
-╠══════════════════════════════════════════════════════════════╣
+╠══════════════════════════════════════════════════════════════════════╣
 ║  模式: ${stateless ? '无状态 (Stateless)' : '有状态 (Stateful) '}                                 ║
 ║  会话超时: ${String(timeout).padEnd(3)} 分钟                                        ║
 ║                                                              ║`);
@@ -1532,7 +1574,7 @@ async function main() {
 ║    POST   /mcp      发送 MCP 请求                            ║
 ║    GET    /mcp      SSE 流 (服务器推送)                      ║
 ║    GET    /health   健康检查                                 ║
-╚══════════════════════════════════════════════════════════════╝
+╚══════════════════════════════════════════════════════════════════════╝
 `);
                 console.log(`[${new Date().toISOString()}] 所有服务器已启动，监听 ${processedHosts.length} 个地址`);
                 console.log(`[${new Date().toISOString()}] 总计监听: ${processedHosts.join(', ')}`);
@@ -1540,20 +1582,16 @@ async function main() {
         });
         servers.push(httpServer);
     });
-    processedHosts.length;
     const shutdown = async ()=>{
         console.log('\n正在关闭服务器...');
-        try {
-            await transport.close();
-            console.log('Transport 已关闭');
-        } catch (error) {
-            console.error('关闭 transport 时出错:', error);
-        }
+        for (const [sessionId, info] of sessions)try {
+            await info.transport.close();
+        } catch  {}
+        sessions.clear();
         let closedCount = 0;
         servers.forEach((server, index)=>{
             server.close(()=>{
                 closedCount++;
-                console.log(`[${new Date().toISOString()}] 服务器 ${index + 1}/${servers.length} 已关闭`);
                 if (closedCount === servers.length) {
                     console.log('所有服务器已关闭');
                     process.exit(0);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@douyinfe/semi-mcp",
-  "version": "1.0.12",
+  "version": "1.0.14",
   "description": "Semi Design MCP Server - Model Context Protocol server for Semi Design components and documentation",
   "type": "module",
   "main": "./dist/index.js",
@@ -49,7 +49,7 @@
     "prepublishOnly": "npm run build && npm test"
   },
   "dependencies": {
-    "@modelcontextprotocol/sdk": "^1.0.4",
+    "@modelcontextprotocol/sdk": "^1.25.1",
     "@swc/core": "^1.15.8",
     "oxc-parser": "^0.106.0"
   },