Revert "Split ratelimiters in two (one for events, one for registration)"
This reverts commit d7dbad3526136cfc9fdbd568635be5016fb637db.
5 files changed, 7 insertions, 11 deletions
diff --git a/synapse/handlers/_base.py b/synapse/handlers/_base.py
index a2212e2023..d8d86d6ff3 100644
--- a/synapse/handlers/_base.py
+++ b/synapse/handlers/_base.py
@@ -44,7 +44,7 @@ class BaseHandler(object):
self.notifier = hs.get_notifier()
self.state_handler = hs.get_state_handler()
self.distributor = hs.get_distributor()
- self.ratelimiter = hs.get_events_ratelimiter()
+ self.ratelimiter = hs.get_ratelimiter()
self.clock = hs.get_clock()
self.hs = hs
diff --git a/synapse/handlers/message.py b/synapse/handlers/message.py
index 120aa0d017..c762b58902 100644
--- a/synapse/handlers/message.py
+++ b/synapse/handlers/message.py
@@ -224,7 +224,7 @@ class EventCreationHandler(object):
self.profile_handler = hs.get_profile_handler()
self.event_builder_factory = hs.get_event_builder_factory()
self.server_name = hs.hostname
- self.ratelimiter = hs.get_events_ratelimiter()
+ self.ratelimiter = hs.get_ratelimiter()
self.notifier = hs.get_notifier()
self.config = hs.config
diff --git a/synapse/handlers/register.py b/synapse/handlers/register.py
index 03130edc54..47d5e276f8 100644
--- a/synapse/handlers/register.py
+++ b/synapse/handlers/register.py
@@ -61,7 +61,7 @@ class RegistrationHandler(BaseHandler):
self.user_directory_handler = hs.get_user_directory_handler()
self.captcha_client = CaptchaServerHttpClient(hs)
self.identity_handler = self.hs.get_handlers().identity_handler
- self.ratelimiter = hs.get_registration_ratelimiter()
+ self.ratelimiter = hs.get_ratelimiter()
self._next_generated_user_id = None
diff --git a/synapse/rest/client/v2_alpha/register.py b/synapse/rest/client/v2_alpha/register.py
index 6f34029431..b7f354570c 100644
--- a/synapse/rest/client/v2_alpha/register.py
+++ b/synapse/rest/client/v2_alpha/register.py
@@ -196,7 +196,7 @@ class RegisterRestServlet(RestServlet):
self.identity_handler = hs.get_handlers().identity_handler
self.room_member_handler = hs.get_room_member_handler()
self.macaroon_gen = hs.get_macaroon_generator()
- self.ratelimiter = hs.get_registration_ratelimiter()
+ self.ratelimiter = hs.get_ratelimiter()
self.clock = hs.get_clock()
@interactive_auth_handler
diff --git a/synapse/server.py b/synapse/server.py
index f3ca3e259a..4323e7ff12 100644
--- a/synapse/server.py
+++ b/synapse/server.py
@@ -205,8 +205,7 @@ class HomeServer(object):
self.clock = Clock(reactor)
self.distributor = Distributor()
- self.events_ratelimiter = Ratelimiter()
- self.registration_ratelimiter = Ratelimiter()
+ self.ratelimiter = Ratelimiter()
self.datastore = None
@@ -249,11 +248,8 @@ class HomeServer(object):
def get_distributor(self):
return self.distributor
- def get_events_ratelimiter(self):
- return self.events_ratelimiter
-
- def get_registration_ratelimiter(self):
- return self.registration_ratelimiter
+ def get_ratelimiter(self):
+ return self.ratelimiter
def build_federation_client(self):
return FederationClient(self)
|