如何在EF Core中为索引指定WITH (SORT_IN_TEMPDB = ON) ON [PRIMARY]？

完全可以实现，核心是通过扩展EF Core的IndexBuilder和自定义SQL生成逻辑来达成，具体步骤如下：

1. 定义标记注解类 #

先做一个简单的注解类，用来标记索引是否需要启用SORT_IN_TEMPDB：

public class SortInTempDbAnnotation
{
    public bool IsEnabled { get; }

    public SortInTempDbAnnotation(bool enabled)
    {
        IsEnabled = enabled;
    }
}

2. 给IndexBuilder加扩展方法 #

写一个扩展方法，让你能在Fluent API链里直接调用SortInTempDB()：

public static class IndexBuilderExtensions
{
    public static IndexBuilder SortInTempDB(this IndexBuilder builder, bool enabled = true)
    {
        builder.HasAnnotation("SortInTempDB", new SortInTempDbAnnotation(enabled));
        return builder;
    }
}

3. 重写SQL Server迁移SQL生成器 #

因为EF Core默认不会生成SORT_IN_TEMPDB的语句，所以要自定义SQL生成逻辑，继承SQL Server的迁移生成器：

public class CustomSqlServerMigrationsGenerator : SqlServerMigrationsSqlGenerator
{
    public CustomSqlServerMigrationsGenerator(
        MigrationsSqlGeneratorDependencies dependencies,
        IMigrationsAnnotationProvider annotationsProvider)
        : base(dependencies, annotationsProvider)
    {
    }

    protected override void Generate(CreateIndexOperation operation)
    {
        // 先获取默认生成的SQL
        var originalSql = Dependencies.SqlGenerationHelper.GenerateCreateIndex(operation);
        
        // 检查当前索引是否有SortInTempDB注解
        if (operation.FindAnnotation("SortInTempDB")?.Value is SortInTempDbAnnotation annotation && annotation.IsEnabled)
        {
            // 插入WITH子句，拼接ON [PRIMARY]
            var modifiedSql = originalSql.Insert(originalSql.IndexOf("INDEX"), " WITH (SORT_IN_TEMPDB = ON)")
                             + " ON [PRIMARY];";
            Dependencies.CommandBuilder.Append(modifiedSql);
            return;
        }

        // 没有注解就用默认逻辑
        base.Generate(operation);
    }
}

4. 注册自定义生成器 #

在DbContext的服务配置里替换默认的SQL生成器：

services.AddDbContext<YourDbContext>(options =>
{
    options.UseSqlServer("你的连接字符串");
    options.ReplaceService<IMigrationsSqlGenerator, CustomSqlServerMigrationsGenerator>();
});

5. 最终使用方式 #

现在就能像你期望的那样写Fluent API了：

modelBuilder.Entity<SomeTable>()
    .HasIndex(c => c.SomeField)
    .IsClustered(false)
    .SortInTempDB();

执行Add-Migration和Update-Database后，生成的索引创建SQL会自动带上WITH (SORT_IN_TEMPDB = ON) ON [PRIMARY];。

注意：这个方案只适用于SQL Server，因为SORT_IN_TEMPDB是SQL Server独有的索引选项。

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