Custom MCP Database

🚀 MCP 數據庫服務器

本項目提供了一箇中間件/控制平面（MCP）服務器，允許代碼代理（如 Gemini、Claude Desktop 和 Claude Code）安全地對各種數據庫（PostgreSQL、MySQL、MongoDB、Oracle）執行查詢，而無需直接暴露憑證。數據庫連接配置由 SQLite 數據庫（mcp_config.sqlite3）管理。

🚀 快速開始

安裝項目

按照以下步驟設置項目：

1. 克隆倉庫（若尚未完成）：

git clone <repository_url>
cd custom-mcp-database

2. 安裝依賴： 本項目使用 venv（虛擬環境）來管理依賴。運行以下命令創建虛擬環境並安裝所需的 Python 包：

make install

這將創建一個 venv/ 目錄，並安裝 requirements.txt 中列出的所有內容。

啟動 MCP 服務器

使用 make run 命令啟動 MCP 服務器：

make run

服務器將啟動並監聽來自代碼代理的傳入請求。

📦 安裝指南

克隆倉庫

git clone <repository_url>
cd custom-mcp-database

安裝依賴

make install

📚 詳細文檔

數據庫配置

數據庫連接存儲在 mcp_config.sqlite3 中。您可以使用全局 mcp-db 命令（推薦）或本地 main.py 腳本管理這些連接。

添加數據庫連接

使用 add-db 命令。所需參數因數據庫類型而異。 通用語法（全局命令 - 推薦）：

mcp-db add-db --alias <alias> --type <type> [connection_parameters]

替代方法（本地命令）：

python main.py add-db --alias <alias> --type <type> [connection_parameters]

示例（全局命令）：

• PostgreSQL：

mcp-db add-db --alias pg_dev --type postgres --host localhost --port 5432 --user myuser --password mypassword --dbname mydb

• MySQL：

mcp-db add-db --alias mysql_prod --type mysql --host 192.168.1.10 --port 3306 --user root --password secret --dbname production_db

• Oracle：

mcp-db add-db --alias oracle_test --type oracle --host oracle.example.com --port 1521 --user system --password oraclepass --dbname ORCLPDB1

• MongoDB：

mcp-db add-db --alias mongo_cluster --type mongo --uri "mongodb+srv://user:pass@cluster.mongodb.net/" --dbname myapp_db

刪除數據庫連接

使用 remove-db 命令和要刪除的連接別名： 全局命令：

mcp-db remove-db --alias <alias>

本地命令：

python main.py remove-db --alias <alias>

示例：

mcp-db remove-db --alias pg_dev

獲取幫助

如需詳細幫助和示例：

mcp-db --help
mcp-db execute-query --help

與代碼代理集成

Gemini

MCP 服務器啟動（make run）後，Gemini 將自動發現並提供以下工具來與配置的數據庫進行交互：

• list_aliases()：列出所有配置的數據庫別名。
• add_database(...)：添加新的數據庫連接。
• remove_database(...)：刪除數據庫連接。
• execute_query(database_alias, query, params, schema)：對配置的數據庫執行查詢。

Gemini 使用示例：

• 列出別名：

list_aliases()

• 執行 SQL 查詢（PostgreSQL/MySQL/Oracle）：

execute_query(database_alias="pg_dev", query="SELECT * FROM users WHERE id = %s;", params={"id": 1})

• 執行 MongoDB 查詢：

execute_query(database_alias="mongo_cluster", query='''{"name": "John Doe"}''', params={"collection": "users"})

Claude Desktop

要與 Claude Desktop 集成，您需要將其 claude_desktop_config.json 文件配置為指向您的 MCP 服務器。在該文件（通常位於 Claude Desktop 配置目錄中）中創建或修改一個類似如下的條目：

{
  "mcpServers": {
    "Custom DB Server": {
      "command": "/your-path-to/custom-mcp-database/venv/bin/python",
      "args": [
        "/your-path-to/custom-mcp-database/main.py"
      ],
      "workingDirectory": "/your-path-to/custom-mcp-database"
    }
  }
}

重要提示： 將 /your-path-to/custom-mcp-database 替換為 custom-mcp-database 目錄的實際絕對路徑。 配置完成後，重啟 Claude Desktop。MCP 工具隨後可在 Claude Desktop 中使用。

AI 代理的工具配置

要使 Claude Code 和 Gemini 等 AI 代理能夠自動發現並使用 MCP 服務器的工具，您需要將它們配置為啟動或連接到 MCP 服務器。這通常涉及提供 main.py 腳本的路徑並指定工作目錄。 以下是如何配置 AI 代理的 mcpServers 部分的示例：

Claude Code

{
  "mcpServers": {
    "Custom DB Server": {
      "command": "/your-path-to/custom-mcp-database/venv/bin/python",
      "args": [
        "/your-path-to/custom-mcp-database/main.py"
      ],
      "workingDirectory": "/your-path-to/custom-mcp-database"
    }
  }
}

Gemini

{
  "mcpServers": {
    "Custom DB Server": {
      "command": "/your-path-to/custom-mcp-database/venv/bin/python",
      "args": [
        "/your-path-to/custom-mcp-database/main.py"
      ],
      "workingDirectory": "/your-path-to/custom-mcp-database"
    }
  }
}

重要提示： 將 /your-path-to/custom-mcp-database 替換為 custom-mcp-database 目錄的實際絕對路徑。 請參考您特定 AI 代理的官方文檔，以獲取有關配置外部 MCP 服務器的最準確和最新說明。

💻 使用示例

基礎用法

啟動服務器

make run

列出數據庫別名（Gemini）

list_aliases()

高級用法

執行 SQL 查詢（PostgreSQL/MySQL/Oracle - Gemini）

execute_query(database_alias="pg_dev", query="SELECT * FROM users WHERE id = %s;", params={"id": 1})

執行 MongoDB 查詢（Gemini）

execute_query(database_alias="mongo_cluster", query='''{"name": "John Doe"}''', params={"collection": "users"})