aboutsummaryrefslogtreecommitdiffstats
path: root/util/iterators.lua
diff options
context:
space:
mode:
authorKim Alvefur <zash@zash.se>2016-04-15 13:39:56 +0200
committerKim Alvefur <zash@zash.se>2016-04-15 13:39:56 +0200
commitc5c18714c40d8b13616bbbb85c37403f6d2037f1 (patch)
treeb69d14abe9f602425aa02b59e944a346aea31eee /util/iterators.lua
parent7ea91caa95e5f1e65a03032559241ec4832203df (diff)
parent3f87977ddb135a709b9b86eb491aacfdef2cbdf0 (diff)
downloadprosody-c5c18714c40d8b13616bbbb85c37403f6d2037f1.tar.gz
prosody-c5c18714c40d8b13616bbbb85c37403f6d2037f1.zip
Merge 0.10->trunk
Diffstat (limited to 'util/iterators.lua')
-rw-r--r--util/iterators.lua33
1 files changed, 21 insertions, 12 deletions
diff --git a/util/iterators.lua b/util/iterators.lua
index e553229e..bd150ff2 100644
--- a/util/iterators.lua
+++ b/util/iterators.lua
@@ -23,18 +23,18 @@ function it.reverse(f, s, var)
while true do
local ret = { f(s, var) };
var = ret[1];
- if var == nil then break; end
+ if var == nil then break; end
t_insert(results, 1, ret);
end
-- Then return our reverse one
local i,max = 0, #results;
return function (_results)
- if i<max then
- i = i + 1;
- return unpack(_results[i]);
- end
- end, results;
+ if i<max then
+ i = i + 1;
+ return unpack(_results[i]);
+ end
+ end, results;
end
-- Iterate only over keys in a table
@@ -54,6 +54,15 @@ function it.values(t)
end, t;
end
+-- Iterate over the n:th return value
+function it.select(n, f, s, var)
+ return function (_s)
+ local ret = pack(f(_s, var));
+ var = ret[1];
+ return ret[n];
+ end, s, var;
+end
+
-- Given an iterator, iterate only over unique items
function it.unique(f, s, var)
local set = {};
@@ -62,8 +71,8 @@ function it.unique(f, s, var)
while true do
local ret = pack(f(s, var));
var = ret[1];
- if var == nil then break; end
- if not set[var] then
+ if var == nil then break; end
+ if not set[var] then
set[var] = true;
return unpack(ret, 1, ret.n);
end
@@ -77,7 +86,7 @@ function it.count(f, s, var)
while true do
var = f(s, var);
- if var == nil then break; end
+ if var == nil then break; end
x = x + 1;
end
@@ -110,7 +119,7 @@ function it.tail(n, f, s, var)
while true do
local ret = pack(f(s, var));
var = ret[1];
- if var == nil then break; end
+ if var == nil then break; end
results[(count%n)+1] = ret;
count = count + 1;
end
@@ -157,7 +166,7 @@ function it.to_array(f, s, var)
local t = {};
while true do
var = f(s, var);
- if var == nil then break; end
+ if var == nil then break; end
t_insert(t, var);
end
return t;
@@ -169,7 +178,7 @@ function it.to_table(f, s, var)
local t, var2 = {};
while true do
var, var2 = f(s, var);
- if var == nil then break; end
+ if var == nil then break; end
t[var] = var2;
end
return t;