SQLAlchemy：深入浅出 flush() 与 commit() 的区别

2024-03-20 20:05:56

SQLAlchemy：精通 flush() 和 commit() 的区别

引言

作为数据处理的重中之重，SQLAlchemy 赋予了我们掌控数据库交互的强大工具。在处理大量数据时，理解 flush() 和 commit() 方法之间的差异至关重要。本文将深入探讨它们的独特作用，指导你优化内存使用，并巧妙地管理事务处理。

揭开 flush() 的面纱

flush() 方法如同一位勤劳的清洁工，将当前会话中的所有未决更改（包括新增、更新和删除）打包送往数据库。然而，它不会像 commit() 那样提交事务，这意味着你的更改仍然处于可撤销的状态。

flush() 的魅力在于，它只将未决更改发送给数据库，而无需提交整个事务。这就像在每次小改动后立即整理你的桌面，避免凌乱不堪，从而节省内存空间。

揭秘 commit() 的威力

与 flush() 的短暂性不同，commit() 方法犹如法官的一锤定音，将当前会话中的所有未决更改提交到数据库，同时宣告事务的终结。一旦提交，这些更改就不可逆转，牢牢地刻在了数据库的石碑之上。

与 flush() 相比，commit() 需要更多内存，因为它需要存储所有更改，直至事务提交。但这额外的开销也带来了永久性的好处：你的更改将安全无虞地保存在数据库中。

内存使用：一场争夺战

在内存管理的战场上，flush() 是效率的化身。它只将未决更改发送给数据库，减少了内存占用。而 commit() 则更像一个贪心的大户，将所有更改都收入囊中，占据了更多的内存空间。

当面对庞大的数据集时，flush() 的优势尤为明显。它可以定期释放内存，避免系统不堪重负。

事务处理：分道扬镳

事务处理是数据库管理的基石，commit() 和 flush() 在此领域扮演着截然不同的角色。commit() 负责提交事务，而 flush() 则保持事务处于开放状态。

在提交事务之前，你可以使用 flush() 对更改进行额外的处理或验证。这就像在提交报告前仔细检查一遍，确保一切无误。

最佳实践：协同合作

在处理大量数据时，将 flush() 和 commit() 巧妙结合，可以达到事半功倍的效果。定期调用 flush() 以释放内存，并在适当的时候提交事务以永久保存更改。以下是遵循的准则：

大型数据集：定期 flush()，释放内存。
更改处理：在提交前使用 flush()，进行额外处理或验证。
暂缓提交：在不需要立即保存更改时，避免 commit()。

示例：亲身体验

想象你有一个包含 500 万条记录的会话。使用以下代码，你可以领略 flush() 和 commit() 的协同作用：

# 设定会话
session = Session()

# 循环处理数据
for row in rows:
    session.add(row)
    # 每处理 10 万条记录就刷新一次
    if row.id % 100000 == 0:
        session.flush()

# 所有数据处理完毕，提交事务
session.commit()