5. Xây Dựng MCP Server Bằng TypeScript & Node.js: Kết Nối CSDL & REST API

// index.ts
import { Server } from "@modelcontextprotocol/sdk/server/index.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
import { z } from "zod";

const server = new Server(
  { name: "ts-db-server", version: "1.0.0" },
  { capabilities: { tools: {} } }
);

// 1. Khai báo danh sách Tools
server.setRequestHandler(ListToolsRequestSchema, async () => {
  return {
    tools: [
      {
        name: "query_inventory",
        description: "Tra cứu số lượng tồn kho sản phẩm theo mã SKU",
        inputSchema: {
          type: "object",
          properties: {
            sku: { type: "string", description: "Mã SKU sản phẩm (ví dụ: AITG-001)" }
          },
          required: ["sku"]
        }
      }
    ]
  };
});

// 2. Xử lý thực thi Tool
server.setRequestHandler(CallToolRequestSchema, async (request) => {
  if (request.params.name === "query_inventory") {
    const sku = String(request.params.arguments?.sku);
    // Giả lập truy vấn Database
    return {
      content: [{ type: "text", text: `Sản phẩm ${sku}: Còn 45 chiếc trong kho TP.HCM.` }]
    };
  }
  throw new Error("Tool không tồn tại");
});

async function main() {
  const transport = new StdioServerTransport();
  await server.connect(transport);
}
main();