初探MongoDB中的数据库事务

背景

最近有一个业务需要同时写两个表，并且需要保证数据的正确性。我们的项目线上的 MongoDB 版本是 4.0，查了一下发现 4.0 版本的 MongoDB 已经支持副本集中的事务了，于是做了一下调研。

准备环境

鉴于 MongoDB 的事务支持需要在副本集中才生效，我用 Docker-Compose 部署了一个一主两从结构的 4.0 版本 MongoDB 副本集(部署过程参考了文章)。

初始化数据

假设有两个表，订单表(orders)和库存表(inventory), 对于一个下单行为：创建一个购买了商品 A x 件的订单，需要增加一条 orders 表记录表明购买了 x 件商品 A，同时商品 A 对应的库存表记录的库存也要减少 x 件。

import pymongo

client = pymongo.MongoClient(host="mongodb://root:123@localhost:37011")

db = client.get_database("test")
db.inventory.insert_one({"sku": "abc", "qty": 100000})

在代码中我们初始化了十万件商品 abc。

编写事务代码

import pymongo
import sys

client = pymongo.MongoClient(host="mongodb://root:123@localhost:37011")

db = client.get_database("test")

with client.start_session() as session:
    with session.start_transaction():

        print(
            db.orders.insert_one(
                {"sku": "abc", "qty": 1000},
                session=session,
            ).inserted_id
        )
        # sys.exit(0)
        print(
            db.inventory.find_one_and_update(
                {"sku": "abc"},
                {"$inc": {"qty": -1000}},
                return_document=pymongo.ReturnDocument.AFTER,
                session=session,
            )
        )

print(
    list(
        db.orders.aggregate(
            [
                {"$match": {"sku": "abc"}},
                {"$group": {"_id": None, "total": {"$sum": "$qty"}}},
            ]
        )
    )[0]
)