如何在启动时根据数据库模式检查模型

时间:2015-05-11 10:14:04

标签: c# .net entity-framework-6 code-first

我有一个缓存数据库,如果架构中缺少新字段,可以简单地删除并重新创建。我不会被移民困扰,它总是让我烦恼。

那我该如何进行这次启动检查呢? db.Database.CompatibleWithModel返回true,但针对不匹配的DataSet运行Linq查询将引发异常。

围绕此类事情的大多数答案都指向迁移。

1 个答案:

答案 0 :(得分:0)

只需使用反射测试所有DataSet的可能性:

try
{
    using (var db = new MyContext())
    {
        //db.myDataSet.Take(0).SingleOrDefault(); //The following reflection achieves what this line would do
        var t = db.GetType();
        var props = t.GetProperties(BindingFlags.Public | BindingFlags.Instance);
        foreach (var prop in props)
        {
            if (!prop.PropertyType.Name.StartsWith("DbSet"))
                continue;
            var getMethod = prop.GetMethod;
            var dataSet = getMethod.Invoke(db, null);
            var dataSetType = dataSet.GetType();
            var queryableType = typeof(Queryable);
            var takeMethod = queryableType.GetMethod("Take");
            takeMethod = takeMethod.MakeGenericMethod(dataSetType.GenericTypeArguments);
            var takeResult = takeMethod.Invoke(null, new object[] { dataSet, 0 });
            var sodMethod = queryableType.GetMethods().Where(m => m.Name == "SingleOrDefault" && m.GetParameters().Length == 1).Single();
            sodMethod = sodMethod.MakeGenericMethod(dataSetType.GenericTypeArguments);
            var finalResult = sodMethod.Invoke(null, new object[] { takeResult });
        }
    }
}
catch (Exception ex)
{
    if (ex.InnerException != null && ex.InnerException.InnerException != null && ex.InnerException.InnerException.Message.StartsWith("Invalid column"))
    {
        MyHelperFunctions.MarkDatabaseForReset();
    }
    else
        throw; //Show error to user - we didn't expect this one
}

有点粗糙,但它对我有用。很多问题的范围,但应该在99%的时间内工作(例如,代码优先的datacontext上的所有属性通常都是数据集)