From 760cb387bf8e1e5175eca21a0b6b5b02a5467be5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Arturo=20Cabral=20Mej=C3=ADa?= Date: Tue, 15 Nov 2022 08:44:31 -0500 Subject: [PATCH] fix: inject sw rate limiter in event strategies --- src/factories/message-handler-factory.ts | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/src/factories/message-handler-factory.ts b/src/factories/message-handler-factory.ts index c3bcbba..9ebda77 100644 --- a/src/factories/message-handler-factory.ts +++ b/src/factories/message-handler-factory.ts @@ -7,6 +7,7 @@ import { eventStrategyFactory } from './event-strategy-factory' import { IEventRepository } from '../@types/repositories' import { isDelegatedEvent } from '../utils/event' import { IWebSocketAdapter } from '../@types/adapters' +import { slidingWindowRateLimiterFactory } from './rate-limiter-factory' import { SubscribeMessageHandler } from '../handlers/subscribe-message-handler' import { UnsubscribeMessageHandler } from '../handlers/unsubscribe-message-handler' @@ -20,11 +21,17 @@ export const messageHandlerFactory = ( return new DelegatedEventMessageHandler( adapter, delegatedEventStrategyFactory(eventRepository), - createSettings + createSettings, + slidingWindowRateLimiterFactory, ) } - return new EventMessageHandler(adapter, eventStrategyFactory(eventRepository), createSettings) + return new EventMessageHandler( + adapter, + eventStrategyFactory(eventRepository), + createSettings, + slidingWindowRateLimiterFactory, + ) } case MessageType.REQ: return new SubscribeMessageHandler(adapter, eventRepository, createSettings)