日韩无码专区无码一级三级片|91人人爱网站中日韩无码电影|厨房大战丰满熟妇|AV高清无码在线免费观看|另类AV日韩少妇熟女|中文日本大黄一级黄色片|色情在线视频免费|亚洲成人特黄a片|黄片wwwav色图欧美|欧亚乱色一区二区三区

RELATEED CONSULTING
相關(guān)咨詢
選擇下列產(chǎn)品馬上在線溝通
服務(wù)時(shí)間：8:30-17:00
你可能遇到了下面的問(wèn)題
關(guān)閉右側(cè)工具欄

新聞中心

這里有您想知道的互聯(lián)網(wǎng)營(yíng)銷解決方案
Redis做接口限流，一個(gè)注解的事！

今天想和小伙伴們聊聊用 Redis 處理接口限流，這也是最近的 TienChin 項(xiàng)目涉及到這個(gè)知識(shí)點(diǎn)了，我就拎出來(lái)和大家聊聊這個(gè)話題。

10年積累的成都做網(wǎng)站、成都網(wǎng)站建設(shè)經(jīng)驗(yàn)，可以快速應(yīng)對(duì)客戶對(duì)網(wǎng)站的新想法和需求。提供各種問(wèn)題對(duì)應(yīng)的解決方案。讓選擇我們的客戶得到更好、更有力的網(wǎng)絡(luò)服務(wù)。我雖然不認(rèn)識(shí)你，你也不認(rèn)識(shí)我。但先制作網(wǎng)站后付款的網(wǎng)站建設(shè)流程，更有信陽(yáng)免費(fèi)網(wǎng)站建設(shè)讓你可以放心的選擇與我們合作。

1. 準(zhǔn)備工作

首先我們創(chuàng)建一個(gè) Spring Boot 工程，引入 Web 和 Redis 依賴，同時(shí)考慮到接口限流一般是通過(guò)注解來(lái)標(biāo)記，而注解是通過(guò) AOP 來(lái)解析的，所以我們還需要加上 AOP 的依賴，最終的依賴如下：


    org.springframework.boot
    spring-boot-starter-data-redis


    org.springframework.boot
    spring-boot-starter-web


    org.springframework.boot
    spring-boot-starter-aop

然后提前準(zhǔn)備好一個(gè) Redis 實(shí)例，這里我們項(xiàng)目配置好之后，直接配置一下 Redis 的基本信息即可，如下：

spring.redis.host=localhost
spring.redis.port=6379
spring.redis.password=123

好啦，準(zhǔn)備工作就算是到位了。

2. 限流注解

接下來(lái)我們創(chuàng)建一個(gè)限流注解，我們將限流分為兩種情況：

  • 針對(duì)當(dāng)前接口的全局性限流，例如該接口可以在 1 分鐘內(nèi)訪問(wèn) 100 次。
  • 針對(duì)某一個(gè) IP 地址的限流，例如某個(gè) IP 地址可以在 1 分鐘內(nèi)訪問(wèn) 100 次。

針對(duì)這兩種情況，我們創(chuàng)建一個(gè)枚舉類：

public enum LimitType {
    /**
     * 默認(rèn)策略全局限流
     */
    DEFAULT,
    /**
     * 根據(jù)請(qǐng)求者IP進(jìn)行限流
     */
    IP
}

接下來(lái)我們來(lái)創(chuàng)建限流注解：

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface RateLimiter {
    /**
     * 限流key
     */
    String key() default "rate_limit:";

    /**
     * 限流時(shí)間,單位秒
     */
    int time() default 60;

    /**
     * 限流次數(shù)
     */
    int count() default 100;

    /**
     * 限流類型
     */
    LimitType limitType() default LimitType.DEFAULT;
}

第一個(gè)參數(shù)限流的 key，這個(gè)僅僅是一個(gè)前綴，將來(lái)完整的 key 是這個(gè)前綴再加上接口方法的完整路徑，共同組成限流 key，這個(gè) key 將被存入到 Redis 中。

另外三個(gè)參數(shù)好理解，我就不多說(shuō)了。

好了，將來(lái)哪個(gè)接口需要限流，就在哪個(gè)接口上添加 注解，然后配置相關(guān)參數(shù)即可。@RateLimiter

3. 定制 RedisTemplate

小伙伴們知道，在 Spring Boot 中，我們其實(shí)更習(xí)慣使用 Spring Data Redis 來(lái)操作 Redis，不過(guò)默認(rèn)的 RedisTemplate 有一個(gè)小坑，就是序列化用的是 JdkSerializationRedisSerializer，不知道小伙伴們有沒(méi)有注意過(guò)，直接用這個(gè)序列化工具將來(lái)存到 Redis 上的 key 和 value 都會(huì)莫名其妙多一些前綴，這就導(dǎo)致你用命令讀取的時(shí)候可能會(huì)出錯(cuò)。

例如存儲(chǔ)的時(shí)候，key 是 name，value 是 javaboy，但是當(dāng)你在命令行操作的時(shí)候， 卻獲取不到你想要的數(shù)據(jù)，原因就是存到 redis 之后 name 前面多了一些字符，此時(shí)只能繼續(xù)使用 RedisTemplate 將之讀取出來(lái)。get name

我們用 Redis 做限流會(huì)用到 Lua 腳本，使用 Lua 腳本的時(shí)候，就會(huì)出現(xiàn)上面說(shuō)的這種情況，所以我們需要修改 RedisTemplate 的序列化方案。

可能有小伙伴會(huì)說(shuō)為什么不用 StringRedisTemplate 呢?StringRedisTemplate 確實(shí)不存在上面所說(shuō)的問(wèn)題，但是它能夠存儲(chǔ)的數(shù)據(jù)類型不夠豐富，所以這里不考慮。

修改 RedisTemplate 序列化方案，代碼如下：

@Configuration
public class RedisConfig {

    @Bean
    public RedisTemplate redisTemplate(RedisConnectionFactory connectionFactory) {
        RedisTemplate redisTemplate = new RedisTemplate<>();
        redisTemplate.setConnectionFactory(connectionFactory);
        // 使用Jackson2JsonRedisSerialize 替換默認(rèn)序列化(默認(rèn)采用的是JDK序列化)
        Jackson2JsonRedisSerializer jackson2JsonRedisSerializer = new Jackson2JsonRedisSerializer<>(Object.class);
        ObjectMapper om = new ObjectMapper();
        om.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY);
        om.enableDefaultTyping(ObjectMapper.DefaultTyping.NON_FINAL);
        jackson2JsonRedisSerializer.setObjectMapper(om);
        redisTemplate.setKeySerializer(jackson2JsonRedisSerializer);
        redisTemplate.setValueSerializer(jackson2JsonRedisSerializer);
        redisTemplate.setHashKeySerializer(jackson2JsonRedisSerializer);
        redisTemplate.setHashValueSerializer(jackson2JsonRedisSerializer);
        return redisTemplate;
    }
}
這個(gè)其實(shí)也沒(méi)啥好說(shuō)的，key 和 value 我們都使用 Spring Boot 中默認(rèn)的 jackson 序列化方式來(lái)解決。


4. 開發(fā) Lua 腳本


這個(gè)其實(shí)我在之前 vhr 那一套視頻中講過(guò)，Redis 中的一些原子操作我們可以借助 Lua 腳本來(lái)實(shí)現(xiàn)，想要調(diào)用 Lua 腳本，我們有兩種不同的思路：


    

  • 在 Redis 服務(wù)端定義好 Lua 腳本，然后計(jì)算出來(lái)一個(gè)散列值，在 Java 代碼中，通過(guò)這個(gè)散列值鎖定要執(zhí)行哪個(gè) Lua 腳本。
    • 

  • 直接在 Java 代碼中將 Lua 腳本定義好，然后發(fā)送到 Redis 服務(wù)端去執(zhí)行。
    • 



Spring Data Redis 中也提供了操作 Lua 腳本的接口，還是比較方便的，所以我們這里就采用第二種方案。
我們?cè)?resources 目錄下新建 lua 文件夾專門用來(lái)存放 lua 腳本，腳本內(nèi)容如下：
local key = KEYS[1]
local count = tonumber(ARGV[1])
local time = tonumber(ARGV[2])
local current = redis.call('get', key)
if current and tonumber(current) > count then
    return tonumber(current)
end
current = redis.call('incr', key)
if tonumber(current) == 1 then
    redis.call('expire', key, time)
end
return tonumber(current)
這個(gè)腳本其實(shí)不難，大概瞅一眼就知道干啥用的。KEYS 和 ARGV 都是一會(huì)調(diào)用時(shí)候傳進(jìn)來(lái)的參數(shù)，tonumber 就是把字符串轉(zhuǎn)為數(shù)字，redis.call 就是執(zhí)行具體的 redis 指令，具體流程是這樣：


    

  • 首先獲取到傳進(jìn)來(lái)的 key 以及 限流的計(jì)數(shù) 和時(shí)間 time。
    • 

  • 通過(guò) get 獲取到這個(gè) key 對(duì)應(yīng)的值，這個(gè)值就是當(dāng)前時(shí)間窗內(nèi)這個(gè)接口可以訪問(wèn)多少次。
    • 

  • 如果是第一次訪問(wèn)，此時(shí)拿到的結(jié)果為 nil，否則拿到的結(jié)果應(yīng)該是一個(gè)數(shù)字，所以接下來(lái)就判斷，如果拿到的結(jié)果是一個(gè)數(shù)字，并且這個(gè)數(shù)字還大于 count，那就說(shuō)明已經(jīng)超過(guò)流量限制了，那么直接返回查詢的結(jié)果即可。
    • 

  • 如果拿到的結(jié)果為 nil，說(shuō)明是第一次訪問(wèn)，此時(shí)就給當(dāng)前 key 自增 1，然后設(shè)置一個(gè)過(guò)期時(shí)間。
    • 

  • 最后把自增 1 后的值返回就可以了。
    • 



其實(shí)這段 Lua 腳本很好理解。
接下來(lái)我們?cè)谝粋€(gè) Bean 中來(lái)加載這段 Lua 腳本，如下：
@Bean
public DefaultRedisScript limitScript() {
    DefaultRedisScript redisScript = new DefaultRedisScript<>();
    redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("lua/limit.lua")));
    redisScript.setResultType(Long.class);
    return redisScript;
}
可以啦，我們的 Lua 腳本現(xiàn)在就準(zhǔn)備好了。


5. 注解解析


接下來(lái)我們就需要自定義切面，來(lái)解析這個(gè)注解了，我們來(lái)看看切面的定義：
@Aspect
@Component
public class RateLimiterAspect {
    private static final Logger log = LoggerFactory.getLogger(RateLimiterAspect.class);

    @Autowired
    private RedisTemplate redisTemplate;

    @Autowired
    private RedisScript limitScript;

    @Before("@annotation(rateLimiter)")
    public void doBefore(JoinPoint point, RateLimiter rateLimiter) throws Throwable {
        String key = rateLimiter.key();
        int time = rateLimiter.time();
        int count = rateLimiter.count();

        String combineKey = getCombineKey(rateLimiter, point);
        List keys = Collections.singletonList(combineKey);
        try {
            Long number = redisTemplate.execute(limitScript, keys, count, time);
            if (number==null || number.intValue() > count) {
                throw new ServiceException("訪問(wèn)過(guò)于頻繁，請(qǐng)稍候再試");
            }
            log.info("限制請(qǐng)求'{}',當(dāng)前請(qǐng)求'{}',緩存key'{}'", count, number.intValue(), key);
        } catch (ServiceException e) {
            throw e;
        } catch (Exception e) {
            throw new RuntimeException("服務(wù)器限流異常，請(qǐng)稍候再試");
        }
    }

    public String getCombineKey(RateLimiter rateLimiter, JoinPoint point) {
        StringBuffer stringBuffer = new StringBuffer(rateLimiter.key());
        if (rateLimiter.limitType() == LimitType.IP) {
            stringBuffer.append(IpUtils.getIpAddr(((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest())).append("-");
        }
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        Class targetClass = method.getDeclaringClass();
        stringBuffer.append(targetClass.getName()).append("-").append(method.getName());
        return stringBuffer.toString();
    }
}
這個(gè)切面就是攔截所有加了 注解的方法，在前置通知中對(duì)注解進(jìn)行處理。@RateLimiter


    

  • 首先獲取到注解中的 key、time 以及 count 三個(gè)參數(shù)。
    • 

  • 獲取一個(gè)組合的 key，所謂的組合的 key，就是在注解的 key 屬性基礎(chǔ)上，再加上方法的完整路徑，如果是 IP 模式的話，就再加上 IP 地址。以 IP 模式為例，最終生成的 key 類似這樣：(如果不是 IP 模式，那么生成的 key 中就不包含 IP 地址)。rate_limit:127.0.0.1-org.javaboy.ratelimiter.controller.HelloController-hello
    • 

  • 將生成的 key 放到集合中。
    • 

  • 通過(guò) redisTemplate.execute 方法取執(zhí)行一個(gè) Lua 腳本，第一個(gè)參數(shù)是腳本所封裝的對(duì)象，第二個(gè)參數(shù)是 key，對(duì)應(yīng)了腳本中的 KEYS，后面是可變長(zhǎng)度的參數(shù)，對(duì)應(yīng)了腳本中的 ARGV。
    • 

  • 將 Lua 腳本執(zhí)行的結(jié)果與 count 進(jìn)行比較，如果大于 count，就說(shuō)明過(guò)載了，拋異常就行了。
    • 



好了，大功告成了。


6. 接口測(cè)試


接下來(lái)我們就進(jìn)行接口的一個(gè)簡(jiǎn)單測(cè)試，如下：
@RestController
public class HelloController {
    @GetMapping("/hello")
    @RateLimiter(time = 5,count = 3,limitType = LimitType.IP)
    public String hello() {
        return "hello>>>"+new Date();
    }
}
每一個(gè) IP 地址，在 5 秒內(nèi)只能訪問(wèn) 3 次。
這個(gè)自己手動(dòng)刷新瀏覽器都能測(cè)試出來(lái)。


7. 全局異常處理


由于過(guò)載的時(shí)候是拋異常出來(lái)，所以我們還需要一個(gè)全局異常處理器，如下：
@RestControllerAdvice
public class GlobalException {
    @ExceptionHandler(ServiceException.class)
    public Map serviceException(ServiceException e) {
        HashMap map = new HashMap<>();
        map.put("status", 500);
        map.put("message", e.getMessage());
        return map;
    }
}
這是一個(gè)小 demo，我就不去定義實(shí)體類了，直接用 Map 來(lái)返回 JSON 了。
好啦，大功告成。
最后我們看看過(guò)載時(shí)的測(cè)試效果：
好啦，這就是我們使用 Redis 做限流的方式。
                                                

                                                分享文章：Redis做接口限流，一個(gè)注解的事！                                                

                                                本文網(wǎng)址：http://www.5511xx.com/article/cdioedc.html