util/timer.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2010 Matthew Wild
   3 -- Copyright (C) 2008-2010 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9
  10 local ns_addtimer = require "net.server".addtimer;
  11 local event = require "net.server".event;
  12 local event_base = require "net.server".event_base;
  13
  14 local math_min = math.min
  15 local math_huge = math.huge
  16 local get_time = require "socket".gettime;
  17 local t_insert = table.insert;
  18 local t_remove = table.remove;
  19 local ipairs, pairs = ipairs, pairs;
  20 local type = type;
  21
  22 local data = {};
  23 local new_data = {};
  24
  25 module "timer"
  26
  27 local _add_task;
  28 if not event then
  29         function _add_task(delay, callback)
  30                 local current_time = get_time();
  31                 delay = delay + current_time;
  32                 if delay >= current_time then
  33                         t_insert(new_data, {delay, callback});
  34                 else
  35                         local r = callback();
  36                         if r and type(r) == "number" then
  37                                 return _add_task(r, callback);
  38                         end
  39                 end
  40         end
  41
  42         ns_addtimer(function()
  43                 local current_time = get_time();
  44                 if #new_data > 0 then
  45                         for _, d in pairs(new_data) do
  46                                 t_insert(data, d);
  47                         end
  48                         new_data = {};
  49                 end
  50
  51                 local next_time = math_huge;
  52                 for i, d in pairs(data) do
  53                         local t, callback = d[1], d[2];
  54                         if t <= current_time then
  55                                 data[i] = nil;
  56                                 local r = callback(current_time);
  57                                 if type(r) == "number" then
  58                                         _add_task(r, callback);
  59                                         next_time = math_min(next_time, r);
  60                                 end
  61                         else
  62                                 next_time = math_min(next_time, t - current_time);
  63                         end
  64                 end
  65                 return next_time;
  66         end);
  67 else
  68         local EVENT_LEAVE = (event.core and event.core.LEAVE) or -1;
  69         function _add_task(delay, callback)
  70                 local event_handle;
  71                 event_handle = event_base:addevent(nil, 0, function ()
  72                         local ret = callback();
  73                         if ret then
  74                                 return 0, ret;
  75                         elseif event_handle then
  76                                 return EVENT_LEAVE;
  77                         end
  78                 end
  79                 , delay);
  80         end
  81 end
  82
  83 add_task = _add_task;
  84
  85 return _M;