diff --git a/src/main/java/com/vimaltech/contactapi/config/RateLimitConfig.java b/src/main/java/com/vimaltech/contactapi/config/RateLimitConfig.java index cc58a12..ae74205 100644 --- a/src/main/java/com/vimaltech/contactapi/config/RateLimitConfig.java +++ b/src/main/java/com/vimaltech/contactapi/config/RateLimitConfig.java @@ -11,11 +11,17 @@ public class RateLimitConfig { @Bean - public FilterRegistrationBean rateLimitFilterRegistration( - IpRateLimitFilter filter) { + public IpRateLimitFilter ipRateLimitFilter() { + return new IpRateLimitFilter(); + } + + @Bean + public FilterRegistrationBean rateLimitFilterRegistration() { + + FilterRegistrationBean registration = + new FilterRegistrationBean<>(); - FilterRegistrationBean registration = new FilterRegistrationBean<>(); - registration.setFilter(filter); + registration.setFilter(ipRateLimitFilter()); registration.addUrlPatterns("/*"); registration.setOrder(1); diff --git a/src/main/java/com/vimaltech/contactapi/security/IpRateLimitFilter.java b/src/main/java/com/vimaltech/contactapi/security/IpRateLimitFilter.java index b4cb178..0a0f7c8 100644 --- a/src/main/java/com/vimaltech/contactapi/security/IpRateLimitFilter.java +++ b/src/main/java/com/vimaltech/contactapi/security/IpRateLimitFilter.java @@ -18,8 +18,6 @@ import java.util.Optional; import java.util.concurrent.ConcurrentHashMap; -@Component -@Profile("prod") public class IpRateLimitFilter extends OncePerRequestFilter { private final Map buckets = new ConcurrentHashMap<>();