Bỏ qua để đến nội dung
@aws/nx-plugin
Blog

MCP Server đến Cơ sở dữ liệu Quan hệ

Generator connection kết nối một TypeScript MCP Server với dự án Cơ sở dữ liệu Quan hệ, làm cho Prisma client khả dụng bên trong createServer và các điểm vào transport của nó.

Điều kiện tiên quyết

Phần tiêu đề “Điều kiện tiên quyết”

Trước khi sử dụng generator này, hãy đảm bảo bạn có:

  1. Một dự án ts#mcp-server
  2. Một dự án ts#rdb

Cách sử dụng

Phần tiêu đề “Cách sử dụng”

Chạy Generator

Phần tiêu đề “Chạy Generator”
  1. Cài đặt Nx Console VSCode Plugin nếu bạn chưa cài đặt
  2. Mở Nx Console trong VSCode
  3. Nhấp Generate (UI) trong phần "Common Nx Commands"
  4. Tìm kiếm @aws/nx-plugin - connection
  5. Điền các tham số bắt buộc
    • Nhấp Generate

    Chọn dự án MCP server của bạn làm nguồn và dự án cơ sở dữ liệu quan hệ của bạn làm đích. Nếu dự án chứa nhiều thành phần MCP server, hãy chỉ định sourceComponent để phân biệt.

    Tùy chọn

    Phần tiêu đề “Tùy chọn”
    Tham số Kiểu Mặc định Mô tả
    sourceProject Bắt buộc string - Dự án nguồn
    targetProject Bắt buộc string - Dự án đích để kết nối tới
    sourceComponent string - Component nguồn để kết nối từ đó (tên component, đường dẫn tương đối so với thư mục gốc của dự án nguồn, hoặc generator id). Sử dụng '.' để chọn rõ ràng dự án làm nguồn.
    targetComponent string - Component đích để kết nối tới (tên component, đường dẫn tương đối so với thư mục gốc của dự án đích, hoặc generator id). Sử dụng '.' để chọn rõ ràng dự án làm đích.

    Kết quả của Generator

    Phần tiêu đề “Kết quả của Generator”

    Generator sửa đổi ba tệp trong thư mục nguồn của MCP server của bạn:

    • Thư mụcpackages/my-service/src/my-mcp
      • server.ts createServer được cập nhật để chấp nhận tham số { db }
      • http.ts Prisma client được tạo phía trên const server = createServer({ db })
      • stdio.ts Prisma client được tạo trước await createServer({ db }).connect(transport)

    Ngoài ra, target <mcp-server-name>-serve-local được cập nhật để phụ thuộc vào target serve-local của cơ sở dữ liệu.

    Cách hoạt động

    Phần tiêu đề “Cách hoạt động”

    Định nghĩa Server

    Phần tiêu đề “Định nghĩa Server”

    createServer nhận Prisma client như một tham số để nó vẫn đồng bộ và có thể truyền client cho các công cụ và tài nguyên:

    packages/my-service/src/my-mcp/server.ts
    import { getPrisma as getMyDb } from ':my-scope/my-db';
    

    export const createServer = ({
      myDb,
    }: {
      myDb: Awaited<ReturnType<typeof getMyDb>>;
    }) => {
      const server = new McpServer({ name: 'my-service', version: '1.0.0' });
      // register tools/resources that use myDb
      return server;
    };

    HTTP Transport

    Phần tiêu đề “HTTP Transport”

    Prisma client được tạo bên trong request handler, phía trên lời gọi createServer:

    packages/my-service/src/my-mcp/http.ts
    import { getPrisma as getMyDb } from ':my-scope/my-db';
    

    app.post('/mcp', async (req, res) => {
      try {
        const myDb = await getMyDb();
        const server = createServer({ myDb });
        // ...
      }
    });

    Stdio Transport

    Phần tiêu đề “Stdio Transport”

    Cùng một mẫu được áp dụng cho điểm vào stdio transport:

    packages/my-service/src/my-mcp/stdio.ts
    import { getPrisma as getMyDb } from ':my-scope/my-db';
    

    export const startMcpServer = async () => {
      const transport = new StdioServerTransport();
      const myDb = await getMyDb();
      await createServer({ myDb }).connect(transport);
    };

    Nhiều Cơ sở dữ liệu

    Phần tiêu đề “Nhiều Cơ sở dữ liệu”

    Chạy generator lại với một đích khác sẽ thêm cơ sở dữ liệu thứ hai bên cạnh cơ sở dữ liệu đầu tiên. Cả hai đều được truyền cho createServer cùng nhau:

    packages/my-service/src/my-mcp/http.ts
    const postgresDb = await getPostgresDb();
    const mySqlDb = await getMySqlDb();
    const server = createServer({ postgresDb, mysqlDb });

    Cơ sở hạ tầng

    Phần tiêu đề “Cơ sở hạ tầng”

    Construct MCP server được tạo ra triển khai IGrantable và IConnectable, vì vậy bạn có thể cấp quyền truy cập mạng và IAM cho cơ sở dữ liệu trực tiếp trên construct.

    packages/infra/src/stacks/application-stack.ts
    import { MyDatabase } from ':my-scope/common-constructs';
    

    const db = new MyDatabase(this, 'Db', { vpc, ... });
    const myMcpServer = new MyMcpServer(this, 'MyMcpServer', { vpc, ... });
    

    db.allowDefaultPortFrom(myMcpServer);
    db.grantConnect(myMcpServer);

    allowDefaultPortFrom mở quy tắc security group để runtime MCP server có thể truy cập cổng cơ sở dữ liệu. grantConnect cấp quyền IAM rds-db:connect cho execution role của server.

    Phát triển Cục bộ

    Phần tiêu đề “Phát triển Cục bộ”
    Terminal window
    pnpm nx <mcp-server-name>-serve-local <project-name>

    Điều này khởi động MCP server và tất cả các cơ sở dữ liệu được kết nối. Biến môi trường SERVE_LOCAL=true khiến mỗi Prisma client kết nối với cơ sở dữ liệu Docker cục bộ của nó thay vì Aurora.