使用反射自动化EF5 Code First DbModelBuilder

时间:2013-05-15 03:27:31

标签: c# reflection entity-framework-5

我正在尝试使用反射在OnModelCreating中自动化DbModelBuilder对象。

我有许多继承自基类的类:枚举 在数据库中,我希望能够设置这些表的Id,以便在OnModelCreating中调用

modelBuilder.Entity<SomeClass>()
            .Property(sc => sc.SomeClassId)
            .HasDatabaseGeneratedOption(DatabaseGeneratedOption.None);

我想使用反射自动执行此操作,因为如果我将模型应用于数据库时忘记执行此操作,则必须将其删除并重新开始。

我知道我可以在我的班级模型上加上一个属性,但我不想在那里。

这里的代码。出于某种原因,当我尝试调用我的通用方法时,它失败了。有没有人有任何建议?

//Get all of my Models / Tables that I want to be able to specify a proimary key for.

var enumerationPropertyInfos = thisDbContext
    .GetType()
    .GetProperties();
    .Where(p => p.PropertyType.IsGenericType
        && p.PropertyType.GetGenericTypeDefinition() == typeof(DbSet<>)
        && p.PropertyType.GetGenericArguments().First().BaseType == typeof(MyEnumerationBase));


foreach (PropertyInfo pi in enumerationPropertyInfos)
{
    //modelBuilder.Entity<SomeClass>()
    var config = modelBuilder.GetType()
        .GetMethod("Entity")
        .MakeGenericMethod(pi.PropertyType)
        .Invoke(modelBuilder, null);

    //Prepare .Property(...)
    var property = config.GetType().GetMethods().Where(m => m.Name == "Property").First();

    var propertyExpression = typeof(Expression<>)
        .MakeGenericType(typeof(Func<,>)
        .MakeGenericType(pi.PropertyType, typeof(int)));

    //Prepare e => e.SomeClassId
    var paramEx = Expression.Parameter(pi.PropertyType.GetGenericArguments().First(), "e");
    var lambdaEx = Expression.Lambda(Expression.Property(paramEx, pi.PropertyType.GetGenericArguments().First().Name + "Id"), paramEx);

    //Execute .Property(e => e.SomeClassId)
    PrimitivePropertyConfiguration propertyResult = (PrimitivePropertyConfiguration)property
        .MakeGenericMethod(typeof(Expression))
        .Invoke(config, new[] { lambdaEx });

    propertyResult.HasDatabaseGeneratedOption(DatabaseGeneratedOption.None);
}

1 个答案:

答案 0 :(得分:2)

您的代码中有两个错误,首先是这一行:

//modelBuilder.Entity<SomeClass>()
var config = modelBuilder.GetType()
    .GetMethod("Entity")
    .MakeGenericMethod(pi.PropertyType)
    .Invoke(modelBuilder, null);

这相当于:

modelBuilder.Entity<DbSet<SomeClass>>();

将其更改为:

var config = modelBuilder.GetType()
    .GetMethod("Entity")
    .MakeGenericMethod(pi.PropertyType.GetGenericArguments().First())
    .Invoke(modelBuilder, null);

第二个错误出现在这一行:

var propertyResult = (PrimitivePropertyConfiguration)property
    .MakeGenericMethod(typeof(Expression))
    .Invoke(config, new[] { lambdaEx });

您需要将SomeClassId的类型作为通用参数传递给Property()方法,但您要传递typeof(Expression)。假设SomeClassId的类型为int,请将行更改为:

var propertyResult = (PrimitivePropertyConfiguration)property
    .MakeGenericMethod(typeof(int))
    .Invoke(config, new[] { lambdaEx });