.net - 为EF提供即时的自定义数据库提供程序

Question

作为概要分析工具的一部分，我有一个自定义的ADO.NET堆栈，它充当标准ADO.NET的“装饰器”，因此它实际上不做任何工作-它只是传递调用(但记录日志，等等)。 )。除其他外，我从我的自定义连接中提供了一个DbProviderFactory，它实现了IServiceProvider并提供了一个自定义DbProviderServices。

这对于大多数工具(包括LINQ-to-SQL)都非常有效-但是， Entity Framework 不满意。

例如-说我有:

MetadataWorkspace workspace = new MetadataWorkspace(
     new string[] { "res://*/" }, 
     new Assembly[] { Assembly.GetExecutingAssembly() });
using(var conn = /* my custom wrapped DbConnection */)
{
    var provider = DbProviderServices
          .GetProviderServices(conn); // returns my custom DbProviderServices
    var factory = DbProviderServices
          .GetProviderFactory(conn); // returns my custom DbProviderFactory
    ...

到目前为止一切顺利-以上两行有效；返回正确的(自定义)提供程序信息。

现在我们可以添加一个EF模型:

    using (var ec = new EntityConnection(workspace,conn))
    using (var model = new Entities(ec))
    {
        count = model.Users.Count(); // BOOM!
    }

失败并出现以下异常:

Unable to cast object of type '(my custom connection)' to type 'System.Data.SqlClient.SqlConnection'.

这是在分配连接到命令的过程中；本质上，它已默认使用 SSpace的sql-server提供程序，并生成了一个裸露的 SqlCommand。然后，它尝试将 conn分配给生成的命令，该命令不起作用(如果所有装饰器都就位，它将正确运行，而使用装饰的 DbCommand代替)。

现在，将其动态包装的整个要点意味着我真的不需要更改EDMX来注册一个单独的工厂。我只想让它知道我的谎言，该死的谎言和装饰品。

通过侵入 SSpace的胆量(设置一个我无权了解或滥用的 private readonly字段)，可以进行以下工作:

    StoreItemCollection itemCollection =
        (StoreItemCollection)workspace.GetItemCollection(DataSpace.SSpace);
    itemCollection.GetType().GetField("_providerFactory",
        BindingFlags.NonPublic | BindingFlags.Instance)
        .SetValue(itemCollection, factory);

正确放置此位置后， SSpace将使用正确的工厂。但是，这显然令人讨厌。

所以:我在这里错过了一个把戏吗？我该如何用较少的苛刻措施拦截EF提供者？

Answer 1

应该可以使用EFProviderWrappers中的方法。我知道您提到您尝试过，但是我自己还是成功地克服了这种困难。

在上面的代码示例中，您无法将标准的MetadataWorkspace传递给EntityConnection构造函数，因为MetadataWorkspace仍将指向SSDL部分中的原始DbProviderFactory(在您的情况下为SqlClient)。我知道您不想直接修改EDMX/SSDL(我也没有)，但是EFProviderWrappers工具箱提供了一些可能对您有用的辅助方法。 EntityConnectionWrapperUtils.cs类特别有用。它将使用您原始的EDMX(甚至可以从嵌入式资源中提取它)并更新XML，使其指向您的自定义DbProviderFactory。它在运行时完成了所有这些操作，因此您无需进行任何更改。您需要为您的DbProviderFactory命名一个不变的名称并注册它-如果您还没有这样做的话。

然后，您可以将自定义MetadataWorkspace以及自定义DbConnection传递给EntityConnection构造函数，然后EF在需要执行DbCommand时应调用您的工厂。