流量拦截控制

本文分享一下如何实现流量拦截

首先，流量拦截是要在某一段时间内控制访问次数，如果访问次数超过阈值，则拒绝访问。

所以，要有两个配置化信息，一个是流量监控缓存失效时间内可以调用的次数，一个是流量监控缓存失效时间

实现方式是使用Redis缓存记录调用次数和拦截器用来拦截http请求调用。

@Service
public class RateLimitInterceptor extends HandlerInterceptorAdapter {//继承拦截器的适配器类HandlerInterceptorAdapter
	private static final Logger LOGGER = LoggerFactory.getLogger(RateLimitInterceptor.class);
	private String rateControlFlag = "0";//是否开启流量控制
	private Long lastLoadCacheTime = System.currentTimeMillis();
	
	@Autowired
	private RateLimitService rateLimitService;

	/**
	 * 流量控制检查入口
	 */
	@Override
	public boolean preHandle(HttpServletRequest request,
							 HttpServletResponse response, Object handler) throws Exception {
		//根据服务器状态或请求用户的限制策略来进行控制

		// this is where magic happens
		super.preHandle(request, response, handler);
		
		//暂定userKey用请求方地址，可能对方是局域网。。
		String userKey = request.getRemoteAddr();
		
		Long thisTime = System.currentTimeMillis();
		//配置的刷新时间
		long configTime = 300000;
		if(thisTime - lastLoadCacheTime >configTime){
			lastLoadCacheTime = thisTime;//刷新上次加载缓存的时间
			//是否开启流量控制
			rateControlFlag = SystemConfigUtils.getValueByCodeAndBranchChannel(SystemConfigEnum.RATE_CONTROL_SWITCH,"","");
		}
		
		if("1".equals(rateControlFlag)){//开启流量控制
			//获取请求方在一定时间内 剩余可以调用的次数
			Long remainingCount = rateLimitService.decrease(userKey);
			
			if(LOGGER.isDebugEnabled()){
			  LOGGER.debug(userKey+" remainingCount："+ remainingCount);
			}
				
			// 小于0时表示该userKey的已经超过当前时间段能访问的次数
			if (remainingCount < 0) {//429表示你需要限制客户端请求某个服务数量时，该状态码就很有用，也就是请求速度限制。
			    response.sendError(429, "Rate limit exceeded, wait for the next quarter minute");
			    return false;
			}
			response.addHeader("Remaining request count", remainingCount == null? null :remainingCount.toString());
		}
		
		return true;
	}

}

下面看一下获取剩余可以调用的次数

	/*
	 * 减少当前userKey的剩余访问次数，并返回增加后的次数
	 */
	@Override
	public long decrease(String userKey) {

		// 查询缓存，如果没有则初始化，并设置缓存失效时间
		Long value = MyRedisCacheUtil.getCache(CacheEnum.RATE_LIMIT.getCode(), userKey);
		if (value == null) {
			this.doRefreshConfig();//刷新缓存配置

			value = RATE_LIMIT;//初始化剩余可以调用的次数为配置化的阈值
			MyRedisCacheUtil.putCache(CacheEnum.RATE_LIMIT.getCode(), userKey, value);
			// 设置key的失效时间
			MyRedisCacheUtil.expire(CacheEnum.RATE_LIMIT.getCode(), userKey, this.RATE_EXPIRES, TimeUnit.SECONDS);
		} else {
			value -= 1;//剩余次数减一
			MyRedisCacheUtil.putCache(CacheEnum.RATE_LIMIT.getCode(), userKey, value);//更新缓存
		}
		return value;
	}

	/**
	 * 刷新流量限额的配置信息
	 */
	public void doRefreshConfig() {

		String rateExpires = SystemConfigUtils.getValueByCodeAndBranchChannel(SystemConfigEnum.RATE_EXPIRES,"","");
		String rateLimitStr = SystemConfigUtils.getValueByCodeAndBranchChannel(SystemConfigEnum.RATE_LIMIT,"","");
		if (StringUtils.hasText(rateLimitStr)) {
			RATE_LIMIT = Long.parseLong(rateLimitStr);
		}
		if (StringUtils.hasText(rateExpires)) {
			RATE_EXPIRES = Long.parseLong(rateExpires);
		}
	}