MCP Cosmosdb Go

🚀 使用 Go SDK 實現 Azure Cosmos DB 的 MCP 服務器

本項目是一個藉助 Go SDK 構建的 Azure Cosmos DB 的 MCP 服務器示例。mcp - go 項目作為 MCP 的 Go 實現方案。

此 MCP 服務器為與 Azure Cosmos DB 交互提供了一系列實用工具，具體如下：

• 列出數據庫：可檢索特定 Cosmos DB 帳戶中的所有數據庫。
• 列出容器：能檢索特定數據庫中的所有容器。
• 讀取容器元數據：獲取特定容器的元數據或配置信息。
• 創建容器：在指定數據庫中創建新容器，並可定義分區鍵。
• 向容器添加項：將新項添加到指定數據庫中的容器。
• 讀取消息：使用項的 ID 和分區鍵從容器中讀取特定項。
• 執行查詢：在 Cosmos DB 容器上執行 SQL 查詢，還可選擇指定分區鍵範圍。

以下是演示視頻（建議以 2x 速度觀看 😄），該視頻使用了 VS Code Insiders in Agent mode：

[](https://www.youtube.com/watch?v=CsM - mspWJeM)

🚀 快速開始

📦 安裝指南

克隆項目倉庫並進入項目目錄：

git clone https://github.com/abhirockzz/mcp_cosmosdb_go
cd mcp_cosmosdb_go

go build -o mcp_azure_cosmosdb main.go

配置 MCP 服務器：

mkdir -p .vscode

# 定義 mcp.json 的內容
MCP_JSON_CONTENT=$(cat <<EOF
{
  "servers": {
    "CosmosDB Golang MCP": {
      "type": "stdio",
      "command": "$(pwd)/mcp_azure_cosmosdb"
    }
  }
}
EOF
)

# 將內容寫入 mcp.json
echo "$MCP_JSON_CONTENT" > .vscode/mcp.json

🔐 Azure Cosmos DB 的 RBAC 權限和身份驗證

• 權限要求：您所使用的用戶主體應具備相應權限（控制平面 和 數據平面），以便執行數據庫、容器和項目的 CRUD 操作。
• 身份驗證方式：
  • 本地憑據：使用 Azure CLI（az login）進行本地登錄即可，MCP 服務器將自動採用 DefaultAzureCredential 實現。
  • 環境變量方式：您也可以在 MCP 服務器配置中設置 COSMOSDB_ACCOUNT_KEY 環境變量，示例如下：

  {
    "servers": {
      "CosmosDB Golang MCP": {
        "type": "stdio",
        "command": "/Users/demo/mcp_azure_cosmosdb",
        "env": {
          "COSMOSDB_ACCOUNT_KEY": "輸入密鑰"
        }
      }
    }
  }
  

完成上述步驟後，您就可以啟動 VS Code Insider 的代理模式，開始使用該 MCP 服務器啦。

🔧 本地開發和測試

您可以按照上述步驟進行本地開發和測試，確保各項功能正常運行。