vimaltech-contact-api/src/main/java/com/vimaltech/contactapi/config/RateLimitConfig.java at 6597a16243b6e086dc1ff92797eb651a5eefec5e · vimal-java-dev/vimaltech-contact-api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
package com.vimaltech.contactapi.config;

import com.vimaltech.contactapi.security.IpRateLimitFilter;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;

@Configuration
@Profile("prod")
public class RateLimitConfig {

    @Bean
    public IpRateLimitFilter ipRateLimitFilter() {
        return new IpRateLimitFilter();
    }

    @Bean
    public FilterRegistrationBean<IpRateLimitFilter> rateLimitFilterRegistration() {

        FilterRegistrationBean<IpRateLimitFilter> registration =
                new FilterRegistrationBean<>();

        registration.setFilter(ipRateLimitFilter());
        registration.addUrlPatterns("/*");
        registration.setOrder(1);

        return registration;
    }
}