mentos1386/send
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Prevent possible download counter race condition

Browse source
This commit is contained in:
Robert 2019-04-19 13:10:49 +02:00
parent ee45a835be
commit f94918bebd
2 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
server/routes/download.js
View file

@ -34,7 +34,7 @@ module.exports = async function(req, res) {
if (dl >= dlimit) { if (dl >= dlimit) {
await storage.del(id); await storage.del(id);
} else { } else {
await storage.setField(id, 'dl', dl); await storage.incrementField(id, 'dl');
} }
} catch (e) { } catch (e) {
log.info('StorageError:', id); log.info('StorageError:', id);

4
server/storage/index.js
View file

@ -62,6 +62,10 @@ class DB {
this.redis.hset(id, key, value); this.redis.hset(id, key, value);
} }
incrementField(id, key, increment = 1) {
this.redis.hincrby(id, key, increment);
}
async del(id) { async del(id) {
const filePath = await this.getPrefixedId(id); const filePath = await this.getPrefixedId(id);
this.storage.del(filePath); this.storage.del(filePath);