从代理对象检索实体框架DbContext

时间:2015-07-03 08:01:13

标签: entity-framework

我正在使用Entity framework 6,DBcontext Database First。 我有一些情况,当我有一个代理对象,我想得到DBContext。 我正在使用此代码:

using System.Data.Entity;
using System.Data.Entity.Infrastructure;
using System.Data.Objects;

namespace Sample
{
    public class MyDbContext : DbContext
{
    private static readonly Dictionary<ObjectContext, MyDbContext> contexts = 
        new Dictionary<ObjectContext, MyDbContext>();

    public static MyDbContext FromObjectContext(ObjectContext context)
    {
        lock (contexts)
        {
            if (contexts.ContainsKey(context))
                return contexts[context];
            return null; 
        }
    }

    public static MyDbContext FromObject(object obj)
    {
        var field = obj.GetType().GetField("_entityWrapper");
        var wrapper = field.GetValue(obj);
        var property = wrapper.GetType().GetProperty("Context");
        var context = (ObjectContext)property.GetValue(wrapper, null);
        return FromObjectContext(context);
    }

    public MyDbContext()
    {
        lock (contexts)
            contexts[((IObjectContextAdapter)this).ObjectContext] = this; 
    }

    protected override void Dispose(bool disposing)
    {
        lock (contexts)
            contexts.Remove(((IObjectContextAdapter)this).ObjectContext);
        base.Dispose(disposing);
    }

}

}

现在,使用此代码,我可以使用此代码获取DBContext:

var ctx = MyDataContext.FromObject(MyObj1);

除了一个案例外,此代码正在运行:

如果我添加一个新对象并调用SaveChanges,并且在此对象之后尝试获取DbContext,则会出现错误。在线:

Dim wrapper = field.GetValue(obj)

错误:

An unhandled exception of type 'System.NullReferenceException' occurred in myprog.exe

Additional information: Object reference not set to an instance of an object.

我也发现在这种情况下,这行不会返回任何内容:

var field = obj.GetType().GetField("_entityWrapper");

我该怎么办? 谢谢!

1 个答案:

答案 0 :(得分:0)

添加新对象时,新对象仍然是原始对象(不是具有代理的对象)。您可以使用DbSet.Create而不是new,因此您已经拥有了代理的实体。