Merge pull request #1299 from slootjes/master

Prevent possible download counter race condition
This commit is contained in:
Danny Coates 2019-04-19 10:37:58 -07:00 committed by GitHub
commit 147ed4c736
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 1 deletions

View File

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

View File

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