util/timer.lua


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

-- Prosody IM v0.3
-- Copyright (C) 2008-2009 Matthew Wild
-- Copyright (C) 2008-2009 Waqas Hussain
-- 
-- This project is MIT/X11 licensed. Please see the
-- COPYING file in the source package for more information.
--


local ns_addtimer = require "net.server".addtimer;
local get_time = os.time;
local t_insert = table.insert;
local t_remove = table.remove;
local ipairs, pairs = ipairs, pairs;
local type = type;

local data = {};
local new_data = {};

module "timer"

local function _add_task(delay, func)
	local current_time = get_time();
	delay = delay + current_time;
	if delay >= current_time then
		t_insert(new_data, {delay, func});
	else func(); end
end

add_task = _add_task;

ns_addtimer(function()
	local current_time = get_time();
	if #new_data > 0 then
		for _, d in ipairs(new_data) do
			t_insert(data, d);
		end
		new_data = {};
	elseif #data == 0 then
		return;
	end
	
	for i, d in pairs(data) do
		local t, func = d[1], d[2];
		if t <= current_time then
			t_remove(data, i);
			local r = func();
			if type(r) == "number" then _add_task(r, func); end
		end
	end
end);

return _M;