リソースリークの修正

Author: poad

example/lambda/index.ts

@@ -6,23 +6,25 @@ import { createHonoApp } from 'aws-lambda-mcp-server';
 
 const logger = new Logger();
 
-const server = new McpServer({
-  name: 'hello-server',
-  version: '1.0.0',
-});
-
-server.tool(
-  'say_hello',
-  { who: z.string() },
-  async ({ who }) => ({
-    content: [{
-      type: 'text',
-      text: `${who} さん、こんにちは！`,
-    }],
-  }),
-);
+const createMcpServer = () => {
+  const server = new McpServer({
+    name: 'hello-server',
+    version: '1.0.0',
+  });
 
-const app = createHonoApp(server);
+  server.tool(
+    'say_hello',
+    { who: z.string() },
+    async ({ who }) => ({
+      content: [{
+        type: 'text',
+        text: `${who} さん、こんにちは！`,
+      }],
+    }),
+  );
+  return server;
+};
+const app = createHonoApp(createMcpServer);
 
 // Lambda handler
 export const handler = handle(app);

package/src/index.ts

@@ -119,32 +119,33 @@ const closeResources = async (server: McpServer, transport: StreamableHTTPTransp
  * @remarks
  * サーバーとトランスポートの接続・リクエスト処理・エラーハンドリングを行います。
  *
- * @param server MCPサーバーインスタンス
+ * @param createMcpServer MCPサーバーインスタンスを生成するファクトリ関数
  * @param c Honoのコンテキスト
  * @returns MCPレスポンス
  * @private
  */
-const handleRequest = async (server: McpServer, c: Context<BlankEnv, '/mcp', BlankInput>) => {
+const handleRequest = async (createMcpServer: () => McpServer, c: Context<BlankEnv, '/mcp', BlankInput>) => {
   const transport = new StreamableHTTPTransport({
     sessionIdGenerator: undefined, // セッションIDを生成しない（ステートレスモード）
     enableJsonResponse: true,
   });
+  const server = createMcpServer();
   try {
     await server.connect(transport);
     logger.trace('MCP リクエストを受信');
     return await transport.handleRequest(c);
   } catch (error) {
+    return handleError(c, error, 'MCP 接続中のエラー:');
+  } finally {
+    // エラーの有無に関わらず必ずリソースをクローズ
     try {
       await closeResources(server, transport);
-    } catch (closeError) {
-      const errorDetails = closeError instanceof Error
-        ? { message: closeError.message, stack: closeError.stack }
-        : closeError;
-      logger.error('Transport close failed after connection error:', { closeError: errorDetails });
+    } catch {
+      // クローズエラーは既にcloseResources内でログ出力されているため、
+      // ここでは追加のエラーハンドリングは不要
+      logger.warn('リソースクローズ中に追加エラーが発生しましたが、処理を継続します');
     }
-    return handleError(c, error, 'MCP 接続中のエラー:');
   }
-
 };
 
 /**
@@ -167,22 +168,11 @@ export const createHonoApp = (createMcpServer: () => McpServer) => {
   const app = new Hono();
 
   app.post('/mcp', async (c) => {
-    const server = createMcpServer();
-
-    try {
-      return await handleRequest(server, c);
-    } finally {
-      await server.close();
-    }
+    return await handleRequest(createMcpServer, c);
   });
 
   app.get('/mcp', async (c) => {
-    const server = createMcpServer();
-    try {
-      return await handleRequest(server, c);
-    } finally {
-      await server.close();
-    }
+    return await handleRequest(createMcpServer, c);
   });
 
   app.put('/mcp', methodNotAllowedHandler);