doris-mcp-server 的项目扩展与二次开发

项目的基础介绍

doris-mcp-server 是一个基于 Python 和 FastAPI 开发的后端服务项目，实现了 Model Context Protocol (MCP) 协议，允许客户端通过定义的 "Tools" 与其交互。该项目主要设计用于连接 Apache Doris 数据库，并可能利用大型语言模型（LLMs）执行如自然语言查询到 SQL 转换（NL2SQL）、查询执行、元数据管理和分析等任务。

项目的核心功能

• MCP 协议实现：提供标准的 MCP 接口，支持工具调用、资源管理和提示符交互。
• 流式 HTTP 通信：统一的 HTTP 端点，支持请求/响应和流式通信，以实现最佳性能和可靠性。
• 标准输入/输出通信：支持与 MCP 客户端如 Cursor 的直接集成。
• 企业级架构：模块化设计，包含工具管理器、监控工具模块、查询信息工具、资源管理器和提示符管理器等综合功能。
• 高级数据库特性：包括高性能的 SQL 执行、安全管理和元数据提取等。

项目使用了哪些框架或库？

• FastAPI：用于构建 API 的现代、快速（高性能）的 Web 框架，使用 Python 3.7+ 的异步和等待关键字。
• Python：项目基于 Python 3.12+ 版本开发。
• 其他依赖：项目依赖多种第三方库，如数据库连接库、安全库等，具体可见项目 requirements.txt 文件。

项目的代码目录及介绍

项目的代码目录结构如下：

.
├── doris_mcp_client
├── doris_mcp_server
│   ├── tools
│   │   ├── tools_manager.py
│   │   ├── resources_manager.py
│   │   ├── prompts_manager.py
│   │   └── ...
│   ├── utils
│   │   ├── query_executor.py
│   │   ├── security.py
│   │   ├── schema_extractor.py
│   │   └── ...
│   ├── config.py
│   ├── start_server.sh
│   └── ...
├── examples
├── test
├── .asf.yaml
├── Dockerfile
├── LICENSE.txt
├── Makefile
├── README.md
├── docker-compose.yml
├── generate_requirements.py
├── pyproject.toml
├── requirements-dev.txt
├── requirements.txt
└── ...

• doris_mcp_server：包含服务器的核心逻辑，包括工具管理、资源管理、提示符管理等功能模块。
• utils：包含查询执行、安全管理、模式提取等实用工具。
• config.py：集中管理配置信息。
• start_server.sh：启动服务器的脚本。

对项目进行扩展或者二次开发的方向

• 增加新的工具模块：根据业务需求，可以开发新的工具模块，扩展服务器的功能。
• 集成更多的数据库：除了 Apache Doris，可以扩展支持其他数据库，如 MySQL、PostgreSQL 等。
• 增强安全性：进一步完善安全框架，增加如认证、授权等安全特性。
• 性能优化：针对查询执行、资源管理等模块进行性能优化。
• 扩展监控和日志：增强监控和日志功能，便于问题的追踪和性能分析。
• 用户界面开发：为项目开发一个用户友好的界面，提升用户体验。
• 支持更多协议：扩展服务器以支持更多协议，提高与其他系统的兼容性。