Mercurial > prosody-modules
diff mod_http_upload/mod_http_upload.lua @ 3366:6d1c5ecf72c1
mod_http_upload: Report statistics on upload sizes (0.10+)
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Mon, 29 Oct 2018 01:21:29 +0100 |
parents | 1e1dbd7e5b6c |
children | fca95f1b8870 |
line wrap: on
line diff
--- a/mod_http_upload/mod_http_upload.lua Mon Oct 29 01:10:10 2018 +0100 +++ b/mod_http_upload/mod_http_upload.lua Mon Oct 29 01:21:29 2018 +0100 @@ -109,6 +109,13 @@ return sum < quota; end +local measure_slot = function () end +if module.measure then + -- COMPAT 0.9 + -- module:measure was added in 0.10 + measure_slot = module:measure("sizes", "slot"); +end + local function handle_request(origin, stanza, xmlns, filename, filesize) local username, host = origin.username, origin.host; -- local clients only @@ -157,6 +164,8 @@ pending_slots[slot] = nil; end); + measure_slot(filesize); + origin.log("debug", "Given upload slot %q", slot); local base_url = module:http_url(); @@ -213,6 +222,13 @@ return true; end); +local measure_upload = function () end +if module.measure then + -- COMPAT 0.9 + -- module:measure was added in 0.10 + measure_upload = module:measure("sizes", "upload"); +end + -- http service local function upload_data(event, path) local uploader = pending_slots[path]; @@ -252,6 +268,7 @@ os.remove(full_filename); return 500; end + measure_upload(#event.request.body); module:log("info", "File uploaded by %s to slot %s", uploader, random_dir); return 201; end