收益回報完成後關閉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()執行枚舉這不起作用尚未處理。

從GetObjects返回時調用.ToList()是實際執行枚舉的內容，但此時連接已被破壞，並且IDataReader失敗。

在我的實例CloseDBConnections不能從新ProcessReader函數中調用，因爲IDataReader可能來自其他來源（再保在這種情況下，整點）

是否有一個合理的辦法解決這個或者我將不得不復制枚舉代碼？

我試圖把調用ProcessReader爲yield return，而不是return但這並不工作，因爲C＃（可以理解）認爲我想添加一個IEnumerable到IEnumerable！

Answer 1

如何通過回調調用ProcessReaderCloseDBConnections？

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

Answer 2

不漂亮，但這應該工作。

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