[心得] IEnumerator, IEnumerable, and Yield

看板C_Sharp (C#)作者 (Easy)時間9年前 (2016/03/28 00:56), 9年前編輯推噓3(300)
留言3則, 3人參與, 最新討論串1/1
我來討鞭,以下是我在 C# 使用 IEnumerator, IEnumerable, and Yield 的整理心得, 目前無廣告無音樂網誌版本 http://goo.gl/d7fimi --- 這篇是因為工作上同事對於 yield return 的疑問,所以開始寫這篇文章,順便整理一下 自己的經驗並且複習。 IEnumerator,迭代器,是一個巡迴容器的介面: // namespace System.Collections public interface IEnumerator { object Current { get; } bool MoveNext(); void Reset(); } 使用方法很簡單,在一個迴圈中呼叫 MoveNext,檢查是否還有可列舉的元素,利用 Current 來取得目前的元素(Element): for (var itr = something.GetEnumerator(); itr.MoveNext(); ) { Console.Write(itr.Current); } 此外有泛型(Generic)的迭代器 IEnumerator,需多實做用於釋放資源的 Dispose 的介 面: // namespace System.Collections.Generic using System.Collections; public interface IEnumerator : IDisposable, IEnumerator { T Current { get; } bool MoveNext(); void Reset(); void Dispose(); } 使用上,建議在列舉完元素後,呼叫 Dispose 來釋放迭代器的資源,可以使用 using 使 得程式碼更易讀: using (var itr = something.GetEnumerator()) { while (itr.MoveNext()) { Console.Write(itr.Current); } } IEnumerable,提供一個取得迭代器方法,用該迭代器來列舉元素: // namespace System.Collections public interface IEnumerable { IEnumerator GetEnumerator(); } 另有定義較常使用的泛型的版本: // namespace System.Collections.Generic using System.Collections; public interface IEnumerable : IEnumerable { IEnumerator GetEnumerator(); } 在 C# 中,陣列 (Array) 以及大多的容器 (System.Collections 或是 System.Collections.Generic) 都有實做 IEnumerable 介面,意思是可以透過 GetEnumerator() 來列舉該容器中的元素。例如 List、int[] 或是 string,都可以使 用其迭代器來列舉容器中的元素。 以下是使用 C# 語法 foreach 來列舉 List 容器中的元素: List numbers = new List { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }; foreach (var number in numbers) { Console.Write(number); // Output: 0123456789 } 該 foreach 在編譯實作上,使用 IEnumerable 來取得 IEnumerator,透過迭代器列舉元 素,可轉換程式碼如下: List numbers = new List { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }; using(var itr = numbers.GetEnumerator()) { while (itr.MoveNext()) { var number = itr.Current; Console.Write(number); // Output: 0123456789 } } 也可以自訂不同的迭代器,來製作出自定的列舉方法,例如下面定義對於 List 容器的 Reverse 迭代器,將元素順序反轉輸出: struct ReverseListEnumerable<T> : IEnumerable<T> { List<T> data; public ReverseListEnumerable(List<T> data) { this.data = data; } public IEnumerator<T> GetEnumerator() { return new Enumerator(this.data); } IEnumerator IEnumerable.GetEnumerator() { return this.GetEnumerator(); } struct Enumerator : IEnumerator<T> { int index; List<T> data; public Enumerator(List<T> data) { this.data = data; this.index = data.Count; } public T Current { get { return this.data[this.index]; } } object IEnumerator.Current { get { return this.Current; } } public void Dispose() { // 這裡把 data 設定 null,當作釋放資源,之後就無法再使用 this.data = null; } public bool MoveNext() { this.index -= 1; return this.index >= 0; } public void Reset() { // 重置 Index this.index = this.data.Count; } } } List<int> numbers = new List<int> { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }; foreach(var number in new ReverseListEnumerable<int>(numbers)) { Console.Write(number); // Output: 9876543210 } 沒錯!自定義迭代器怎麼可以這麼又臭又長…,但 C# 提供一項語法,可以讓上述自定義 迭代器的程式碼,變得相當容易撰寫又好閱讀,那就是 yield。 Yield 上節的 Reverse 例子使用 yield 可以改成: IEnumerable<T> ReverseListEnumerable<T>(List<T> data) { for (var index = data.Count - 1; index >= 0; index--) { yield return data[index]; } } List<int> numbers = new List<int> { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }; foreach(var number in ReverseListEnumerable(numbers)) { Console.Write(number); // Output: 9876543210 } 沒有搞錯,就是如此簡單明瞭,C# 編譯器會把它處理成上節的那一個樣子,如果有興趣 可以參考 Reference 的文章連結 The implementation of iterators in C# and its consequences。 當第一次呼叫迭代器的 MoveNext 時,程式會一直執行,直到 yield return,此時 yield return 的元素回傳,並保留程式位置以及狀態,當下一次呼叫 MoveNext() 時, 會從該保留位置以及狀態繼續執行。 此外也可以使用 yield break 強制中斷迭代器列舉。下面例子,當列舉元素為 3 的倍數 時就中斷迭代: IEnumerable<int> MyEnumerable(IEnumerable<int> enumerable) { foreach (var itr in enumerable) { if (itr % 3 == 0) { yield break; } yield return itr; } } List<int> numbers = new List<int> { 1, 2, 3, 4, 5, 6, 7, 8, 9 }; foreach(var number in MyEnumerable(numbers)) { Console.Write(number); // Output: 12 } 延伸閱讀 System.Linq 前兩節中,範例實做 List 容器中 Reverse 迭代器,但這功能在 System.Linq 中已經有 所實做。System.Linq 提供許多對於 IEnumerable 物件的迭代器處理擴充,有哪些可以 使用,可以參考官方文件。以下是使用 System.Linq 來完成 Reverse 功能: using System.Collections.Generic; using System.Linq; List numbers = new List { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }; foreach(var number in numbers.Reverse()) { Console.Write(number); // Output: 9876543210 } -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 211.23.144.23 ※ 文章網址: https://www.ptt.cc/bbs/C_Sharp/M.1459097776.A.528.html

03/28 01:10, , 1F
03/28 01:10, 1F

03/28 04:00, , 2F
說到這個怎麼可以不提 Iterator Pattern 呢XD?
03/28 04:00, 2F
感謝你的建議,這部分我在寫的過程中確實是沒有想到,之後研究後再補充上去 ※ 編輯: easy0519 (60.251.127.173), 03/28/2016 11:02:39

03/28 11:08, , 3F
03/28 11:08, 3F
文章代碼(AID): #1M-12mKe (C_Sharp)
文章代碼(AID): #1M-12mKe (C_Sharp)