sqlalchemy 如何优雅写“存在则更新,不存在则插入”

SQLAlchemy实现upsert有五种方式:一、PostgreSQL用on_conflict_do_update;二、通用ORM用merge();三、MySQL用on_duplicate_key_update;四、SQLite用prefix_with(“OR REPLACE”);五、跨库兼容用SELECT+INSERT/UPDATE。

sqlalchemy 如何优雅写“存在则更新,不存在则插入”

如果您在使用 SQLAlchemy 时需要实现“存在则更新,不存在则插入”的逻辑,即通常所说的 upsert 操作,但又希望避免重复查询或手动判断主键/唯一约束冲突,则需依赖数据库原生支持与 SQLAlchemy 的适配机制。以下是几种不同数据库后端下的具体实现方式:

一、使用 on_conflict_do_update(PostgreSQL)

PostgreSQL 提供了 INSERT … ON CONFLICT DO UPDATE 语法,SQLAlchemy 通过 on_conflict_do_update 方法直接封装该能力,需指定冲突目标(如唯一索引列)和更新字段。

1、确保模型已定义 __table_args__ 中包含 UniqueConstraint 或主键。

2、构造 insert 语句并调用 on_conflict_do_update,传入 index_elements 参数指定冲突列。

3、在 set_ 方法中传入字典,声明冲突时要更新的字段及值,例如 {“name”: excluded.name, “updated_at”: func.now()}

二、使用 merge(通用 ORM 方式)

SQLAlchemy 的 merge() 方法基于主键或标识符查找对象:若 session 中已存在相同主键的对象,则同步属性;若不存在,则添加新实例。该方法不依赖数据库层 upsert 语法,但要求模型定义明确主键且数据一致性由应用层保障。

1、创建目标模型实例,并设置其 主键字段(如 id)或 唯一标识字段(如 name)。

2、调用 session.merge(instance),此时 SQLAlchemy 自动执行 SELECT + INSERT/UPDATE 判断逻辑。

3、执行 session.commit() 完成持久化。

三、使用 insert().on_duplicate_key_update(MySQL)

MySQL 支持 INSERT … ON DUPLICATE KEY UPDATE,SQLAlchemy 通过 on_duplicate_key_update 扩展实现,需确保表上存在 PRIMARY KEY 或 UNIQUE 约束。

1、使用 insert(Model) 构造语句。

Petalica Paint

Petalica Paint

用AI为你的画自动上色!

下载

2、链式调用 on_duplicate_key_update,传入关键字参数形式的更新字段映射,例如 name=inserted.name, updated_at=func.now()

3、执行 session.execute(stmt) 并提交事务。

四、使用 SQLite 的 INSERT OR REPLACE

SQLite 不支持标准 upsert 语法,但提供 INSERT OR REPLACE 行为:当违反 PRIMARY KEY 或 UNIQUE 约束时,自动删除旧行并插入新行。SQLAlchemy 可通过 prefix_with(“OR REPLACE”) 实现等效效果。

1、构造标准 insert 语句:insert(Model).values(…)

2、添加前缀 prefix_with(“OR REPLACE”),使生成 SQL 为 INSERT OR REPLACE INTO …

3、注意该方式会触发 DELETE + INSERT,可能影响外键级联或自增 ID 连续性。

五、手动执行 SELECT + INSERT/UPDATE(跨库兼容方案)

当目标数据库不支持原生 upsert 或需精细控制冲突逻辑时,可显式分离查询与写入步骤。此方式完全可控,但引入额外 round-trip 开销与竞态风险,建议配合 SELECT FOR UPDATE 或事务隔离级别提升安全性。

1、使用 session.query(Model).filter(…).with_for_update() 查询是否存在匹配记录。

2、根据查询结果调用 session.add()session.query().update()

3、统一提交事务以保证原子性。

https://www.php.cn/faq/1991518.html

发表回复

Your email address will not be published. Required fields are marked *