基于Redis的分布式限流实现

本节基于Redis实现了系统的限流控制。

Lua脚本准备

local val = redis.call('incr', KEYS[1]) local ttl = redis.call('ttl', KEYS[1]) redis.log(redis.LOG_NOTICE, "incr "..KEYS[1].." "..val); if val == 1 then redis.call('expire', KEYS[1], tonumber(ARGV[1])) else if ttl == -1 then redis.call('expire', KEYS[1], tonumber(ARGV[1])) end end if val > tonumber(ARGV[2]) then return 0 end return 1

对传入的key做incr操作，如果key首次生成，设置超时时间ARGV[1]；

ttl是为防止某些key在未设置超时时间并长时间已经存在的情况下做的保护的判断；

判断当前key的val是否超过限制次数ARGV[2]。

Redis客户端

public Long limit(String key) { return redisClient.eval(key, expire, reqCount); }

加载lua脚本，并对外提供Redis调用的接口。

AOP

@Before("@annotation(com.snatch.deal.shop.common.annotations.RateLimit)") public void before(JoinPoint pjp) throws Throwable { Signature sig = pjp.getSignature(); if (!(sig instanceof MethodSignature)) { throw new IllegalArgumentException("该注解只能用在方法上"); } MethodSignature msig = (MethodSignature) sig; String methodName = pjp.getTarget().getClass().getName() + "." + msig.getName(); String limitKey = Md5Utils.encrypt(methodName); if (rateLimiter.limit(limitKey) != 1){ throw new OverLimitException("触发限流控制"); } }

@Target({ElementType.METHOD}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface RateLimit { String value() default ""; }

利用AOP的before实现方法执行前的拦截，所有注解了RateLimit的方法都会被缓存限流。一旦触发限流，对外抛出异常。

统一异常处理

@ControllerAdvice public class GlobalExceptionHandler { @ExceptionHandler(OverLimitException.class) @ResponseBody public String OverLimitExceptionHandler(OverLimitException ole){ return ole.getMessage(); } }

利用ControllerAdvice处理对应的异常的处理。

详细代码在 https://gitee.com/lawlet/shop