当前位置:首页 >> 脚本专栏

Lua中的迭代器浅析

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

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循环可以这么方便地处理迭代器。