Entity Framework中如何设置IsPrimary与SiteId关联的唯一性约束

这个需求在分站点或多租户场景里挺常见的，咱们可以通过Entity Framework的Fluent API结合数据库层面的过滤唯一约束来实现，下面分两种情况给你详细说明：

方法一：EF Core 5+ 用Fluent API配置过滤唯一索引 #

从EF Core 5开始，官方支持了过滤索引（Filtered Indexes），刚好能匹配你的需求。只需要在DbContext的OnModelCreating方法里添加如下配置：

protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    // 配置组合索引，仅对IsPrimary=true的记录启用唯一性
    modelBuilder.Entity<Test>()
        .HasIndex(t => new { t.SiteId, t.IsPrimary })
        .HasFilter("IsPrimary = 1") // 注意：这里用数据库原生语法，SQL Server中1代表true
        .IsUnique();
}

配置说明： #

• 这个索引是SiteId和IsPrimary的组合，但通过HasFilter只针对IsPrimary=true的记录生效
• 同一SiteId下最多只能有一条IsPrimary=true的记录，而IsPrimary=false的记录不受该约束限制，可以存在多条
• 如果使用MySQL/PostgreSQL，把过滤条件改成IsPrimary = TRUE即可

方法二：手动添加数据库层面的唯一约束（兼容旧版EF/EF Core） #

如果你用的是EF Core 5之前的版本，或者希望直接在数据库层面强制约束，可以通过迁移文件添加自定义SQL语句：

1. 先生成空迁移文件：Add-Migration AddUniqueConstraintForPrimaryTest
2. 打开生成的迁移文件，修改Up和Down方法：

protected override void Up(MigrationBuilder migrationBuilder)
{
    // 创建过滤唯一约束，只限制IsPrimary=true的SiteId唯一性
    migrationBuilder.Sql(@"
        CREATE UNIQUE INDEX IX_Test_SiteId_Primary ON Test (SiteId)
        WHERE IsPrimary = 1;
    ");
}

protected override void Down(MigrationBuilder migrationBuilder)
{
    // 回滚时删除索引
    migrationBuilder.DropIndex(
        name: "IX_Test_SiteId_Primary",
        table: "Test");
}

3. 执行迁移：Update-Database

额外建议：业务逻辑层提前校验 #

虽然数据库层面的约束已经能保证数据一致性，但建议在业务逻辑层也添加校验逻辑，比如在添加/更新Test实体时：

public async Task<bool> AddTestAsync(Test test)
{
    if (test.IsPrimary)
    {
        // 检查当前SiteId下是否已有IsPrimary=true的记录
        var existingPrimary = await _context.Tests
            .AnyAsync(t => t.SiteId == test.SiteId && t.IsPrimary);
        
        if (existingPrimary)
        {
            // 抛出异常或者返回错误提示
            throw new InvalidOperationException($"SiteId {test.SiteId} 已经存在主记录");
        }
    }
    
    _context.Tests.Add(test);
    await _context.SaveChangesAsync();
    return true;
}

这样可以在数据提交到数据库之前就拦截错误，提升用户体验，也减少数据库层面的异常抛出。

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