Question

我正在制作一个遵循工作单元模式的界面。我的界面如下所示：

public interface IDataContext : IDisposable
{
    void SaveChanges();   
    TSource Create<TSource>(TSource toCreate) where TSource : class;
    TSource Update<TSource>(TSource toUpdate) where TSource : class;
    bool Delete<TSource>(TSource toDelete) where TSource : class;
    IQueryable<TSource> Query<TSource>();
}

到目前为止一切顺利。现在我在我的数据层中实现它，它使用EF4作为数据提供者。我想出了“查询”方法的代码，但我认为它不是很干净，我觉得有一种聪明的方法可以做到，但我无法弄明白。

public IQueryable<TSource> Query<TSource>()
    {
        var type = typeof(TSource);
        IQueryable item = null;

        if (type == typeof(Activity)) item = _entities.ActivitySet;
        if (type == typeof(Company)) item = _entities.CompanySet;
        if (type == typeof(Phase)) item = _entities.PhasesSet;
        if (type == typeof(Project)) item = _entities.ProjectSet;
        if (type == typeof(ProjectState)) item = _entities.ProjectStateSet;
        if (type == typeof(ProjectType)) item = _entities.ProjectTypeSet;
        if (type == typeof(User)) item = _entities.UserSet;
        if (type == typeof(UserType)) item = _entities.UserTypeSet;
        if (item != null) return item as IQueryable<TSource>;

        throw new NotImplementedException(string.Format("Query not implemented for type {0}", type.FullName));
    }

我在这里看到的问题是所有的IF每次都经过测试，虽然我可以将它们连接在一起，但是看起来仍然非常糟糕。其他问题是我必须为每个可能添加的新实体手动添加一行，但这不是我主要考虑的问题。

有人对此有什么好的建议吗？感谢。

Answer 1

如果您使用的是EF4，则可以调用CreateObjectSet＆lt;＆gt;。

using System;
using System.Collections.Generic;
using System.Data.Objects;
using System.Linq;

namespace WindowsFormsApplication1
{
    static class Program
    {
        [STAThread]
        static void Main()
        {
            var context = new DataContext(new NorthwindEntities());
            var list = context.Query<Customer>().ToList();
            var list2 = context.Query<Customer>().ToList();
        }
    }

    public class DataContext : IDataContext
    {
        private Dictionary<Type, object> _objectSets = new Dictionary<Type,object>();
        private ObjectContext _entities;

        public DataContext(ObjectContext objectContext)
        {
            this._entities = objectContext;
        }

        public IQueryable<T> Query<T>()
            where T : class
        {
            Type entityType = typeof(T);
            ObjectSet<T> objectSet;

            if (this._objectSets.ContainsKey(entityType))
            {
                objectSet = this._objectSets[entityType] as ObjectSet<T>;
            }
            else
            {
                objectSet = this._entities.CreateObjectSet<T>();
                this._objectSets.Add(entityType, objectSet);
            }

            return objectSet;
        }
    }

    interface IDataContext
    {
        IQueryable<T> Query<T>() where T : class;
    }
}

如果您使用的是EF1，可以调用CreateQuery＆lt;＆gt;但您还需要找到实体设置名称。

using System;
using System.Collections.Generic;
using System.Data.Metadata.Edm;
using System.Data.Objects;
using System.Linq;

namespace WindowsFormsApplication2
{
    static class Program
    {
        [STAThread]
        static void Main()
        {
            var context = new DataContext(new NorthwindEntities());
            var list = context.Query<Customer>().ToList();
        }
    }

    public class DataContext : IDataContext
    {
        private Dictionary<string, string> _entitySets;
        private ObjectContext _entities;

        public DataContext(ObjectContext objectContext)
        {
            this._entities = objectContext;
        }

        public IQueryable<T> Query<T>()
            where T : class
        {
            return this._entities.CreateQuery<T>(this.GetEntitySetName<T>());
        }

        private string GetEntitySetName<T>()
            where T : class
        {
            if (this._entitySets == null)
            {
                // create a dictionary of the Entity Type/EntitySet Name
                this._entitySets = this._entities.MetadataWorkspace
                                                 .GetItems<EntityContainer>(DataSpace.CSpace)
                                                 .First()
                                                 .BaseEntitySets.OfType<EntitySet>().ToList()
                                                 .ToDictionary(d => d.ElementType.Name, d => d.Name);
            }

            Type entityType = typeof(T);

            // lookup the entity set name based on the entityType
            return this._entitySets[entityType.Name];
        }
    }

    interface IDataContext
    {
        IQueryable<T> Query<T>() where T : class;
    }
}

Answer 2

一种方法可能是不使用自动生成的ObjectContext和所有预定义的ObjectSet，而是调用：

public IQueryable<TSource> Query<TSource>
{
  return _entities.CreateObjectSet<TSource>();
}

但是我不确定每次要执行查询时创建对象集会产生什么性能影响。我通常对对象集进行一些惰性初始化（使用字典来存储已经创建的对象集）并将它们重用于单个工作单元实例。

工作单位和EF

2 个答案: