Lua迭代器详解（附加红点功能实例）

1. 什么是迭代器

迭代器是一种能让我们遍历一个集合中的所有元素的代码结构。比如常用ipairs()和pairs()。

2. 为什么需要理解迭代器的原理

对于常见的table，无论key，value都可以通过ipairs()和pairs()，甚至访问table下标[*]，next()遍历到table中所有的元素，而对于诸如链表，树，图等结构时，只依靠Lua原生的迭代器并不能方便的“以一条链或者路径的方式进行输出”，这时，我们需要根据所需自定义迭代器。

      对于一些还是在校学生等接触Lua时间不长的人来说，可能没法立刻体会到链表，树，图在Lua中的使用场景（除了练习数据结构）。
      其实，在结合Lua进行游戏开发的商业项目中，这些数据结构会得到广泛的使用，比如：

1. LRU动态管理内存资源，通常LRU会由链表去解决地址冲突问题；
2. 游戏的红点树，红点一般都是用内向外（一般游戏主界面可以理解为最外一层）逐级传递，通过树来构建整个游戏中所有界面的红点依赖，就可以通过设置某一个节点从而使得这条路径上所有的节点红点都能被响应（设置显隐）。

3. 迭代器的实现

0. 闭包

闭包（closure）是一个函数以及其引用的非局部变量的组合。闭包允许函数携带其执行环境，使得函数可以访问并操作其定义时的变量，即使这些变量超出了其定义的作用域。

function createCounter()local count = 0local function counter()count = count + 1return countendreturn counter
endlocal myCounter = createCounter()
print(myCounter()) -- 输出 1
print(myCounter()) -- 输出 2
print(myCounter()) -- 输出 3

1. 有状态迭代器

有状态迭代器需要在外部环境中维护状态信息，通常利用Lua的闭包实现

-- statefulRange 返回一个闭包函数，闭包中维护了 current 状态，这个状态在每次迭代时更新。function statefulRange(start, stop)local current = start - 1return function()current = current + 1if current <= stop thenreturn currentendend
endfor i in statefulRange(1, 5) doprint(i) -- 1 2 3 4 5
end

2. 无状态迭代器

无状态迭代器在每次调用迭代函数时，通过传入参数来计算下一个值。这些参数包含了必要的状态信息，但这些信息是通过函数调用参数传递的，而不是通过外部变量或表格。

-- 迭代函数
-- range 函数返回三个值：迭代函数 rangeIterator、初始状态 start 和初始控制变量 start - 1。
-- for 循环每次调用 rangeIterator 时，传入当前状态（start）和控制变量（current）。
-- 在 rangeIterator 内部，控制变量 current 被递增，然后返回递增后的值，直到达到 stop。function rangeIterator(start, stop, current)if current < stop thencurrent = current + 1return current, currentend
end-- 迭代器生成函数
function range(start, stop)return rangeIterator, start, start - 1
end

4. 红点树系统基础

下面给出的是根据策划配置构建红点树的大致过程，如需使用，可结合项目，在此基础上进行添加并完善对应功能。

local RedPoint = {}local RedPointTable = {}function RedPoint.GetAll() -- 获取红点所有配置表return RedPointTable
end-- 获取一个节点的配置项
function RedPoint.GetConfig(NodeName)return RedPointTable[NodeName]
endfunction RedPoint.AddNode(NodeName,ParentNodeName,bEnd) -- 这种方式插入节点有个问题是要按照树的顺序从上到下，如果先插子节点，而父节点不存在，导致父节点的Child子节点索引表错误return function(cfg)if ParentNodeName ~= "none" thenlocal parentNode = RedPoint.GetConfig(ParentNodeName)if not parentNode.Child thenparentNode.Child = {}endparentNode.Child[NodeName] = true -- 以hash的方式存储，方便做一些子节点快速查找操作，如果不需要也可以直接按顺序插入endcfg.bEndNode = bEnd -- bEnd = false 添加非叶子节点 bEnd = true 添加非叶子节点cfg.parent = ParentNodeNameRedPointTable[NodeName] = cfgend
endfunction RedPoint.AddNodeEX(NodeName,ParentNodeName,bEnd) -- 这种方式插入可以不管父子节点的顺序，但是需要在使用前先对该配置表RedPointTable进行一次预处理，来建立每个节点的子节点表return function (cfg)cfg.bCatalog = bEndcfg.parent = ParentNodeNameRedPointTable[NodeName] = cfgend
end
------------------ 策划配置部分 Begin ------------------
--- 路径 1
RedPoint.AddNode("root","none",false){NodeValue = 1
}RedPoint.AddNode("child1","root",false){NodeValue = 2
}RedPoint.AddNode("grandchild1","child1",false){NodeValue = 5
}RedPoint.AddNode("grandchild2","grandchild1",true){NodeValue = 7
}--- 路径 2
RedPoint.AddNode("child2","root",true){NodeValue = 3
}--- 路径 3
RedPoint.AddNode("child3","root",true){NodeValue = 4
}--- 路径 4
RedPoint.AddNode("grandchild3","child1",true){NodeValue = 6
}
------------------ 策划配置部分 End ------------------------------------ 功能函数测试 -------------------
-- 实际开发中会有一个单独的红点功能来管理下面这些接口
-- 迭代器函数
local InnerNextNode = function (CurNodeName, CurNodeParentName)if not CurNodeParentName thenreturn CurNodeName,RedPoint.GetConfig(CurNodeName)elselocal Config = RedPoint.GetConfig(CurNodeParentName)local parent = Config.parentif parent ~= "none" thenreturn parent, RedPoint.GetConfig(parent)elsereturn nil, nilendend
end
-- 自定义for无状态迭代器
local NextNode = function(CurNodeName)return InnerNextNode,CurNodeName,nil
end
-- 由当前节点向父节点依次遍历，直到遍历到没有父节点为止
function RedPoint.GetEachNode(CurNodeName)for NodeName,NodeConfig in NextNode(CurNodeName) doprint("NodeName:",NodeName," NodeConfig:",NodeConfig.NodeValue)end
end
-- 获取一个节点的所有直接子节点
function RedPoint.GetNodeChild(CurNodeName)local CurNodeCfg = RedPointTable[CurNodeName]if CurNodeCfg and CurNodeCfg.Child thenlocal Child = CurNodeCfg.Childfor NodeName,bChild in pairs(Child) doif bChild thenlocal CurChildConfig = RedPoint.GetConfig(NodeName)if CurChildConfig thenprint("CurNodeName's Child NodeName is:",NodeName,CurChildConfig.NodeValue)elseerror("Cur Child Node is Invalid:",NodeName)endendendend
end
-- 遍历以CurNodeName节点为根节点时下面所有节点
function RedPoint.GetEachNodeAll(CurNodeName)local CurNodeConfig = RedPoint.GetConfig(CurNodeName)if CurNodeConfig thenprint("Cur Node====",CurNodeConfig.NodeValue)local CurChildNode = CurNodeConfig.Childif CurChildNode thenfor NodeName,NodeConfig in pairs(CurChildNode) doRedPoint.GetEachNodeAll(NodeName)endendendenddoRedPoint.GetEachNode("grandchild2")-- NodeName:    grandchild2  NodeConfig:    7-- NodeName:    grandchild1  NodeConfig:    5-- NodeName:    child1   NodeConfig:    2-- NodeName:    root     NodeConfig:    1RedPoint.GetNodeChild("root")-- CurNodeName's Child NodeName is: child2  3-- CurNodeName's Child NodeName is: child3  4-- CurNodeName's Child NodeName is: child1  2RedPoint.GetEachNodeAll("root")-- Cur Node==== 1-- Cur Node==== 3-- Cur Node==== 4-- Cur Node==== 2-- Cur Node==== 5-- Cur Node==== 7-- Cur Node==== 6
endreturn RedPoint