Lua中的迭代器浅析

勤奋小炮手 2014-09-13

今天学习的内容还蛮有意思的,让我兴奋了一下~

1.迭代器

什么是迭代器?别傻了,我最讨厌的就是名词解释了,反正就是用来遍历集合的一种方式。
比如,我们最常用的pairs,如下代码:

代码如下:

    local t = {"fdsd", "445"};

   

    for k, v in pairs(t) do

        print("k=" .. k .. ", v=" .. v);

    end

 
这是一次遍历table的操作,然后打印出table的key值和value值。
输出结果如下:

代码如下:

[LUA-print] k=1, v=fdsd

[LUA-print] k=2, v=445

2.自己写一个迭代器

要想了解迭代器,那还是要自己写一个才行,迭代器没有什么神奇的地方,它很简单。如这样一个函数:

代码如下:

function dieDaiQi(t)

    local i = 0;

    return function()

        i = i + 1;            

        return t[i];

    end

end

这函数故名思议,叫做迭代器,英文不好的也没关系,知道是这意思就好了,呵呵(小若:英文你个头啊!分明是拼音啊!)
 
有没有发现这dieDaiQi函数有点特别?没错,它就是之前的文章提到过的“闭合函数”,正是利用了闭合函数的特性来实现迭代功能的。
来看看如何使用这个迭代器吧:

代码如下:

    local iter = dieDaiQi(t);

    while true do

        local value = iter();

        if value == nil then

            break;

        end

       

        print(value);

    end

因为每一次调用dieDaiQi函数,就会产生一个新的闭合函数,所以我们要用一个iter变量保存这个闭合函数,避免重复创建。

如果你对闭合函数已经很生疏了,可以看看我之前的这篇文章:【笨木头Lua专栏】基础补充03:闭合函数、非全局函数与函数的尾调用

接着,只要循环调用iter闭合函数即可,因为闭合函数的特点,i变量是会一直增加的,所以每次调用iter函数,返回的都是下一个table元素。
最终输出结果如下:

代码如下:

[LUA-print] fdsd

[LUA-print] 445

3.更简洁的迭代器调用

刚刚调用迭代器的方式也太粗暴了,这么长一片代码,不太合理。
所以,我们又有了偷懒的方式――使用for循环调用迭代器。
 
直接看代码,刚刚的迭代器可以这么调用:

代码如下:

    local t = {"fdsd", "445"};

    for value in dieDaiQi(t) do

        print(value);

    end

 
这里大家可能会有一个疑问,每一次的循环,都会调用一次dieDaiQi函数,那不就会产生很多个闭合函数?那i的值不就每次都是0?

答案是:不会的。

因为for循环只会调用一次dieDaiQi函数,然后把它的返回值保存起来。

4.结束

这篇的介绍似乎是异常地简短,其实不是的,还有下篇,因为我怕接下来要说的东西比较多,导致文章太长。

所以,还是分开来介绍吧,下一篇,我们来深入了解一下为什么for循环可以这么方便地处理迭代器。

相关推荐