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_file_share: Update cached value while it is reasonably fresh

Browse source 
This should ensure that cache entries until the oldest file that counted
to the last 24h becomes older than 24h.
This commit is contained in:
Kim Alvefur 2021-01-31 17:44:19 +01:00
parent 1af0932ecf
commit a609c0902b
1 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
plugins/mod_http_file_share.lua
View file

@ -79,13 +79,15 @@ function get_daily_quota(uploader)
local iter, err = uploads:find(nil, {with = uploader; start = max_age }); local iter, err = uploads:find(nil, {with = uploader; start = max_age });
if not iter then return iter, err; end if not iter then return iter, err; end
local total_bytes = 0; local total_bytes = 0;
local oldest_upload; local oldest_upload = now;
for _, slot, when in iter do for _, slot, when in iter do
local size = tonumber(slot.attr.size); local size = tonumber(slot.attr.size);
if size then total_bytes = total_bytes + size; end if size then total_bytes = total_bytes + size; end
if not oldest_upload then oldest_upload = when; end if when < oldest_upload then oldest_upload = when; end
end end
quota_cache:set(uploader, { time = oldest_upload or now, size = total_bytes }); -- If there were no uploads then we end up caching [now, 0], which is fine
-- since we increase the size on new uploads
quota_cache:set(uploader, { time = oldest_upload, size = total_bytes });
return total_bytes; return total_bytes;
end end
@ -167,8 +169,11 @@ function handle_slot_request(event)
return true; return true;
end end
-- Invalidate cache local cached_quota = quota_cache:get(uploader);
quota_cache:set(uploader, nil); if cached_quota and cached_quota.time > os.time()-86400 then
cached_quota.size = cached_quota.size + filesize;
quota_cache:set(uploader, cached_quota);
end
local authz = get_authz(uploader, filename, filesize, filetype, slot); local authz = get_authz(uploader, filename, filesize, filetype, slot);
local slot_url = get_url(slot, filename); local slot_url = get_url(slot, filename);