EF CodeFirst：如何获取上下文所有实体？用于OnModelCreating创建索引

当然有办法啦！刚好你的需求是在OnModelCreating里获取实体，给带特定特性的属性创建索引，我给你两种适配不同EF版本的实用方案：

方案1：适配EF6（你的代码风格对应的版本） #

直接在OnModelCreating中获取（推荐，更准确） #

EF6的DbModelBuilder可以直接获取已注册的所有实体类型，不会遗漏任何通过Fluent API注册的实体：

protected override void OnModelCreating(DbModelBuilder modelBuilder)
{
    // 从模型构建器中提取所有实体类型
    var allEntityTypes = modelBuilder.Models
        .SelectMany(model => model.EntityTypes)
        .Select(entityType => entityType.ClrType)
        .ToList();

    // 遍历实体，查找带目标特性的属性并创建索引
    foreach (var entityType in allEntityTypes)
    {
        // 替换成你要检测的自定义特性
        var targetProperties = entityType.GetProperties()
            .Where(prop => prop.GetCustomAttributes(typeof(YourCustomIndexAttribute), true).Any());

        foreach (var prop in targetProperties)
        {
            // 给属性添加索引（EF6的写法）
            modelBuilder.Entity(entityType)
                .Property(prop.Name)
                .HasColumnAnnotation("Index", new IndexAnnotation(new IndexAttribute()));
        }
    }

    base.OnModelCreating(modelBuilder);
}

封装成GetAllEntities()方法 #

如果需要复用这个逻辑，也可以在MyContext里封装一个方法，通过反射提取所有DbSet<T>对应的实体类型：

public class MyContext : DbContext 
{
    public virtual DbSet<Customer> Customers { get; set; }
    public virtual DbSet<Provider> Providers { get; set; }
    public virtual DbSet<Product> Products { get; set; }

    public IEnumerable<Type> GetAllEntities()
    {
        // 反射获取所有DbSet<T>类型的属性，提取泛型参数T（实体类型）
        return GetType().GetProperties()
            .Where(prop => prop.PropertyType.IsGenericType 
                           && prop.PropertyType.GetGenericTypeDefinition() == typeof(DbSet<>))
            .Select(prop => prop.PropertyType.GetGenericArguments()[0]);
    }

    // 你的OnModelCreating方法...
}

⚠️ 注意：这种反射方法会漏掉没有对应DbSet属性但通过Fluent API注册的实体，所以优先推荐第一种从modelBuilder获取的方式。

方案2：适配EF Core（如果后续升级版本） #

EF Core的API更简洁，直接通过ModelBuilder的Model属性就能获取所有实体：

protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    // 获取所有实体类型
    var allEntityTypes = modelBuilder.Model.GetEntityTypes()
        .Select(entityType => entityType.ClrType)
        .ToList();

    // 处理带特性的属性
    foreach (var entityType in allEntityTypes)
    {
        var targetProperties = entityType.GetProperties()
            .Where(prop => prop.GetCustomAttributes(typeof(YourCustomIndexAttribute), inherit: true).Any());

        foreach (var prop in targetProperties)
        {
            // EF Core中创建索引的写法更直观
            modelBuilder.Entity(entityType)
                .HasIndex(prop.Name);
        }
    }

    base.OnModelCreating(modelBuilder);
}

如果要封装方法，EF Core里直接用模型API更可靠：

public IEnumerable<Type> GetAllEntities()
{
    return Model.GetEntityTypes()
        .Select(entityType => entityType.ClrType);
}

内容的提问来源于stack exchange，提问作者Rafael León