c# - 在完成返回后关闭 IDataReader

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

通过回调在 ProcessReader 中调用 CloseDBConnections 怎么样？

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();
  }
}