----1:failed
—1:success
—@param key:key限流关键字
—@param apply:申请的令牌数量
local function acquire(key, apply)
local times = redis.call(‘TIME’);
–times[1] 秒数 --times[2] 微秒数
local curr_mill_second = times[1] *1000000 + times[2];
curr_mill_second = curr_mill_second / 1000;
local cacheInfo = redis.pcall(“HMGET”, key, “last_mill_second”, “curr_permits”, “max_permits”, “rate”)
—局部变量:上次申请的时间
local last_mill_second = cacheInfo[1];
—局部变量:之前的令牌数
local curr_permits = tonumber(cacheInfo[2]);
—局部变量:桶的容量
local max_permits = tonumber(cacheInfo[3]);
—局部变量:令牌的发放速率
local rate = cacheInfo[4];
—局部变量:本次的令牌数
local local_curr_permits = max_permits;
if (type(last_mill_second) ~= ‘boolean’ and last_mill_second ~= nil) then
–计算时间段内的令牌数
local reverse_permits = math.floor(((curr_mill_second - last_mill_second) / 1000) *rate);
–令牌总数
local expect_curr_permits = reverse_permits + curr_permits;
–可以申请的令牌总数
local_curr_permits = math.min(expect_curr_permits, max_permits);
else
–第一次获取令牌
redis.pcall(“HSET”, key, “last_mill_second”, curr_mill_second)
end
local result = -1;
–有足够的令牌可以申请
if (local_curr_permits - apply >= 0) then
–保存剩余的令牌
redis.pcall(“HSET”, key, “curr_permits”, local_curr_permits - apply);
–保存时间,下次令牌获取时使用
redis.pcall(“HSET”, key, “last_mill_second”, curr_mill_second)
–返回令牌获取成功
result = 1;
else
–保存令牌总数
redis.pcall(“HSET”, key, “curr_permits”, local_curr_permits);
–返回令牌获取失败
result = -1;
end
return result
end
—方法:初始化限流器
—1 success
—@param key key
—@param max_permits 桶的容量
—@param rate 令牌的发放速率
local function init(key, max_permits, rate)
local rate_limit_info = redis.pcall(“HMGET”, key, “last_mill_second”, “curr_permits”, “max_permits”, “rate”)
local org_max_permits = tonumber(rate_limit_info[3])
local org_rate = rate_limit_info[4]
if (org_max_permits == nil) or (rate ~= org_rate or max_permits ~= org_max_permits) then
redis.pcall(“HMSET”, key, “max_permits”, max_permits, “rate”, rate, “curr_permits”, max_permits)
end
return 1;
end
—方法:删除限流Key
local function delete(key)
redis.pcall(“DEL”, key) return 1;
end
local key = KEYS[1]
local method = ARGV[1]
if method == ‘acquire’ then
return acquire(key, ARGV[2], ARGV[3])
elseif method == ‘init’ then
return init(key, ARGV[2], ARGV[3])
elseif method == ‘delete’ then
return delete(key)
else
–ignore
end
该脚本有3个方法,其中两个方法比较重要,分别说明如下:
(1)限流器初始化方法init(key,max_permits,rate),此方法在限流开始时被调用。
(2)限流检测的方法acquire(key,apply),此方法在请求到来时被调用。
Java分布式令牌桶限流
============
rate_limiter.lua脚本既可以在Java中调用,又可以在Nginx中调用。本小节先介绍其在Java中的使用,第10章再介绍其在Nginx中的使用。
Java分布式令牌桶限流器的实现就是通过Java代码向Redis加载rate_limiter.lua脚本,然后封装其令牌桶初始化方法init(…)和限流监测方法acquire(…),以供外部调用。它的代码如下:
package com.crazymaker.springcloud.standard.ratelimit;
…
/**
*实现:令牌桶限流服务
*create by尼恩 @ 疯狂创客圈
**/
@Slf4j
public class RedisRateLimitImpl implements RateLimitService, InitializingBean
{
/**
*限流器的redis key前缀
*/
private static final String RATE_LIMITER_KEY_PREFIX = “rate_limiter:”;
//private ScheduledExecutorService executorService = Executors.newScheduledThreadPool(1);
private RedisRateLimitProperties redisRateLimitProperties;
private RedisTemplate redisTemplate;
//lua脚本的实例
private static RedisScript rateLimiterScript = null;
//lua脚本的类路径
private static String rateLimitLua = “script/rate_limiter.lua”;
static
{
//从类路径文件中加载令牌桶lua脚本
String script = IOUtil.loadJarFile(RedisRateLimitImpl.class.getClassLoader(), rateLimitLua);
if (StringUtils.isEmpty(script))
{
log.error(“lua script load failed:” + rateLimitLua);
} else
{
//创建Lua脚本实例
rateLimiterScript = new DefaultRedisScript<>(script, Long.class);
}
}
public RedisRateLimitImpl(
RedisRateLimitProperties redisRateLimitProperties,
RedisTemplate redisTemplate)
{
this.redisRateLimitProperties = redisRateLimitProperties;
this.redisTemplate = redisTemplate;
}
private Map<String, LimiterInfo> limiterInfoMap = new HashMap<>();
/**
*限流器的信息
*/
@Builder
@Data
public static class LimiterInfo
{
/**
*限流器的key,如秒杀的id
*/ private String key;
/**
*限流器的类型,如seckill
*/
private String type = “default”;
/**
*限流器的最大桶容量
*/
private Integer maxPermits;
/**
*限流器的速率
*/
private Integer rate;
/**
*限流器的redis key
*/
public String fullKey()
{
return RATE_LIMITER_KEY_PREFIX + type + “:” + key;
}
/**
*限流器在map中的缓存key
*/
public String cashKey()
{
return type + “:” + key;
}
}
/**
*限流检测:是否超过redis令牌桶限速器的限制
*@param cacheKey计数器的key
*@return true or false
*/
@Override
public Boolean tryAcquire(String cacheKey)
{
if (cacheKey == null)
{
return true;
}
if (cacheKey.indexOf(“:”) <= 0)
自我介绍一下,小编13年上海交大毕业,曾经在小公司待过,也去过华为、OPPO等大厂,18年进入阿里一直到现在。
深知大多数Java工程师,想要提升技能,往往是自己摸索成长或者是报班学习,但对于培训机构动则几千的学费,着实压力不小。自己不成体系的自学效果低效又漫长,而且极易碰到天花板技术停滞不前!
因此收集整理了一份《2024年Java开发全套学习资料》,初衷也很简单,就是希望能够帮助到想自学提升又不知道该从何学起的朋友,同时减轻大家的负担。
既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Java开发知识点,真正体系化!
由于文件比较大,这里只是将部分目录大纲截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且后续会持续更新
如果你觉得这些内容对你有帮助,可以添加V获取:vip1024b (备注Java)
最后
终极手撕架构师的学习笔记:分布式+微服务+开源框架+性能优化
一个人可以走的很快,但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎扫码加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!
文件比较大,这里只是将部分目录大纲截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且后续会持续更新**
如果你觉得这些内容对你有帮助,可以添加V获取:vip1024b (备注Java)
[外链图片转存中…(img-yGpB0M7Z-1712713584446)]
最后
终极手撕架构师的学习笔记:分布式+微服务+开源框架+性能优化
[外链图片转存中…(img-bZH97HGq-1712713584446)]
一个人可以走的很快,但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎扫码加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!
[外链图片转存中…(img-yqeytHqC-1712713584447)]