cq-mcp-server 0.3.0 → 0.3.2

package/dist/index.js

@@ -2,6 +2,7 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
 import { createServer } from "node:http";
 import { randomUUID } from "node:crypto";
 import { z } from "zod";
@@ -330,42 +331,74 @@ log.info(`CloudQuery MCP Server 启动  transport=${MCP_TRANSPORT}`);
 if (MCP_TRANSPORT === "http") {
     // session_id → transport 映射，支持多用户并发
     const sessions = new Map();
+    // SSE transport 映射（sessionId → transport），用于兼容 Dify 等旧版客户端
+    const sseSessions = new Map();
     const httpServer = createServer(async (req, res) => {
+        // 允许跨域（Dify 可能从不同域访问）
+        res.setHeader("Access-Control-Allow-Origin", "*");
+        res.setHeader("Access-Control-Allow-Methods", "GET, POST, DELETE, OPTIONS");
+        res.setHeader("Access-Control-Allow-Headers", "Content-Type, mcp-session-id");
+        res.setHeader("Access-Control-Expose-Headers", "mcp-session-id");
+        if (req.method === "OPTIONS") {
+            res.writeHead(204).end();
+            return;
+        }
         if (!req.url?.startsWith("/mcp")) {
             res.writeHead(404).end("Not Found");
             return;
         }
-        // 每个新 session（首次 initialize 请求）创建独立的 server + transport
-        const sessionId = req.headers["mcp-session-id"];
-        if (!sessionId) {
-            // 新 session：为该用户创建独立实例
-            const transport = new StreamableHTTPServerTransport({
-                sessionIdGenerator: () => randomUUID(),
+        // ── SSE transport（兼容 Dify）──
+        // GET /mcp → 建立 SSE 连接
+        if (req.method === "GET") {
+            const sseTransport = new SSEServerTransport("/mcp", res);
+            sseSessions.set(sseTransport.sessionId, sseTransport);
+            log.info(`SSE 连接建立: ${sseTransport.sessionId}`);
+            const mcpServer = createMcpServer();
+            await mcpServer.connect(sseTransport);
+            req.on("close", () => {
+                sseSessions.delete(sseTransport.sessionId);
+                log.info(`SSE 连接关闭: ${sseTransport.sessionId}，当前活跃: ${sseSessions.size}`);
             });
-            const server = createMcpServer();
-            await server.connect(transport);
-            transport.onclose = () => {
-                const sid = transport.sessionId;
-                if (sid) {
-                    sessions.delete(sid);
-                    log.info(`Session 关闭: ${sid}，当前活跃: ${sessions.size}`);
-                }
-            };
-            await transport.handleRequest(req, res);
-            // handleRequest 后 sessionId 已生成，存入 map
-            if (transport.sessionId) {
-                sessions.set(transport.sessionId, transport);
-                log.info(`新 Session: ${transport.sessionId}，当前活跃: ${sessions.size}`);
-            }
+            return;
         }
-        else {
-            // 已有 session：路由到对应 transport
+        // POST /mcp?sessionId=xxx → SSE 消息（Dify 旧版协议）
+        const urlObj = new URL(req.url, `http://${req.headers.host}`);
+        const sseSessionId = urlObj.searchParams.get("sessionId");
+        if (sseSessionId && sseSessions.has(sseSessionId)) {
+            await sseSessions.get(sseSessionId).handlePostMessage(req, res);
+            return;
+        }
+        // ── Streamable HTTP transport（新版协议）──
+        const sessionId = req.headers["mcp-session-id"];
+        if (sessionId) {
             const transport = sessions.get(sessionId);
             if (!transport) {
                 res.writeHead(404).end("Session not found");
                 return;
             }
             await transport.handleRequest(req, res);
+            return;
+        }
+        if (req.method !== "POST") {
+            res.writeHead(400).end("Missing mcp-session-id");
+            return;
+        }
+        const transport = new StreamableHTTPServerTransport({
+            sessionIdGenerator: () => randomUUID(),
+        });
+        const mcpServer = createMcpServer();
+        await mcpServer.connect(transport);
+        transport.onclose = () => {
+            const sid = transport.sessionId;
+            if (sid) {
+                sessions.delete(sid);
+                log.info(`Session 关闭: ${sid}，当前活跃: ${sessions.size}`);
+            }
+        };
+        await transport.handleRequest(req, res);
+        if (transport.sessionId) {
+            sessions.set(transport.sessionId, transport);
+            log.info(`新 Session: ${transport.sessionId}，当前活跃: ${sessions.size}`);
         }
     });
     httpServer.listen(MCP_PORT, () => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cq-mcp-server",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "description": "MCP Server for CloudQuery platform — list databases, list tables, inspect columns, and execute SQL via MCP tools.",
   "type": "module",
   "main": "dist/index.js",