summaryrefslogtreecommitdiffhomepage
path: root/libs/httpd/luasrc
diff options
context:
space:
mode:
authorSteven Barth <steven@midlink.org>2008-07-03 18:49:29 +0000
committerSteven Barth <steven@midlink.org>2008-07-03 18:49:29 +0000
commit280872baa79d20de093314e3cfcf2cea14434f83 (patch)
tree39e012bbd6285563abfdefbd560cd33fdcaca4ee /libs/httpd/luasrc
parentf0a0e503783d6571c1a0a90bf14e7531b2f2a176 (diff)
* libs/httpd: Add limit for luci handler
Diffstat (limited to 'libs/httpd/luasrc')
-rw-r--r--libs/httpd/luasrc/httpd/handler/luci.lua12
1 files changed, 11 insertions, 1 deletions
diff --git a/libs/httpd/luasrc/httpd/handler/luci.lua b/libs/httpd/luasrc/httpd/handler/luci.lua
index 093741393..65f3f01b2 100644
--- a/libs/httpd/luasrc/httpd/handler/luci.lua
+++ b/libs/httpd/luasrc/httpd/handler/luci.lua
@@ -23,12 +23,16 @@ local ltn12 = require("luci.ltn12")
Luci = luci.util.class(luci.httpd.module.Handler)
Response = luci.httpd.module.Response
-function Luci.__init__(self)
+function Luci.__init__(self, limit)
luci.httpd.module.Handler.__init__(self)
+ self.limit = limit or 5
+ self.running = {}
+ setmetatable(self.running, {__mode = "v"})
end
function Luci.handle_head(self, ...)
local response, sourceout = self:handle_get(...)
+ self.running = self.running - 1
return response
end
@@ -37,6 +41,11 @@ function Luci.handle_post(self, ...)
end
function Luci.handle_get(self, request, sourcein, sinkerr)
+ if self.limit and #self.running >= self.limit then
+ return self:failure(503, "Overload")
+ end
+ table.insert(self.running, coroutine.running())
+
local r = luci.http.Request(
request.env,
sourcein,
@@ -57,6 +66,7 @@ function Luci.handle_get(self, request, sourcein, sinkerr)
status = 500
headers["Content-Type"] = "text/plain"
local err = {id}
+ self.running = self.running - 1
return Response( status, headers ), function() return table.remove(err) end
end