[特殊字符] PostgreSQL MCP 开发指南

简介 🚀

PostgreSQL MCP 是一个基于 FastMCP 框架的 PostgreSQL 数据库交互服务。它提供了一套简单易用的工具函数,让你能够通过 API 方式与 PostgreSQL 数据库进行交互。

功能特点 ✨

  • 🔄 数据库连接管理与重试机制
  • 🔍 执行 SQL 查询并获取结果
  • 📋 数据库元数据操作(列出表、模式等)
  • 📝 表结构管理(创建表、描述表结构)
  • 💾 数据操作(插入、更新数据)

快速开始 🏃‍♂️

配置选项

PostgreSQL MCP 支持通过环境变量或命令行参数进行配置:

配置项 环境变量 默认值
主机地址 POSTGRES_HOST localhost
端口 POSTGRES_PORT 5432
用户名 POSTGRES_USER postgres
密码 POSTGRES_PASSWORD postgres
数据库 POSTGRES_DATABASE postgres
连接超时 POSTGRES_CONNECTION_TIMEOUT 10
重试次数 POSTGRES_CONNECT_RETRY_COUNT 3

运行服务

bash 复制代码
python postgresql-mcp.py --host localhost --port 5432 --user postgres --password yourpassword --database yourdb

API 功能 📚

执行 SQL 查询

python 复制代码
response = await execute_query(
    query="SELECT * FROM users WHERE age > %s",
    params=[18],
    db_config={"database": "custom_db"}
)

列出数据库中的表

python 复制代码
tables = await list_tables(schema_name="public")

获取表结构

python 复制代码
table_info = await describe_table(
    table_name="users",
    schema_name="public"
)

列出所有模式

python 复制代码
schemas = await list_schemas()

创建新表

python 复制代码
result = await create_table(
    table_name="new_table",
    columns=[
        {"name": "id", "type": "SERIAL", "primary_key": True},
        {"name": "name", "type": "VARCHAR(100)", "nullable": False},
        {"name": "created_at", "type": "TIMESTAMP", "default": "CURRENT_TIMESTAMP"}
    ]
)

插入数据

python 复制代码
result = await insert_data(
    table_name="users",
    data=[
        {"name": "张三", "age": 30, "email": "zhangsan@example.com"},
        {"name": "李四", "age": 25, "email": "lisi@example.com"}
    ]
)

更新数据

python 复制代码
result = await update_data(
    table_name="users",
    data={"status": "inactive", "updated_at": "CURRENT_TIMESTAMP"},
    condition="user_id = %s",
    params=[1001]
)

错误处理 🔧

服务会提供详细的错误信息,常见问题包括:

  • 连接失败(主机/端口错误)
  • 身份验证失败(用户名/密码错误)
  • 数据库不存在
  • SQL 语法错误
  • 表或列不存在
  • 违反表约束条件

实现细节 🔍

该服务基于以下关键技术:

  • psycopg2 库用于 PostgreSQL 连接
  • FastMCP 框架提供 API 接口
  • 参数化查询防止 SQL 注入
  • 自动重试机制提高可靠性

最佳实践 💡

  • 始终使用参数化查询防止 SQL 注入
  • 为所有更新操作提供 WHERE 条件
  • 利用连接配置优化数据库性能
  • 根据实际需求调整连接超时和重试次数

⚠️ 注意:在生产环境中使用时,请确保设置安全的数据库密码,并适当限制数据库用户权限。

GIthub源码

相关推荐
2301_793086873 分钟前
Mysql group by
数据库·mysql
jllllyuz19 分钟前
Spring中的事务是如何实现的
数据库·sql·spring
wangmengxxw40 分钟前
Spring-常用注解
java·数据库·spring·注解
m0_747266091 小时前
contentprovider实验+SQLite数据库的实现
数据库·sqlite
TDengine (老段)1 小时前
TDengine 中 TDgp 中添加机器学习模型
大数据·数据库·算法·机器学习·数据分析·时序数据库·tdengine
网安Ruler2 小时前
Web开发-PHP应用&Cookie脆弱&Session固定&Token唯一&身份验证&数据库通讯
前端·数据库·网络安全·php·渗透·红队
zfoo-framework4 小时前
线上redis的使用
数据库·redis·缓存
典孝赢麻崩乐急4 小时前
Redis学习-----Redis的基本数据类型
数据库·redis·学习
止水编程 water_proof5 小时前
MySQL——事务详解
数据库·mysql
爱喝水的鱼丶5 小时前
SAP-ABAP:SAP ABAP OpenSQL JOIN 操作权威指南高效关联多表数据
运维·开发语言·数据库·sap·abap