mirror of https://gitgud.io/fatchan/jschan.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
612 B
37 lines
612 B
'use strict';
|
|
|
|
const Mongo = require(__dirname+'/db.js')
|
|
, db = Mongo.client.db('jschan').collection('ratelimit');
|
|
|
|
module.exports = {
|
|
|
|
db,
|
|
|
|
resetQuota: (ip, suffix) => {
|
|
return db.deleteOne({ '_id': `${ip}-${suffix}` });
|
|
},
|
|
|
|
incrmentQuota: (ip, suffix, amount) => {
|
|
return db.findOneAndUpdate(
|
|
{
|
|
'_id': `${ip}-suffix`
|
|
},
|
|
{
|
|
'$inc': {
|
|
'sequence_value': amount
|
|
},
|
|
'$setOnInsert': {
|
|
'expireAt': new Date()
|
|
}
|
|
},
|
|
{
|
|
'upsert': true
|
|
}
|
|
).then(r => { return r.value ? r.value.sequence_value : 0 });
|
|
},
|
|
|
|
deleteAll: () => {
|
|
return db.deleteMany({});
|
|
},
|
|
|
|
}
|
|
|