SQLAlchemy 实现 PostgreSQL 的 `UPDATE ... LIMIT 1`，哪种方法更优？

## SQLAlchemy 中实现 PostgreSQL 的 UPDATE ... LIMIT 1

在使用 SQLAlchemy 时，你可能会遇到在 PostgreSQL 数据库中执行 UPDATE ... LIMIT 1 操作的需求。然而，由于 SQLAlchemy 中无法在调用 update() 时使用 limit()，因此需要采用变通方法来实现此功能。

方法一：子查询

子查询方法通过获取要更新的行的 ID 来实现 UPDATE ... LIMIT 1 操作。

# 获取要更新行的 ID
updated_row_id = session.query(Log.id) \
    .order_by(Log.id.desc()) \
    .limit(1) \
    .scalar()

# 使用 ID 过滤更新查询
session.query(Log) \
    .filter(Log.id == updated_row_id) \
    .update({ 'analyzed': True })

方法二：CTE

CTE（公共表表达式）方法在 SQLAlchemy 1.4 或更高版本中可用，它通过创建一个临时表来存储要更新行的 ID，从而实现 UPDATE ... LIMIT 1 操作。

# 创建 CTE
updated_row_id = session.query(Log.id) \
    .order_by(Log.id.desc()) \
    .limit(1) \
    .cte('recent_log') \
    .column('id')

# 使用 CTE 过滤更新查询
session.query(Log) \
    .filter(Log.id == updated_row_id) \
    .update({ 'analyzed': True })

性能考虑

子查询方法比 CTE 方法的性能通常较差，因为子查询需要执行两次。而 CTE 方法只需要执行一次查询。

示例代码

from sqlalchemy import Column, Integer, Boolean, create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.sql import Select, subquery

engine = create_engine('postgresql://user:password@host:port/database')
Session = sessionmaker(bind=engine)
session = Session()

class Log(object):
    __tablename__ = 'logs'
    id = Column(Integer, primary_key=True)
    analyzed = Column(Boolean)

# 子查询方法
updated_row_id = session.query(Log.id) \
    .order_by(Log.id.desc()) \
    .limit(1) \
    .scalar()

session.query(Log) \
    .filter(Log.id == updated_row_id) \
    .update({ 'analyzed': True })

# CTE 方法
updated_row_id = session.query(Log.id) \
    .order_by(Log.id.desc()) \
    .limit(1) \
    .cte('recent_log') \
    .column('id')

session.query(Log) \
    .filter(Log.id == updated_row_id) \
    .update({ 'analyzed': True })

session.commit()

结论

在 PostgreSQL 中实现 UPDATE ... LIMIT 1 操作可以通过子查询或 CTE 方法。CTE 方法性能更优，但仅适用于 SQLAlchemy 1.4 或更高版本。

常见问题解答

1. 为什么 SQLAlchemy 中无法在调用 update() 时使用 limit()？
答：UPDATE ... LIMIT 1 是 MySQL 的特有功能，SQLAlchemy 无法支持。

2. 子查询方法和 CTE 方法哪个性能更好？
答：CTE 方法性能更好，因为只需要执行一次查询。

3. 如何在 SQLAlchemy 1.4 之前的版本中使用 CTE 方法？
答：无法使用。

4. 除了子查询和 CTE 方法外，还有其他实现 UPDATE ... LIMIT 1 的方法吗？
答：没有。

5. 如何知道哪种方法更适合我的具体场景？
答：如果你使用的是 SQLAlchemy 1.4 或更高版本，则建议使用 CTE 方法。对于较早版本的 SQLAlchemy，则只能使用子查询方法。