lua自定义迭代器

Posted 2020-08-06 LightSong@计海拾贝

迭代器

http://www.tutorialspoint.com/lua/lua_iterators.htm

迭代器能够让你遍历某个集合或者容器中的每一个元素。 对于lua来说， 集合通常指代 table， 用于创建变化的数据结构， 类似数组。

Iterator is a construct that enables you to traverse through the elements of the so called collection or container. In Lua, these collections often refer to tables, which are used to create various data structures like array.

 

通用For迭代器

 

通常使用的for循环， 配合in使用， in后的参数 就是一个迭代器函数。

A generic for iterator provides the key value pairs of each element in the collection. A simple example is given below.

array = {"Lua", "Tutorial"}

for key,value in ipairs(array) 
do
   print(key, value)
end

对于迭代器函数， 根据迭代器函数中状态的维护， 可以分为如下两种类型， 有状态迭代器， 和 无状态迭代器。

In Lua we use functions to represent iterators. Based on the state maintenance in these iterator functions, we have two main types −

• Stateless Iterators
• Stateful Iterators

 

无状态迭代器

迭代器函数中不维护任何状态。

By the name itself we can understand that this type of iterator function does not retain any state.

Let us now see an example of creating our own iterator using a simple function that prints the squares of n numbers.

如下例子， 迭代状态， 由for的参数 i 记录。

function square(iteratorMaxCount,currentNumber)

   if currentNumber<iteratorMaxCount
   then
      currentNumber = currentNumber+1
      return currentNumber, currentNumber*currentNumber
   end
	
end

for i,n in square,3,0
do
   print(i,n)
end

 

改进版：

function square(iteratorMaxCount,currentNumber)

   if currentNumber<iteratorMaxCount
   then
      currentNumber = currentNumber+1
      return currentNumber, currentNumber*currentNumber
   end
	
end

function squares(iteratorMaxCount)
   return square,iteratorMaxCount,0
end  

for i,n in squares(3)
do 
   print(i,n)
end

 

有状态迭代器

利用闭包，将状态管理在闭包类， 迭代器函数为闭包。

The previous example of iteration using function does not retain the state. Each time the function is called, it returns the next element of the collection based on a second variable sent to the function. To hold the state of the current element, closures are used. Closure retain variables values across functions calls. To create a new closure, we create two functions including the closure itself and a factory, the function that creates the closure.

Let us now see an example of creating our own iterator in which we will be using closures.

 

如下例子，推荐使用如下写法，减少信息暴漏：

array = {"Lua", "Tutorial"}

function elementIterator (collection)

   local index = 0
   local count = #collection
	
   -- The closure function is returned
	
   return function ()
      index = index + 1
		
      if index <= count
      then
         -- return the current element of the iterator
         return collection[index]
      end
		
   end
	
end

for element in elementIterator(array)
do
   print(element)
end

 

自定义例子

使用有状态迭代器， 实现字符串拆分为固定长度的字符串：

 

local instr = "2334t545dfgjkkkk"


function StrSegIterator (str, segSize)
 local strIndex = 1

 -- The closure function is returned
 return function ()
  local segStart = strIndex
  local segEnd = strIndex + segSize - 1
  local strseg = string.sub(str, segStart, segEnd)

  if #strseg > 0 then
   strIndex = strIndex + segSize

   -- return the current element of the iterator
   return strseg
  end

 end

end


for element in StrSegIterator(instr, 2)
do
   print(element)
end