Merge pull request #52 from onaio/add-throttling

Add throttling configurations

Author: KipSigei

defaults/main.yml

@@ -326,3 +326,8 @@ onadata_crf_cookie_secure: True
 onadata_crf_use_sessions: True
 onadata_session_cookie_secure: True
 onadata_session_cookie_samesite: "Strict"
+
+# throttling
+onadata_enable_requests_throttling: False
+onadata_throttling_rate: 50
+onadata_throttling_user_agents: []

templates/onadata_checkout_path/onadata/preset/local_settings.py.j2

@@ -90,7 +90,27 @@ REST_FRAMEWORK = {
         'rest_framework.renderers.BrowsableAPIRenderer',
         'rest_framework_csv.renderers.CSVRenderer',
     ),
+{% if onadata_enable_requests_throttling|default(false) %}
+    'DEFAULT_THROTTLE_CLASSES': (
+        'onadata.libs.throttle.RequestHeaderThrottle',
+    ),
+    'DEFAULT_THROTTLE_RATES': {
+        'header': '{{ onadata_throttling_rate }}/minute'
+    }
+{% endif %}
 }
+{% if onadata_enable_requests_throttling|default(false) %}
+THROTTLE_HEADERS = {
+    "HTTP_USER_AGENT": [
+{% for user_agent in onadata_throttling_user_agents %}
+        '{{ user_agent }}',
+{% endfor %}
+    ],
+{% if onadata_throttling_remote_user is defined %}
+    "REMOTE_USER": '{{ onadata_throttling_remote_user }}'
+{% endif %}
+}
+{% endif %}
 OAUTH2_PROVIDER['AUTHORIZATION_CODE_EXPIRE_SECONDS'] = 600
 OAUTH2_PROVIDER['PKCE_REQUIRED'] = {{ onadata_oauth2_pkce_required }}
 BROKER_TRANSPORT = 'librabbitmq'