如何将通用配置应用于ef核心中的所有实体

时间:2018-11-13 07:05:22

标签: c# entity-framework-core

我的应用程序中具有从基本实体派生的实体,该实体使用ef核心代码优先方法。

基类

public abstract class BaseEntity<T> : IEntity<T>
{
    [Key]
    [DatabaseGenerated(DatabaseGeneratedOption.Identity)]
    public T Id { get; set; }

    object IEntity.Id { get { return Id; } set { } }

    private DateTime? createdOn;
    [DataType(DataType.DateTime)]
    public DateTime CreatedOn { get => createdOn ?? DateTime.Now; set => createdOn = value; }

    [DataType(DataType.DateTime)]
    public DateTime? ModifiedOn { get; set; }

    public bool IsDeleted { get; set; }
    // Auto increment for all entities.
    public int OrderId { get; set; }
}

和实体

public class UserEntity : BaseEntity<int>
{
    public string EmployeeId { get; set; }
    public string FullName { get; set; }
    public string Email { get; set; }
}

我可以对每个实体的.ValueGeneratedOnAdd()属性OrderId应用OnModelCreating方法每个实体,但是,有一种方法可以对所有实体应用通用规则自己不重复吗?

2 个答案:

答案 0 :(得分:3)

在缺少自定义约定的情况下,您可以使用典型的modelBuilder.Model.GetEntityTypes()循环,标识目标实体类型并调用通用配置。

由于基本的 generic 类,您的情况下的识别有点复杂,但是可以通过迭代Type.BaseType并检查BaseEntity<>来实现。找到它之后,您可以检索稍后将需要的通用参数T

如果您不想使用实现IEnityTypeConfiguration<TEntity> generic 类,则可以将实现方式放入这样的 generic conconstrained 方法中

static void Configure<TEntity, T>(ModelBuilder modelBuilder)
    where TEntity : BaseEntity<T>
{
    modelBuilder.Entity<TEntity>(builder =>
    {
        builder.Property(e => e.OrderId).ValueGeneratedOnAdd();
    });
}

将实际的实体类型TEntity传递到modelBuilder.Enity方法至关重要,因为否则EF Core会将您传递的任何内容视为实体类型并配置TPH继承。

调用方法需要反思-使用MakeGenericMethod然后使用Invoke查找通用方法定义。

以下是封装在静态类中的所有内容:

public static class BaseEntityConfiguration
{
    static void Configure<TEntity, T>(ModelBuilder modelBuilder)
        where TEntity : BaseEntity<T>
    {
        modelBuilder.Entity<TEntity>(builder =>
        {
            builder.Property(e => e.OrderId).ValueGeneratedOnAdd();
        });
    }

    public static ModelBuilder ApplyBaseEntityConfiguration(this ModelBuilder modelBuilder)
    {
        var method = typeof(BaseEntityConfiguration).GetTypeInfo().DeclaredMethods
            .Single(m => m.Name == nameof(Configure));
        foreach (var entityType in modelBuilder.Model.GetEntityTypes())
        {
            if (entityType.ClrType.IsBaseEntity(out var T))
                method.MakeGenericMethod(entityType.ClrType, T).Invoke(null, new[] { modelBuilder });
        }
        return modelBuilder;
    }

    static bool IsBaseEntity(this Type type, out Type T)
    {
        for (var baseType = type.BaseType; baseType != null; baseType = baseType.BaseType)
        {
            if (baseType.IsGenericType && baseType.GetGenericTypeDefinition() == typeof(BaseEntity<>))
            {
                T = baseType.GetGenericArguments()[0];
                return true;
            }
        }
        T = null;
        return false;
    }
}

现在您所需要的只是从OnModelCreating覆盖内调用它:

modelBuilder.ApplyBaseEntityConfiguration();

答案 1 :(得分:1)

在EF6中,您可以使用:

modelBuilder.Properties<int>().Where(p=>p.Name == "OrderId").Configure(c => c.HasDatabaseGeneratedOption(DatabaseGeneratedOption.Identity));