MySQL使用INSERTONDUPLICATEKEYUPDATE实现Upsert方法

在文章实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《SQLAlchemy 实现“存在则更新，不存在则插入”（即 Upsert）有多种方式，具体取决于你使用的数据库类型（如 PostgreSQL、MySQL 等）。以下是几种常见方法的总结和示例。✅ 一、使用 on_conflict_do_update（PostgreSQL）PostgreSQL 提供了 ON CONFLICT 子句来实现 Upsert。示例代码：from sqlalchemy import create_engine, Column, Integer, String from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.dialects.postgresql import insert Base = declarative_base() class User(Base): __tablename__ = 'users' id = Column(Integer, primary_key=True) name = Column(String) email = Column(String, unique=True) engine = create_engine('postgresql://user:password@localhost/dbname') # 插入或更新数据 stmt = insert(User).values(name='Alice', email='alice@example.com') stmt = stmt.on_conflict_do_update( index_elements=['email'], set_={'name': 'Alice'} ) with engine.connect() as conn: conn.execute(stmt)说明：index_elements=['email']：指定唯一索引字段。set_：冲突时更新的字段。✅ 二、使用 INSERT ... ON DUPLICATE KEY UPDATE（MySQL）MySQL 使用 `ON》，聊聊，希望可以帮助到正在努力赚钱的你。

SQLAlchemy实现upsert有三种方式：一是Core层数据库原生语法（PostgreSQL用on_conflict_do_update、MySQL用on_duplicate_key_update、SQLite用on_conflict_do_update），高效且避免竞态；二是ORM层bulk_upsert_mappings（2.0+），批量处理、不触发事件；三是merge()，自动查再更/插但有性能开销。

SQLAlchemy 提供了多种方式实现“存在则更新，不存在则插入”（upsert），具体写法取决于你用的是 Core 还是 ORM，以及后端数据库是否原生支持 upsert（如 PostgreSQL 的 ON CONFLICT、MySQL 的 ON DUPLICATE KEY UPDATE、SQLite 的 ON CONFLICT REPLACE/DO UPDATE）。

使用 Core：database-native upsert（推荐，高效）

这是最直接、性能最好的方式，利用数据库原生语法，避免先查再插/更的竞态问题和额外查询开销。

• PostgreSQL：用 insert().on_conflict_do_update()

from sqlalchemy.dialects.postgresql import insert
stmt = insert(User).values(
id=123,
name='Alice',
email='alice@example.com'
)
stmt = stmt.on_conflict_do_update(
indexelements=['id'],  # 冲突依据（主键或唯一索引列）
set=dict(name=stmt.excluded.name, email=stmt.excluded.email)  # 更新字段
)
conn.execute(stmt)

• MySQL：用 insert().on_duplicate_key_update()

from sqlalchemy.dialects.mysql import insert
stmt = insert(User).values(id=123, name='Alice', email='alice@example.com')
stmt = stmt.on_duplicate_key_update(
name=stmt.inserted.name,
email=stmt.inserted.email
)
conn.execute(stmt)

• SQLite（≥3.24.0）：用 insert().on_conflict_do_update()

from sqlalchemy.dialects.sqlite import insert
stmt = insert(User).values(id=123, name='Alice', email='alice@example.com')
stmt = stmt.on_conflict_do_update(
indexelements=['id'],
set=dict(name=stmt.excluded.name, email=stmt.excluded.email)
)
conn.execute(stmt)

使用 ORM：bulk_upsert_mappings（批量高效，适合多条）

适用于一次性处理多条记录，底层调用对应 dialect 的原生 upsert，比逐条 session.merge() 快得多。

• 要求模型定义了主键或唯一约束（用于判断“存在”）
• 只支持 2.0+ 版本（session.bulk_upsert_mappings()）

data = [
    {'id': 123, 'name': 'Alice', 'email': 'a1@example.com'},
    {'id': 456, 'name': 'Bob', 'email': 'b1@example.com'},
]
session.bulk_upsert_mappings(User, data)
session.commit()

注意：bulk_upsert_mappings 不触发 ORM 事件（如 @event.listens_for），也不做对象实例化，纯数据层操作。

兼容性兜底：用 merge()（简单但有代价）

如果数据库不支持原生 upsert，或你需要 ORM 对象生命周期管理（如事件、关系加载），可用 session.merge()：

• 它会先按主键/identity 查询是否存在；存在则更新属性，不存在则新建
• 每次调用都产生一次 SELECT 查询（可能变慢，尤其高并发时）
• 适合单条、低频、开发/测试场景

user = User(id=123, name='Alice', email='alice@example.com')
merged = session.merge(user)  # 自动判断插入或更新
session.commit()

关键注意事项

• 确保目标列（如 id 或 email）上有主键或唯一约束，否则 upsert 无法判断冲突
• PostgreSQL 的 index_elements 必须匹配某个唯一索引（可以是复合唯一索引的一部分）
• MySQL 要求冲突列属于某个 PRIMARY KEY 或 UNIQUE KEY
• ORM 的 merge() 按主键比较，不支持按非主键字段（如 email）判断“存在”——除非你重写 __hash__ 和 __eq__，但这不推荐

今天关于《MySQL使用INSERTONDUPLICATEKEYUPDATE实现Upsert方法》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！