| | |
|---|
| | | import com.ruoyi.common.annotation.RateLimiter; |
|---|
| | | import com.ruoyi.common.enums.LimitType; |
|---|
| | | import com.ruoyi.common.exception.ServiceException; |
|---|
| | | import com.ruoyi.common.utils.RedisUtils; |
|---|
| | | import com.ruoyi.common.utils.MessageUtils; |
|---|
| | | import com.ruoyi.common.utils.ServletUtils; |
|---|
| | | import com.ruoyi.common.utils.redis.RedisUtils; |
|---|
| | | import lombok.extern.slf4j.Slf4j; |
|---|
| | | import org.aspectj.lang.JoinPoint; |
|---|
| | | import org.aspectj.lang.annotation.Aspect; |
|---|
| | |
|---|
| | | } |
|---|
| | | long number = RedisUtils.rateLimiter(combineKey, rateType, count, time); |
|---|
| | | if (number == -1) { |
|---|
| | | throw new ServiceException("访问过于频繁，请稍候再试"); |
|---|
| | | throw new ServiceException(MessageUtils.message("rate.limiter.message")); |
|---|
| | | } |
|---|
| | | log.info("限制令牌 => {}, 剩余令牌 => {}, 缓存key => '{}'", count, number, combineKey); |
|---|
| | | } catch (ServiceException e) { |
|---|
| | |
|---|
| | | if (rateLimiter.limitType() == LimitType.IP) { |
|---|
| | | // 获取请求ip |
|---|
| | | stringBuffer.append(ServletUtils.getClientIP()).append("-"); |
|---|
| | | } else if (rateLimiter.limitType() == LimitType.CLUSTER){ |
|---|
| | | } else if (rateLimiter.limitType() == LimitType.CLUSTER) { |
|---|
| | | // 获取客户端实例id |
|---|
| | | stringBuffer.append(RedisUtils.getClient().getId()).append("-"); |
|---|
| | | } |
|---|
