Redis

中的应用令牌桶算法是一种常用的限流算法适用于控制请求的频率防止系统过载。

结合

Redis

dependenciesdependencygroupIdorg.springframework.boot/groupIdartifactIdspring-boot-starter-data-redis/artifactId/dependency

/dependencies二.、配置

spring.redis.jedis.pool.max-active8

spring.redis.jedis.pool.max-wait-1

spring.redis.jedis.pool.max-idle8

spring.redis.jedis.pool.min-idle0

三.、实现令牌桶算法

org.springframework.beans.factory.annotation.Autowired;

import

org.springframework.data.redis.core.StringRedisTemplate;

import

org.springframework.data.redis.core.script.DefaultRedisScript;

import

org.springframework.data.redis.core.script.RedisScript;

import

org.springframework.stereotype.Component;import

import

java.nio.charset.StandardCharsets;

import

request_rate_limiter.lua;Autowiredprivate

StringRedisTemplate

Redis提供的一个类用于封装Lua脚本。

RedisScriptLong

redisScript

:timestamp用于存储上一次令牌桶被填充的时间戳。

ListString

keys

Arrays.asList(TOKEN_BUCKET_KEY_PREFIX

key

String.valueOf(refillRate)令牌桶的填充速率。

*

String.valueOf(limitCount)令牌桶的最大容量。

*

String.valueOf(System.currentTimeMillis()

result

stringRedisTemplate.execute(redisScript,

//Redis键列表String.valueOf(refillRate),

//令牌桶的填充速率String.valueOf(limitCount),

//令牌桶的最大容量String.valueOf(System.currentTimeMillis()

1000),

//请求的令牌数量这里假设每次请求需要1个令牌。

);return

result

获取资源流。

我的lua文件在resources根目录下面//使用

InputStreamReader

包装成一个字符流并指定字符编码为UTF-8。

这样可以确保读取的文件内容是正确的编码格式reader

new

InputStreamReader(getClass().getClassLoader().getResourceAsStream(LUA_SCRIPT_PATH),

StandardCharsets.UTF_8);/*创建一个Scanner对象用于读取字符流。

useDelimiter(\\A)设置分隔符为文件的开始标记

\\A

。

这意味着Scanner会将整个文件内容视为一个单一的字符串。

next()

读取并返回整个文件内容*/return

java.util.Scanner(reader).useDelimiter(\\A).next();}

catch

com.lhx.testany.controller;import

org.springframework.beans.factory.annotation.Autowired;

import

org.springframework.web.bind.annotation.GetMapping;

import

org.springframework.web.bind.annotation.RestController;RestController

public

tokenBucket;GetMapping(/test)public

String

我快速在浏览器调用时//生成的令牌不足于被消耗就会执行esle,这样就能做限流if

1))