SQLite 数据库演示

该项目展示了在 FastMCP 应用中使用具有异步支持的 SQLite 数据库的用法。

功能

• 使用 aiosqlite 实现异步 SQLite 数据库操作
• 自动初始化数据库并插入示例数据
• 类型安全的数据库上下文管理
• 正确处理连接生命周期

先决条件

• Python 3.7+
• pip（Python 包安装工具）

安装

1. 克隆此仓库
2. 安装所需的依赖项：

pip install -r requirements.txt

项目结构

• server.py：主要的 FastMCP 服务器实现，带有数据库上下文管理
• database.py：具有异步支持的 SQLite 数据库实现
• demo.db：SQLite 数据库文件（首次运行时自动创建）

数据库模式

演示数据库包含一个 users 表，结构如下：

CREATE TABLE users (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    name TEXT NOT NULL,
    email TEXT UNIQUE NOT NULL
)

首次运行时会自动插入以下示例数据：

• John Doe（john@example.com）
• Jane Smith（jane@example.com）

使用方法

1. 启动服务器：

python server.py

2. 数据库将在首次运行时自动初始化并插入示例数据。

3. 使用 query_db 工具从数据库中检索数据。

数据库操作

Database 类提供以下方法：

• connect()：创建新的数据库连接并初始化模式
• query()：执行样本查询以检索所有用户
• disconnect()：关闭数据库连接

自定义

要修改数据库模式或添加新表：

1. 编辑 database.py 中的 _init_db() 方法
2. 为 Database 类添加新的方法以实现自定义查询
3. 更新 server.py 中的 query_db 工具以使用新方法

错误处理

数据库实现包括：

• 服务器关闭时自动清理连接
• 正确处理异步上下文管理
• 使用 Python 类型提示实现类型安全