Question

实际上它不一定是IDataReader。

我有一个像这样的函数：

public IEnumerable<MyClass> GetObjects() 
{
  IDataReader dr = GetSomeDataReader();
  while (dr.Read())
  {
    yield return new MyClass(dr);
  }
  CloseDBConnections();
}

这样做很好，直到我像这样重构它：

public IEnumerable<MyClass> GetObjects() 
{
  IDataReader dr = GetSomeDataReader();
  try
  {
    return ProcessReader(dr);
  } finally {
    CloseDBConnections();
  }
}
public IEnumerable<MyClass> ProcessReader(IDataReader dr)
{
  while (dr.Read())
  {
    yield return new MyClass(dr);
  }
}

这不起作用，因为执行CloseDBConnections()时枚举尚未处理。

从.ToList()返回时调用GetObjects实际执行枚举，但到那时连接已被破坏，IDataReader失败。

在我的实例中CloseDBConnections无法在新的ProcessReader函数中调用，因为IDataReader可能来自其他来源（在此实例中重新分解的整个点）

是否有合理的解决方法或者我是否必须复制枚举代码？

我尝试将ProcessReader作为yield return而不是return，但这不起作用，因为C＃（可以理解）认为我正在尝试添加IEnumerable到IEnumerable！

Answer 1

不漂亮，但这应该有效。

public IEnumerable<MyClass> GetObjects() 
{
  IDataReader dr = GetSomeDataReader();
  try
  {
    foreach (var result in ProcessReader(dr))
    {
      yield return result;
    }
  } finally {
    CloseDBConnections();
  }
}

Answer 2

如何通过回调在CloseDBConnections中呼叫ProcessReader？

public IEnumerable<MyClass> GetObjects() 
{
  return ProcessReader(GetSomeDataReader(), CloseDBConnections);
}

public IEnumerable<MyClass> ProcessReader(IDataReader dr, Action OnFinished)
{
  try
  {
    while (dr.Read())
      yield return new MyClass(dr);
  }
  finally
  {
    if (OnFinished != null) 
      OnFinished();
  }
}

收益率返回完成后关闭IDataReader

2 个答案: