C# 9.0新特性詳解系列之二：擴展方法GetEnumerator支持foreach循環

時間 2020-12-01

標籤測試 this code 對象接口 get string ast 欄目 C# 简体版

原文原文鏈接

1.介紹

咱們知道，咱們要使一個類型支持foreach循環，就須要這個類型知足下面條件之一：測試

該類型實例若是實現了下列接口中的其中之一：this
- System.Collections.IEnumerable
- System.Collections.Generic.IEnumerable<T>
- System.Collections.Generic.IAsyncEnumerable<T>
該類型中有公開的無參GetEnumerator()方法，且其返回值類型必須是類，結構或者接口，同時返回值類型具備公共 Current 屬性和公共無參數且返回類型爲 Boolean的MoveNext 方法。code

上面的第一個條件，歸根結底仍是第二個條件的要求，由於這幾個接口，裏面要求實現的仍是GetEnumerator方法，同時，接口中GetEnumerator的返回值類型IEnumerator接口中要實現的成員和第二條中返回值類型的成員相同。對象

C#9.0以前，是不支持採起擴展方法的方式給類型注入GetEnumerator方法，以支持foreach循環的。從C#9.0以後，這種狀況獲得了支持。接口

2. 應用與示例

在這裏，咱們定義一個People類，它能夠枚舉其全部組員Person，而且在其中定義了MoveNext方法和Current屬性。同時，咱們也經過擴展方法給People注入了GetEnumerator方法。這樣，咱們就可使用foreach來枚舉People對象了。get

首先，咱們來定義一個Person記錄:string

public record Person(string FirstName, string LastName);

下來，咱們來建立People類型,用來描述多個Person對象，並提供GetEnumerator返回值類型中所需的Current屬性和MoveNext方法。在此，咱們沒有實現任何接口:it

public class People：IDisposable//: IEnumerator<Person>
{
    int position = -1;

    private Person[] _people { get; init; }
    public People(Person[] people)
    {
        _people = people;
    }

    public bool MoveNext()
    {
        position++;
        return (position < _people.Length);
    }

    public Person Current
    {
        get
        {
            try
            {
                return _people[position];
            }
            catch (IndexOutOfRangeException)
            {
                throw new InvalidOperationException();
            }
        }
    }

    public void Reset()
    {
        position = -1;
    }

    public void Dispose()
    {
        Reset();
    }
}

須要注意的是People中，因爲沒有經過使用前面的接口來實現支持foreach功能，這樣就存在一個問題，就是第一次foreach循環完成後，狀態尚未恢復到初始狀態，第二次使用foreach進行枚舉就沒有可用項。所以咱們添加了Reset方法用於手工恢復回初始狀態，若是想讓foreach能自動恢復狀態，就讓People實現接口IDisposable，並在其實現中，調用Reset方法。io

而後，咱們定義擴展方法，給People注入GetEnumerator方法ast

static class PeopleExtensions
{
    //public static IEnumerator<T> GetEnumerator<T>(this IEnumerator<T> people) => people;
    public static People GetEnumerator(this People people) => people;
}

最後，只要引用了擴展方法所在的命名空間，foreach循環就可使用了。

var PersonList = new Person[3]
{
    new ("John", "Smith"),
    new ("Jim", "Johnson"),
    new ("Sue", "Rabon"),
};

var people = new People(PersonList);
foreach (var person in people)
{
    Console.WriteLine(person);
}

到這裏，咱們就完成了利用擴展方法來實現foreach循環的示例，爲了方便拷貝測試，咱們全部的代碼放在一塊兒就以下所示：

var PersonList = new Person[3]
{
    new ("John", "Smith"),
    new ("Jim", "Johnson"),
    new ("Sue", "Rabon"),
};

var people = new People(PersonList);
foreach (var person in people)
{
    Console.WriteLine(person);
}

public record Person(string FirstName, string LastName);

public class People：IDisposable//: IEnumerator<Person>
{
    int position = -1;

    private Person[] _people { get; init; }
    public People(Person[] people)
    {
        _people = people;
    }

    public bool MoveNext()
    {
        position++;
        return (position < _people.Length);
    }

    public Person Current
    {
        get
        {
            try
            {
                return _people[position];
            }
            catch (IndexOutOfRangeException)
            {
                throw new InvalidOperationException();
            }
        }
    }

    public void Reset()
    {
        position = -1;
    }

    public void Dispose()
    {
        Reset();
    }
}

static class PeopleExtensions
{
    //public static IEnumerator<T> GetEnumerator<T>(this IEnumerator<T> people) => people;
    public static People GetEnumerator(this People people) => people;
}