Azure Cosmos DB跨容器多操作如何实现回滚？

情况1：所有实体（A、B、C）位于同一个Cosmos DB容器 #

Azure Cosmos DB的TransactionalBatch原生支持单容器内的原子事务，所有操作要么全部成功，要么自动回滚，完全匹配你的需求。

代码示例 #

using Azure;
using Azure.Cosmos;
using Azure.Cosmos.Fluent;

// 初始化Cosmos客户端（建议以单例方式注入到项目中）
var cosmosClient = new CosmosClientBuilder("<你的连接字符串>")
    .WithApplicationName("TransactionalDemo")
    .Build();

var database = cosmosClient.GetDatabase("<数据库名称>");
var container = database.GetContainer("<容器名称>");

// 事务要求所有操作属于**同一个分区键**，需确保A、B、C的分区键值一致
var partitionKey = new PartitionKey("<共同的分区键值>");

using var batch = container.CreateTransactionalBatch(partitionKey);

// 批量添加操作：插入A、更新B、删除C
batch.CreateItem<EntityA>(new EntityA { Id = "A-1", /* 其他属性赋值 */ });
batch.ReplaceItem<EntityB>("B-1", new EntityB { Id = "B-1", /* 更新后的属性 */ });
batch.DeleteItem<EntityC>("C-1");

try
{
    TransactionalBatchResponse response = await batch.ExecuteAsync();
    if (response.IsSuccessStatusCode)
    {
        Console.WriteLine("所有操作执行成功");
    }
    else
    {
        Console.WriteLine($"事务执行失败: {response.StatusCode}");
    }
}
catch (CosmosException ex)
{
    // 捕获异常时，事务会自动回滚所有已提交的变更
    Console.WriteLine($"事务异常回滚: {ex.Message}");
}

关键限制：TransactionalBatch要求所有操作必须归属同一个分区键，如果你的实体分区键不同，需要调整数据模型（比如将关联实体放到同一分区），或者改用跨容器方案。

情况2：实体分布在多个Cosmos DB容器 #

由于Cosmos DB不支持跨容器事务，此时需要通过**补偿事务（Saga模式）**实现最终一致性——每个操作都对应可逆的回滚逻辑，一旦某一步失败，就按逆序回滚之前完成的所有操作。

实现思路 #

1. 按顺序执行核心操作：插入A → 更新B → 删除C
2. 每一步执行成功后标记状态
3. 若某一步失败，按逆序执行补偿：比如删除C失败，先回滚B到更新前状态，再删除已插入的A

代码示例 #

// 假设三个实体分别在不同容器中
var containerA = database.GetContainer("ContainerA");
var containerB = database.GetContainer("ContainerB");
var containerC = database.GetContainer("ContainerC");

EntityB originalB = null;
bool isACreated = false;
bool isBUpdated = false;

try
{
    // 1. 插入实体A
    await containerA.CreateItemAsync(new EntityA { Id = "A-1" }, new PartitionKey("<A的分区键>"));
    isACreated = true;

    // 2. 查询B的原始状态用于回滚，再执行更新
    originalB = await containerB.ReadItemAsync<EntityB>("B-1", new PartitionKey("<B的分区键>"));
    await containerB.ReplaceItemAsync(new EntityB { Id = "B-1", /* 更新属性 */ }, "B-1", new PartitionKey("<B的分区键>"));
    isBUpdated = true;

    // 3. 删除实体C
    await containerC.DeleteItemAsync<EntityC>("C-1", new PartitionKey("<C的分区键>"));

    Console.WriteLine("所有操作执行成功");
}
catch (Exception ex)
{
    Console.WriteLine($"操作失败，开始回滚: {ex.Message}");

    // 逆序执行补偿操作
    if (isBUpdated && originalB != null)
    {
        try
        {
            await containerB.ReplaceItemAsync(originalB, "B-1", new PartitionKey("<B的分区键>"));
            Console.WriteLine("已回滚实体B");
        }
        catch (Exception rollbackEx)
        {
            Console.WriteLine($"回滚B失败: {rollbackEx.Message}");
            // 此处需记录日志，后续人工介入修复
        }
    }

    if (isACreated)
    {
        try
        {
            await containerA.DeleteItemAsync<EntityA>("A-1", new PartitionKey("<A的分区键>"));
            Console.WriteLine("已回滚实体A");
        }
        catch (Exception rollbackEx)
        {
            Console.WriteLine($"回滚A失败: {rollbackEx.Message}");
            // 记录日志，后续人工处理
        }
    }
}

注意：补偿事务无法保证严格原子性，只能保证最终一致性。如果回滚操作本身失败，必须依赖日志记录和人工介入来修复数据不一致问题。

内容的提问来源于stack exchange，提问作者Trần Như Tuấn Anh