AWS Lambda でスケーラブルな MCP サーバーを構築する：実践ガイド #lambda

import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
import express from "express";
import { z } from "zod";

// MCP サーバーを作成
const server = new McpServer({
  name: "Example MCP Server",
  version: "1.0.0"
});

// Express アプリケーションを作成
const app = express();
app.use(express.json());

// ツールを追加 - シンプルな計算機を実装
server.tool(
  "calculate",
  { a: z.number(), b: z.number(), operation: z.enum(["add", "subtract", "multiply", "divide"]) },
  async ({ a, b, operation }) => {
    let result;
    switch (operation) {
      case "add": result = a + b; break;
      case "subtract": result = a - b; break;
      case "multiply": result = a * b; break;
      case "divide": result = a / b; break;
    }
    return {
      content: [{ type: "text", text: `結果: ${result}` }]
    };
  }
);

// Streamable HTTP トランスポート（セッションレス）を設定
const transport = new StreamableHTTPServerTransport({
  sessionIdGenerator: undefined, // セッション管理を無効化
});

// ルートを設定
app.post('/mcp', async (req, res) => {
  try {
    await transport.handleRequest(req, res, req.body);
  } catch (error) {
    console.error('MCP リクエスト処理中のエラー:', error);
    if (!res.headersSent) {
      res.status(500).json({
        jsonrpc: '2.0',
        error: {
          code: -32603,
          message: '内部サーバーエラー',
        },
        id: null,
      });
    }
  }
});

app.get('/mcp', async (req, res) => {
  res.writeHead(405).end(JSON.stringify({
    jsonrpc: "2.0",
    error: {
      code: -32000,
      message: "メソッドは許可されていません。"
    },
    id: null
  }));
});

app.delete('/mcp', async (req, res) => {
  res.writeHead(405).end(JSON.stringify({
    jsonrpc: "2.0",
    error: {
      code: -32000,
      message: "メソッドは許可されていません。"
    },
    id: null
  }));
});

// サーバーを起動
const PORT = process.env.PORT || 8080;
server.connect(transport).then(() => {
  app.listen(PORT, () => {
    console.log(`MCP サーバーがポート ${PORT} でリッスン中`);
  });
}).catch(error => {
  console.error('サーバーのセットアップに失敗しました:', error);
  process.exit(1);
});