Db MCP Server

🚀 數據庫管理控制檯服務器

本數據庫管理控制檯服務器可支持多種主流數據庫，能與OpenAI的Agents SDK完美集成，還具備高可用性設計，可高效穩定地進行數據庫操作。

🚀 快速開始

運行服務器

# 以控制檯模式啟動
./server -mode=console

# 以SSE模式啟動（用於OpenAI代理集成）
./server -mode=sse

最基本的配置文件示例

{
    "connections": [
        {
            "id": "mysql1",
            "type": "mysql",
            "host": "mysql1",
            "port": 3306,
            "name": "db1",
            "user": "user1",
            "password": "pass1"
        }
    ]
}

✨ 主要特性

• 多數據庫支持：支持MySQL、PostgreSQL等多種主流數據庫
• SSE模式集成：完美兼容OpenAI的Agents SDK，支持通過SSE協議與人工智能代理交互
• 查詢超時配置：允許為每個數據庫連接單獨設置查詢超時時間，默認30秒
• 高可用性設計：內置連接池和負載均衡功能，確保數據庫操作高效穩定

💻 使用示例

控制檯操作命令

基本命令

• help：顯示所有可用命令
• query：執行SQL查詢，支持超時參數
• describe：查看錶結構信息
• execute：執行DML操作（如INSERT, UPDATE, DELETE）
• schema：獲取數據庫 schema 信息
• connections：管理數據庫連接狀態

示例

# 執行查詢操作
query mysql1 "SELECT * FROM users LIMIT 10"

# 帶超時設置的查詢
query mysql1 --timeout=60 "SELECT * FROM large_table"

# 查看錶結構
describe mysql1 users

OpenAI 代理 SDK 集成

兼容性說明

• 支持通過SSE協議與OpenAI的GPT-4o等模型交互
• 提供完整的數據庫工具鏈，包括查詢、描述和執行功能

基本集成示例

from openai import OpenAI
from agents.agent import Agent, ModelSettings
from agents.tools.mcp_server import MCPServerSse, MCPServerSseParams

# 初始化 MCP 服務器連接
db_server = MCPServerSse(
    params=MCPServerSseParams(
        url="http://localhost:9095/sse"
    ),
)

# 創建具備數據庫操作能力的代理
agent = Agent(
    name="Database Assistant",
    model="gpt-4o",
    model_settings=ModelSettings(temperature=0.1),
    instructions="""
    你是一個專業的數據庫助手，能夠執行SQL查詢、管理事務和分析表結構。
    """,
    tools=[db_server]
)

📚 詳細文檔

錯誤處理

常見錯誤

• 連接超時：檢查目標數據庫服務是否正常運行
• 權限不足：確認用戶擁有相應的數據庫權限
• 語法錯誤：仔細檢查SQL語句是否有拼寫或語法問題

日誌記錄

服務器默認輸出日誌到logs/目錄，支持DEBUG、INFO、WARNING、ERROR四級日誌級別。

數據庫連接池配置

{
    "connections": [
        {
            "id": "mysql1",
            "type": "mysql",
            "host": "mysql1",
            "port": 3306,
            "name": "db1",
            "user": "user1",
            "password": "pass1",
            "pool_size": 5,
            "max_idle_time": 60
        }
    ]
}

表格輸出格式

文獻記錄

所有文檔和設計思路均可在項目倉庫中找到。

⚠️ 重要提示

使用本工具前，請確保已閱讀並理解相關安全注意事項，特別是在生產環境中使用時要格外小心。