cartodb/CartoDB-SQL-API
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

params existence responsability for UserLimits

Browse Source
This commit is contained in:
Simon Martín 2018-05-10 11:51:58 +02:00
parent a997881048
commit 0bb7cec9f4
4 changed files with 21 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/middlewares/rate-limit.js
View File

@ -51,10 +51,8 @@ function rateLimit(userLimits, endpointGroup = null) {
}
function isRateLimitEnabled(endpointGroup) {
return global.settings.ratelimits &&
global.settings.ratelimits.rateLimitsEnabled &&
return global.settings.ratelimits.rateLimitsEnabled &&
endpointGroup &&
global.settings.ratelimits.endpoints &&
global.settings.ratelimits.endpoints[endpointGroup];
}

3
app/server.js
View File

@ -137,9 +137,10 @@ function App(statsClient) {
var userDatabaseService = new UserDatabaseService(metadataBackend);
UserLimitsService.configure();
const userLimitsServiceOptions = {
limits: {
rateLimitsEnabled: (global.settings.ratelimits && global.settings.ratelimits.rateLimitsEnabled) || false
rateLimitsEnabled: global.settings.ratelimits.rateLimitsEnabled
}
};
const userLimitsService = new UserLimitsService(metadataBackend, userLimitsServiceOptions);

18
app/services/user_limits.js
View File

@ -1,3 +1,14 @@
const DEFAULT_RATE_LIMITS_OPTIONS = {
rateLimitsEnabled: false,
endpoints: {
query: false,
query_format: false,
job_create: false,
job_get: false,
job_delete: false
}
};
/**
* UserLimits
* @param {cartodb-redis} metadataBackend
@ -11,6 +22,13 @@ class UserLimits {
this.preprareRateLimit();
}
static configure() {
// default rate limits
if(!global.settings.ratelimits) {
global.settings.ratelimits = DEFAULT_RATE_LIMITS_OPTIONS;
}
}
preprareRateLimit() {
if (this.options.limits.rateLimitsEnabled) {
this.metadataBackend.loadRateLimitsScript();

8
test/acceptance/rate-limit.js
View File

@ -72,10 +72,6 @@ function assertRequest (status, limit, remaining, reset, retry, done = null) {
describe('rate limit', function() {
before(function() {
if (!global.settings.ratelimits) {
global.settings.ratelimits = {};
global.settings.ratelimits.endpoints = {};
}
global.settings.ratelimits.rateLimitsEnabled = true;
global.settings.ratelimits.endpoints.query = true;
@ -89,10 +85,6 @@ describe('rate limit', function() {
});
after(function() {
if (!global.settings.ratelimits) {
global.settings.ratelimits = {};
global.settings.ratelimits.endpoints = {};
}
global.settings.ratelimits.rateLimitsEnabled = false;
global.settings.ratelimits.endpoints.query = false;