启用EnableRetryOnFailure时，调用SaveChanges前重置DbContext的最优方案？

问题场景 #

需要实现一批实体修改操作：若任意步骤抛出异常，需撤销所有已做更改，并将指定实体标记为故障状态。但启用EnableRetryOnFailure后，手动创建事务会触发如下异常：

System.InvalidOperationException: The configured execution strategy 'SqlServerRetryingExecutionStrategy' does not support user initiated transactions. Use the execution strategy returned by 'DbContext.Database.CreateExecutionStrategy()' to execute all the operations in the transaction as a retriable unit.

ChatGPT提供的ResetChanges方法可实现重置，但存在优化空间，以下是更优方案：

方案1：优化版ChangeTracker重置 #

原ResetChanges方法可通过Reload()替代手动赋值，提升可靠性（尤其适用于含复杂属性/关系的实体）：

private void ResetChanges(DbContext db)
{
    foreach (var entry in db.ChangeTracker.Entries().ToList())
    {
        switch (entry.State)
        {
            case EntityState.Modified:
                entry.Reload(); // 直接从数据库重载实体，确保状态完全恢复
                break;
            case EntityState.Added:
                entry.State = EntityState.Detached; // 移除新增实体的跟踪
                break;
            case EntityState.Deleted:
                entry.State = EntityState.Unchanged; // 恢复已标记删除的实体
                break;
        }
    }
}

// 调用逻辑
try
{
    ChangeAnEntity(db);
    ChangeAnotherEntity(db);
    CreateSomeEntities(db);
    ExecuteMethodThatCanThrow();

    await db.SaveChangesAsync();
}
catch
{
    ResetChanges(db);
    SetAnEntityAsFaulted(db);
    await db.SaveChangesAsync();
}

优势 #

• 比手动赋值OriginalValues更可靠，能自动处理关系属性的状态恢复
• 逻辑清晰，仅处理需要重置的实体状态

方案2：结合EF Core执行策略（推荐） #

既然启用了重试策略，应使用EF Core官方推荐的CreateExecutionStrategy()包装整个操作，既符合执行策略要求，又能处理 transient 异常重试：

var executionStrategy = db.Database.CreateExecutionStrategy();

await executionStrategy.ExecuteAsync(async () =>
{
    try
    {
        // 执行所有实体修改操作
        ChangeAnEntity(db);
        ChangeAnotherEntity(db);
        CreateSomeEntities(db);
        ExecuteMethodThatCanThrow();

        await db.SaveChangesAsync();
    }
    catch (Exception)
    {
        // 重置所有变更
        ResetChanges(db);
        // 标记故障实体
        SetAnEntityAsFaulted(db);
        // 保存故障标记
        await db.SaveChangesAsync();
        // 终止重试（若需要重试可移除throw，需确保操作幂等）
        throw;
    }
});

优势 #

• 完全兼容EnableRetryOnFailure配置，避免手动事务冲突
• 自动处理数据库连接中断等 transient 异常的重试（需确保操作幂等）
• 保证整个流程的一致性，符合EF Core工作单元设计

方案3：新DbContext实例重置（简化场景） #

若当前DbContext的ChangeTracker状态复杂，直接创建新实例可避免状态处理的繁琐：

try
{
    ChangeAnEntity(db);
    ChangeAnotherEntity(db);
    CreateSomeEntities(db);
    ExecuteMethodThatCanThrow();

    await db.SaveChangesAsync();
}
catch
{
    await db.DisposeAsync();
    // 使用原上下文的配置创建新实例
    using var newDbContext = new YourDbContext(db.Options);
    
    SetAnEntityAsFaulted(newDbContext);
    await newDbContext.SaveChangesAsync();
}

适用场景 #

• 实体修改逻辑简单，无需保留原上下文状态
• 不想处理ChangeTracker的复杂状态恢复逻辑

关键注意事项 #

1. 使用Reload()时，需确保数据库连接正常，且目标实体仍存在于数据库中
2. 执行策略包装的操作需保证幂等性，避免重试时重复执行修改逻辑
3. 标记故障实体的操作需独立于原修改流程，确保重置后能正确执行

内容的提问来源于stack exchange，提问作者Kralizek