如何通过接口解决这个问题？

Question

我正在使用Entity Framework，这就是我的问题所在，但它可能不相关，因为我认为这是一个通用的C＃问题。

我有一个MyContext类，它有很多DbSet<T>个属性。

为了对它进行单元测试，我编辑了MyContext，所以这些是IDbSet<T>而是模拟它。 IDbSet<T>是Entity Framework的一部分，DbSet实现它，因此两者几乎完全相同。

这一切都很有效，因为我可以做IDbSet<T>我能做的所有事情DbSet<T>

除了一件事之外的所有事情 - DbSet都有一个名为SqlQuery(..)的方法，我也希望能够从我的IDbSet调用。

由于我无法改变DbSet或IDbSet，这给我留下了一个难题。我无法理解我是如何做到这一点所以我的IDbSet能够以某种合理的方式添加SearchQuery(..)方法。

这有意义吗？我很困惑所以我可能忽略了一些非常简单的东西，例如copy＆amp;粘贴IDbSet并重命名并添加SearchQuery()。有什么想法吗？

Answer 1

修改

我相信我得到了你想要的东西 - 到处都是包装纸，可能比它的价值更麻烦，但值得一看：

// An interface which implements IDbSet<T> and adds on the method you want
public interface IExtendedDbSet<T> : IDbSet<T>
    where T : class
{
    DbSqlQuery<T> SqlQuery(string sql, object[] parameters);
}

// Implement this interface by wrapping around a regular DbSet<T>.
// You implement all the methods and properties by just wrapping the DbSet<T>
// calls
public class ExtendedDbSet<T> : IExtendedDbSet<T>
    where T : class
{
    private readonly DbSet<T> _dbSet;

    public ExtendedDbSet(DbSet<T> dbSet) { _dbSet = dbSet; }

    DbSqlQuery<T> IExtendedDbSet<T>.SqlQuery(string sql, object[] parameters)
    {
        return _dbSet.SqlQuery(sql, parameters);
    }

    T IDbSet<T>.Add(T entity) { return _dbSet.Add(entity); }
    T IDbSet<T>.Attach(T entity) { return _dbSet.Attach(entity); }
    TDerivedEntity IDbSet<T>.Create<TDerivedEntity>() { return _dbSet.Create<TDerivedEntity>(); }
    T IDbSet<T>.Create() { return _dbSet.Create(); }
    T IDbSet<T>.Find(params object[] keyValues) { return _dbSet.Find(keyValues); }
    ObservableCollection<T> IDbSet<T>.Local { get { return _dbSet.Local; } }
    T IDbSet<T>.Remove(T entity) { return _dbSet.Remove(entity); }
    IEnumerator<T> IEnumerable<T>.GetEnumerator() { return ((IEnumerable<T>)_dbSet).GetEnumerator(); }
    IEnumerator IEnumerable.GetEnumerator() { return ((IEnumerable)_dbSet).GetEnumerator(); }
    Type IQueryable.ElementType { get { return ((IQueryable)_dbSet).ElementType; } }
    Expression IQueryable.Expression { get { return ((IQueryable)_dbSet).Expression; } }
    IQueryProvider IQueryable.Provider { get { return ((IQueryable)_dbSet).Provider; } }
}

// A regular context class, no special interfaces to implement or
// custom properties or anything.
public class MyContext : DbContext
{
    public DbSet<Car> Cars { get; set; }
}

// An interface representing your context, which exposes extended DbSet<T>
// for your sets. Also define SaveChanges() and whatever else you may need
// to call on your context object.
public interface IMyContext
    : IDisposable
{
    IExtendedDbSet<Car> Cars { get; }
    int SaveChanges();
}

// A wrapper around your regular context. For each set, return an
// ExtendedDbSet<T> wrapper.
public class MyContextWrapper : IMyContext
{
    private readonly MyContext _myContext;

    public MyContextWrapper(MyContext myContext) { _myContext = myContext; }

    IExtendedDbSet<Car> IMyContext.Cars
    {
        get { return new ExtendedDbSet<Car>(_myContext.Cars); }
    }

    void IDisposable.Dispose()
    {
        _myContext.Dispose();
    }

    int IMyContext.SaveChanges()
    {
        return _myContext.SaveChanges();
    }
}

// Define your context variable as IMyContext, and create it
// by creating a wrapper around a regular context. The properties
// of the interface will be extended wrappers around your sets.
internal class Program
{
    private static void Main(string[] args)
    {
        using (IMyContext context = new MyContextWrapper(new MyContext()))
        {
            Console.WriteLine(context.Cars.SqlQuery("select 1", new object[0]));
        }
    }
}

Answer 2

嗯，它可能不是最优雅的解决方案，但您可以简单地编写一个这样的扩展方法：

public IEnumerable<T> SearchQuery(this IDbSet<T> set, string query)
{
    var dbSet = set as DbSet<T>;
    if (dbSet != null)
    {
        return dbSet.SqlQuery(query);
    }
    else 
    {
        throw new NotSupportedException();
    }
}