FlowControl(流量控制)

public class CustomFlowControlConfigurer implements FlowControlConfigurer {
	
	/***
	 * 配置redis客户端
	 */
	@Override
	public RedissonClient redissonClient() {
		Config config = new Config();
		config.useSingleServer().setAddress("redis://localhost:6379");
		RedissonClient redisson = Redisson.create(config);
		return redisson;
	}
	
	/***
	 * 配置拦截成功后-执行的动作
	 */
	@Override
	public FlowControlInterceptAction flowControlInterceptAction() {
		return new DefaultFlowControlInterceptAction();
	}
	
	/***
	 * 配置拦截数据来源
	 */
	@Override
	public FlowControlDao flowControlDao() {
		return new InMemoryFlowControlDao();
	}
	
	/***
	 * 配置redis key的命名策略
	 */
	@Override
	public RedisKeyNameStrategy redisKeyNameStrategy() {
		return new DefaultRedisKeyNameStrategy();
	}
	
}

@Configuration
public class SpringMvcConfig extends WebMvcConfigurerAdapter {
	
	@Autowired
	private CustomFlowControlConfigurer flowControlConfigurer;
	
	/***
	 * 使用servlet的filter
	 * 
	 * @return
	 */
	@Bean
	public FilterRegistrationBean testFilterRegistration() {
		
		FilterRegistrationBean registration = new FilterRegistrationBean();
		registration.setFilter(new FlowControlFilter(flowControlConfigurer));
		registration.addUrlPatterns("/*");
		registration.setName("flowControlFilter");
		registration.setOrder(1);
		return registration;
	}
	
	
	
}

@Configuration
public class SpringMvcConfig extends WebMvcConfigurerAdapter {
	
	@Autowired
	private CustomFlowControlConfigurer flowControlConfigurer;
	
	/***
	 * 使用springmvc的interceptor
	 */
	@Override
	public void addInterceptors(InterceptorRegistry registry) {
		FlowControlHandlerInterceptor flowControlHandlerInterceptor = new FlowControlHandlerInterceptor(
				flowControlConfigurer);
		registry.addInterceptor(flowControlHandlerInterceptor).addPathPatterns("/**");
		
		super.addInterceptors(registry);
	}
	
}