快捷導(dǎo)航

Redis?Lua腳本實(shí)現(xiàn)ip限流示例

更新時(shí)間：2022年07月15日 11:45:22 作者：yin

這篇文章主要介紹了Redis?Lua腳本實(shí)現(xiàn)ip限流示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

引言

分布式限流最關(guān)鍵的是要將限流服務(wù)做成原子化，而解決方案可以使使用redis+lua或者nginx+lua技術(shù)進(jìn)行實(shí)現(xiàn)，通過(guò)這兩種技術(shù)可以實(shí)現(xiàn)的高并發(fā)和高性能。
首先我們來(lái)使用redis+lua實(shí)現(xiàn)時(shí)間窗內(nèi)某個(gè)接口的請(qǐng)求數(shù)限流，實(shí)現(xiàn)了該功能后可以改造為限流總并發(fā)/請(qǐng)求數(shù)和限制總資源數(shù)。Lua本身就是一種編程語(yǔ)言，也可以使用它實(shí)現(xiàn)復(fù)雜的令牌桶或漏桶算法。
如下操作因是在一個(gè)lua腳本中（相當(dāng)于原子操作），又因Redis是單線程模型，因此是線程安全的。

相比Redis事務(wù)來(lái)說(shuō)，Lua腳本有以下優(yōu)點(diǎn)

減少網(wǎng)絡(luò)開(kāi)銷: 不使用 Lua 的代碼需要向 Redis 發(fā)送多次請(qǐng)求, 而腳本只需一次即可, 減少網(wǎng)絡(luò)傳輸;
原子操作: Redis 將整個(gè)腳本作為一個(gè)原子執(zhí)行, 無(wú)需擔(dān)心并發(fā), 也就無(wú)需事務(wù);
復(fù)用: 腳本會(huì)永久保存 Redis 中, 其他客戶端可繼續(xù)使用.

Lua腳本

local key = KEYS[1] --限流KEY（一秒一個(gè)）
local limit = tonumber(ARGV[1]) --限流大小
local current = tonumber(redis.call('get', key) or "0")
if current + 1 > limit then --如果超出限流大小
    return 0
else --請(qǐng)求數(shù)+1，并設(shè)置2秒過(guò)期
    redis.call("INCRBY", key,"1")
    redis.call("expire", key,"2")
end
return 1

java代碼

import org.apache.commons.io.FileUtils;
import redis.clients.jedis.Jedis;
import java.io.File;
import java.io.IOException;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;
public class RedisLimitRateWithLUA {
    public static void main(String[] args) {
        final CountDownLatch latch = new CountDownLatch(1);
        for (int i = 0; i < 7; i++) {
            new Thread(new Runnable() {
                public void run() {
                    try {
                        latch.await();
                        System.out.println("請(qǐng)求是否被執(zhí)行："+accquire());
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                }
            }).start();
        }
        latch.countDown();
    }
    public static boolean accquire() throws IOException, URISyntaxException {
        Jedis jedis = new Jedis("127.0.0.1");
        File luaFile = new File(RedisLimitRateWithLUA.class.getResource("/").toURI().getPath() + "limit.lua");
        String luaScript = FileUtils.readFileToString(luaFile);
        String key = "ip:" + System.currentTimeMillis()/1000; // 當(dāng)前秒
        String limit = "5"; // 最大限制
        List<String> keys = new ArrayList<String>();
        keys.add(key);
        List<String> args = new ArrayList<String>();
        args.add(limit);
        Long result = (Long)(jedis.eval(luaScript, keys, args)); // 執(zhí)行l(wèi)ua腳本，傳入?yún)?shù)
        return result == 1;
    }
}

運(yùn)行結(jié)果

請(qǐng)求是否被執(zhí)行：true
請(qǐng)求是否被執(zhí)行：true
請(qǐng)求是否被執(zhí)行：false
請(qǐng)求是否被執(zhí)行：true
請(qǐng)求是否被執(zhí)行：true
請(qǐng)求是否被執(zhí)行：true
請(qǐng)求是否被執(zhí)行：fals

從結(jié)果可看出只有5個(gè)請(qǐng)求成功執(zhí)行

IP限流Lua腳本

local key = "rate.limit:" .. KEYS[1]
local limit = tonumber(ARGV[1])
local expire_time = ARGV[2]
local is_exists = redis.call("EXISTS", key)
if is_exists == 1 then
    if redis.call("INCR", key) > limit then
        return 0
    else
        return 1
    end
else
    redis.call("SET", key, 1)
    redis.call("EXPIRE", key, expire_time)
    return 1
end

參考 http://www.dbjr.com.cn/books/561366.html

以上就是Redis Lua腳本實(shí)現(xiàn)ip限流示例的詳細(xì)內(nèi)容，更多關(guān)于Redis Lua限流的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: