背景
最近有一个业务需要同时写两个表,并且需要保证数据的正确性。我们的项目线上的 MongoDB 版本是 4.0,查了一下发现 4.0 版本的 MongoDB 已经支持副本集中的事务了,于是做了一下调研。
准备环境
鉴于 MongoDB 的事务支持需要在副本集中才生效,我用 Docker-Compose 部署了一个一主两从结构的 4.0 版本 MongoDB 副本集(部署过程参考了[文章][1]。
初始化数据
假设有两个表,订单表(orders)和库存表(inventory), 对于一个下单行为:创建一个购买了商品 A x 件的订单,需要增加一条 orders 表记录表明购买了 x 件商品 A,同时商品 A 对应的库存表记录的库存也要减少 x 件。
import pymongoclient = pymongo.MongoClient( host="mongodb://root:123@localhost:37011" )db = client.get_database("test")db.inventory.insert_one({"sku": "abc", "qty": 100000})
在代码中我们初始化了十万件商品 abc。
编写事务代码
import pymongoimport sysclient