我如何才能绕过接口的这个问题

本文关键字：接口 问题 | 更新日期: 2023-09-27 18:24:22

我使用的是实体框架，这就是我的问题所在，但它可能并不相关，因为我认为这是一个通用的C#问题。

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

为了对其进行单元测试，我编辑了MyContext，所以这些都是IDbSet<T>，并对其进行了模拟。IDbSet<T>是实体框架的一部分，DbSet实现了它，因此两者几乎相同。

一切都很好，因为我可以对IDbSet<T>做我能对DbSet<T> 做的一切

除了一件事——DbSet有一个名为SqlQuery(..)的方法，我也想从我的IDbSet中调用它。

由于我不能更改DbSet或IDbSet，这给我留下了一个难题。我不知道该怎么做，所以我的IDbSet能够以某种逻辑方式添加一个SearchQuery(..)方法。

这有道理吗？我很困惑，所以我可能忽略了一些非常简单的东西，比如copy&粘贴CCD_ 14并将其重命名并添加CCD_。有什么想法吗？

我如何才能绕过接口的这个问题

编辑

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

// 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]));
        }
    }
}

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

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();
    }
}
相关文章：