mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-04 13:47:41 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

mod_http_files: Cache data read from disk in a weak table

Browse source
This commit is contained in:
Kim Alvefur 2012-12-11 23:02:06 +01:00
parent bcfa20dc80
commit f2336aa626
1 changed files with 18 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

31
plugins/mod_http_files.lua
View file

@ -27,6 +27,8 @@ local mime_map = {
css = "text/css";
};
local cache = setmetatable({}, { __mode = "kv" }); -- Let the garbage collector have it if it wants to.
function serve_file(event, path)
local request, response = event.request, event.response;
local orig_path = request.path;
@ -44,7 +46,11 @@ function serve_file(event, path)
return 304;
end
if attr.mode == "directory" then
local data = cache[path];
if data then
response.headers.content_type = data.content_type;
data = data.data;
elseif attr.mode == "directory" then
if full_path:sub(-1) ~= "/" then
response.headers.location = orig_path.."/";
return 301;
@ -57,20 +63,19 @@ function serve_file(event, path)
-- TODO File listing
return 403;
else
local f = open(full_path, "rb");
data = f and f:read("*a");
f:close();
if not data then
return 403;
end
local ext = path:match("%.([^.]*)$");
local content_type = mime_map[ext];
cache[path] = { data = data; content_type = content_type; };
response.headers.content_type = content_type;
end
local f, err = open(full_path, "rb");
if not f then
module:log("warn", "Failed to open file: %s", err);
return 404;
end
local data = f:read("*a");
f:close();
if not data then
return 403;
end
local ext = path:match("%.([^.]*)$");
response.headers.content_type = mime_map[ext]; -- Content-Type should be nil when not known
return response:send(data);
end