aboutsummaryrefslogtreecommitdiffstats
path: root/util
diff options
context:
space:
mode:
authorMatthew Wild <mwild1@gmail.com>2011-08-29 13:00:33 -0400
committerMatthew Wild <mwild1@gmail.com>2011-08-29 13:00:33 -0400
commitcfd622372fcdfb31f2dc400be443f7266032f444 (patch)
tree1c494effeca2cc9c488de86aef634fc0ed17d25f /util
parent8114fdc49e5437a8f3eb4a50c96976729ee7a211 (diff)
downloadprosody-cfd622372fcdfb31f2dc400be443f7266032f444.tar.gz
prosody-cfd622372fcdfb31f2dc400be443f7266032f444.zip
util.throttle: Generic module by waqas to limit something over some time
Diffstat (limited to 'util')
-rw-r--r--util/throttle.lua44
1 files changed, 44 insertions, 0 deletions
diff --git a/util/throttle.lua b/util/throttle.lua
new file mode 100644
index 00000000..8b62e797
--- /dev/null
+++ b/util/throttle.lua
@@ -0,0 +1,44 @@
+
+local gettime = require "socket".gettime;
+
+module "throttle"
+
+local throttle = {};
+local throttle_mt = { __index = throttle };
+
+function throttle:update()
+ local newt = gettime();
+ local elapsed = newt - self.t;
+ self.t = newt;
+ local balance = self.rate * elapsed + self.balance;
+ if balance > self.max then
+ self.balance = self.max;
+ else
+ self.balance = balance;
+ end
+ return self.balance;
+end
+
+function throttle:peek(cost)
+ cost = cost or 1;
+ return self.balance >= cost or self:update() >= cost;
+end
+
+function throttle:poll(cost, split)
+ if self:peek(cost) then
+ self.balance = self.balance - cost;
+ return true;
+ else
+ local balance = self.balance;
+ if split then
+ self.balance = 0;
+ end
+ return false, balance, (cost-self.balance);
+ end
+end
+
+function create(max, period)
+ return setmetatable({ rate = max / period, max = max, t = 0, balance = max }, throttle_mt);
+end
+
+return _M;