1、概述

  限流的含義是在單位時間內確保發(fā)往某個模塊的請求數(shù)量小于某個數(shù)值，比如在實現(xiàn)秒殺功能時，需要確保在10秒內發(fā)往支付模塊的請求數(shù)量小于500個。限流的作用是防止某個段時間段內的請求數(shù)過多，造成模塊因高并發(fā)而不可用。

2、zset有序集合相關命令與限流

  zset也叫有序集合，是Redis的一種數(shù)據類型，在其中每個值(value)都會有一個對應的score參數(shù)，以此來描述該值的權重分值?？梢酝ㄟ^如下形式的命令向zset有序集合里添加元素：

zadd key score value

  在限流相關的應用里，可以通過zadd命令往有序集合里存放數(shù)據時引入表示權重的score參數(shù)，并在其中存放時間戳。由于zset是有序集合，因此包含時間戳的score能被排序，這樣就能用zremrangeByScore命令去除指定時間范圍內的數(shù)據。zrenrabgeByScore命令的語法如下：

zremrangeByScore key min max

  通過該命令能在有序集合里刪除鍵為key、score值在min到max范圍內的數(shù)據。通過這種刪除動作，能排除限流時間范圍外的數(shù)據，并能在此基礎上通過zcard命令統(tǒng)計有序集合內元素的數(shù)量，以確保請求數(shù)量小于限流的上限值。zcard命令的語法格式如下：

zcard key

  該命令會返回有序集合內指定key的元素數(shù)量。

3、zset有序結合與限流

有關redis啟動的知識不做過多介紹了，這里我用docker創(chuàng)建的redis容器。

本例實現(xiàn)100秒內只能處理3個請求

LimitRequest.java

package com.baizhi.request;

import redis.clients.jedis.Jedis;

public class LimitRequest {
    public static void main(String[] args) {
        Jedis jedis=new Jedis("192.168.159.33",6379);
        jedis.del("PayRequest");
        //模擬發(fā)5個請求
        int cnt=5;
        for (int i = 0; i < cnt; i++) {
            LimitUtil.canVisit(jedis,"PayRequest",100,3);
        }
    }
}
class LimitUtil{
    //判斷是否需要限流
    public static void canVisit(Jedis jedis,String requestType,int limitTime,int limitNum){
        long currentTime=System.currentTimeMillis();
        //把請求放入zset
        jedis.zadd(requestType,currentTime,Long.valueOf(currentTime).toString());
        //去掉時間范圍外（超時）的請求
        jedis.zremrangeByScore(requestType,0,currentTime-limitTime*1000);
        //統(tǒng)計時間范圍內總數(shù)
        Long count=jedis.zcard(requestType);
        //設置所有請求的超時時間
        jedis.expire(requestType,limitTime+1);
        boolean flag=limitNum>=count;
        if(flag){
            System.out.println("Can visit");
        }else{
            System.out.println("Can not visit");
        }
    }
}

  先通過zadd方法把表示操作類型的requestType作為鍵插入有序集合，插入時用表示當前時間的currentTime作為值，以保證值的唯一性，同時用currentTime作為有序集合里元素的score值。
   隨后通過zremrangeByScore命令去除0到距當前時間limitTime時間范圍內的數(shù)據。比如限流的時間范圍是100秒，那么通過zremrangeByScore命令就能在有序集合里去除score范圍從0到距離當前時間100秒的數(shù)據，這樣就能確保有序集合內只存有最近100秒內發(fā)來的元素。
  其中，用zcard命令統(tǒng)計有序集合內鍵為requestType的個數(shù)，如果通過if語句發(fā)現(xiàn)當前個數(shù)還沒有達到限流的上限，則允許該請求方法，否則不允許。這里通過最后的打印語句來模擬“是否允許請求訪問”的動作。
  同時請注意，需要使用expire語句設置有序集合里相關鍵的超時時間，這樣就能確保在限流動作完成后這些鍵能自動刪除，而不是一致駐留在內存中。
  在main函數(shù)里，首先創(chuàng)建指向虛擬機6379端口的jedis對象，為了確保在多次運行時數(shù)據不相互干擾，在運行前通過del語句刪除相關的鍵。
   隨后使用for循環(huán)里通過調用canVisit方法模擬發(fā)出5個請求。在調用該方法時，通過傳入參數(shù)指定限流所用的鍵為"PayRequest",同時指定了100秒內只能處理3個請求。運行之后控制臺的效果如下：

到此這篇關于用Java實現(xiàn)Redis限流的文章就介紹到這了,更多相關Java實現(xiàn)Redis限流內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

最新評論