基於Redis SortedSet簡單限流算法

如下爲接口及具體實現：

抽象接口，方法有三個參數，第一個參數爲此限流要用到的Redis的key，第二個參數爲限流的時間，第三個參數爲限流次數。

---

package com.shanyuan.platform.ms.base.frelimit;

/**
 * 
 * 限流算法
 * @author 楊健
 * @date 2017/11/14
 *
 */
public interface FrequencyLimitService {
	
	public boolean freLimit(String redisKey, Integer seconds, Integer times);

}

具體實現

package com.shanyuan.platform.ms.base.frelimit.impl;

import java.util.UUID;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;
import org.springframework.stereotype.Component;

import com.shanyuan.platform.ms.base.frelimit.FrequencyLimitService;

/**
*
* 頻率限制工具
* @author yangjian
* @date 2017/11/14
*
*/
@Component
public class FrequencyLimitServiceImpl implements FrequencyLimitService {

protected Logger logger = LoggerFactory.getLogger(getClass());

@Autowired
private StringRedisTemplate stringRedisTemplate;

/**
*
* @param redisKey
* @param seconds
* @param times
* @return
*/
public boolean freLimit(String redisKey, Integer seconds, Integer times){
boolean result = false;
//定義redis操做
ZSetOperations<String, String> opsForZSet = stringRedisTemplate.opsForZSet();

Long endTime = System.currentTimeMillis()/1000;
//設置最後一次發送
Long startTime = endTime - seconds;
//移出以前已無效的記錄
//當前redis中有效總條數
Long count = opsForZSet.count(redisKey, startTime, endTime);
//若是條數大於或等於條數限制，則拋出異常，發送太屢次
if(count >= times){
logger.info("規定時間：{}秒內，請求次數：{}過多", seconds, count);
return result;
}
String value = UUID.randomUUID().toString().replaceAll("-", "");
//向set中添加新紀錄
result = opsForZSet.add(redisKey, value, endTime);
if(!result){
logger.info("add redis set error, key:{}, score:{}", opsForZSet, endTime);
return result;
}
//當前有效的總條數
count = opsForZSet.count(redisKey, startTime, endTime);
if(count >= times){
opsForZSet.removeRangeByScore(redisKey, 0, startTime);
Long rank = opsForZSet.rank(redisKey, value);
if(rank < times){
return true;
}else{
opsForZSet.remove(redisKey, value);
return false;
}
}
return result;
}

}

實現方法中的註釋基本說明了每行是作什麼工做。
本方法主要是作限流，能夠用在訂單，發送短信，或者其餘須要作流量控制以及防止攻擊的地方。