使用GeneratorOf struct

Question

我尝试理解生成器和序列使我想到了实现自己的列表数据结构并实现用于for循环的协议。我的代码：

class GSList<T> : SequenceType
{
    var Next : GSList<T>?
    var Value : T

    init(_ value: T, next : GSList<T>?)
    {
        self.Value = value
        self.Next = next
    }

    func add(list: GSList<T>)
    {
       if (self.Next != nil)
       {
            self.Next?.add(list)
       } else
       {
            self.Next = list
       }
    }

    typealias Generator = GeneratorOf<GSList<T>>

    func generate() -> Generator
    {
        var current: GSList<T>? = self
        println(current?.Value)

        return GeneratorOf
            { () -> GSList<T>? in

                let returnValue = current
                current = self.Next
                println(self.Value)
                println(current?.Value)
                return returnValue
        }
    }
}


var list1 = GSList(1, next: nil)
var list2 = GSList(2, next: nil)
var list3 = GSList(3, next: nil)
var list4 = GSList(4, next: nil)
var list5 = GSList(5, next: nil)


list1.add(list2)
list1.add(list3)
list1.add(list4)
list1.add(list5)

var generator = list1.generate()

generator.next()
generator.next()
generator.next()
generator.next()
generator.next()

它没有任何编译时错误，但问题是current变量未使用Next方法中的generate()值更新：

func generate() -> Generator
        {
            var current: GSList<T>? = self
            println(current?.Value)

            return GeneratorOf
                { () -> GSList<T>? in

                    let returnValue = current
                    current = self.Next
                    println(self.Value)
                    println(current?.Value)
                    return returnValue
            }
        }

因此，呼叫下一个generator.next()始终会返回第一个项目。

我该如何避免这种情况？

Answer 1

你的行

current = self.Next

应该是

current = current?.Next