Entity Framework处理重复键引发的PK/UKC 2601异常问题

我明白你遇到的这个头疼的问题——在事务内处理重复Label的唯一键冲突时，每次异常后重新调用SaveChanges()都会导致已插入的实体被重复写入数据库。这背后的核心原因其实是外部事务和EF上下文状态的脱节：

当你手动开启事务后，第一次SaveChanges()抛出唯一键异常时，EF并不会自动回滚你开启的外部事务。这意味着那些已经成功插入的实体（比如Wall、Post）已经留在数据库里，但EF上下文里这些实体的状态仍然是Added。当你处理完重复Label再次调用SaveChanges()时，EF会把这些状态为Added的实体再次插入，自然就产生了重复数据。

下面给你两个解决方案，优先推荐第一种，因为它从根源上避免了问题：

方案一：提前检查并复用Label（推荐） #

与其等到插入时抛出异常再处理，不如在构建对象图的阶段就先查询数据库，复用已经存在的Label。这种方式不仅避免了异常处理的复杂逻辑，性能也更好。

实现代码 #

首先添加一个获取或创建Label的方法：

public Label GetOrCreateLabel(string name, string kind)
{
    // 使用UPDLOCK和HOLDLOCK锁定查询范围，避免并发场景下的竞态冲突
    var existingLabel = Labels.FromSqlRaw(@"
        SELECT * FROM Labels WITH (UPDLOCK, HOLDLOCK) 
        WHERE Name = {0} AND Kind = {1}", name, kind)
        .SingleOrDefault();
    
    if (existingLabel != null)
    {
        return existingLabel;
    }
    
    var newLabel = new Label { Name = name, Kind = kind };
    Labels.Add(newLabel);
    return newLabel;
}

使用方式 #

在给Post关联Label的时候，调用这个方法即可：

// 替换原来直接创建Label的逻辑
var label = context.GetOrCreateLabel("旅游攻略", "PostTag");
post.Labels.Add(label);

这里的UPDLOCK和HOLDLOCK很重要，它能确保在高并发场景下，多个线程同时查询同一个Label时，只有一个线程能进入创建流程，避免出现短暂的“检查不存在→同时插入”导致的唯一键冲突。

方案二：修复事务内的异常处理流程 #

如果你必须保留异常处理的方式，那需要调整事务和SaveChanges()的逻辑，确保异常发生时回滚事务，并修正上下文状态后重新提交。

修改SaveChanges方法 #

调整SaveChanges()，让它在检测到Label重复时返回标记，通知外部需要重新处理：

private int _saveResult;
public override int SaveChanges()
{
    try
    {
        _saveResult = base.SaveChanges();
        return _saveResult;
    }
    catch (DbUpdateException ex)
    {
        var sqlException = ex.InnerException?.InnerException as SqlException;
        // 检查是否是唯一键冲突（2601/2627）
        if (sqlException != null && 
            sqlException.Errors.Any(se => se.Number is 2601 or 2627))
        {
            var duplicateLabelEntries = ex.Entries
                .Where(e => e.Entity.GetType() == typeof(Label))
                .ToList();
            
            if (duplicateLabelEntries.Any())
            {
                // 处理重复Label：关联数据库中已有的记录
                foreach (var entry in duplicateLabelEntries)
                {
                    HandleLabelDuplicates(entry);
                }
                // 返回-1标记，告诉外部需要重新保存
                return -1;
            }
        }
        // 非Label重复的异常，直接抛出
        throw;
    }
}

private void HandleLabelDuplicates(DbEntityEntry entry)
{
    var newLabel = (Label)entry.Entity;
    // 查询数据库中已存在的Label
    var existingLabel = Labels.Single(l => 
        l.Name.Equals(newLabel.Name) && 
        l.Kind.Equals(newLabel.Kind));
    
    // 替换所有Post中的Label引用
    foreach (var post in newLabel.Posts.ToList())
    {
        post.Labels.Remove(newLabel);
        post.Labels.Add(existingLabel);
    }
    
    // 从上下文中移除待插入的重复Label
    entry.State = EntityState.Detached;
}

修改事务处理逻辑 #

在事务中处理重新保存的逻辑，确保异常时回滚事务，重新提交：

using (var context = new YourDbContext())
{
    bool saveSucceeded = false;
    int finalResult = 0;
    DbContextTransaction transaction = null;

    try
    {
        while (!saveSucceeded)
        {
            transaction = context.Database.BeginTransaction();
            var result = context.SaveChanges();

            if (result != -1)
            {
                // 保存成功，提交事务
                transaction.Commit();
                finalResult = result;
                saveSucceeded = true;
            }
            else
            {
                // 需要重新保存，先回滚当前事务
                transaction.Rollback();
                transaction.Dispose();
                // 重置事务，准备重新提交
            }
        }
        return order.Id;
    }
    catch (Exception ex)
    {
        transaction?.Rollback();
        throw;
    }
    finally
    {
        transaction?.Dispose();
    }
}

这个方案的核心是：当检测到Label重复时，先回滚当前事务（清除已经插入的部分数据），修正上下文里的实体引用后，重新开启事务并保存，这样就不会出现重复插入的问题了。

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