0517_【掌握】Redis流量限制.pptx

《Redis开发实战》Redis流量限制

业务限流处理线上应用经常会受到各种爬虫或者是恶意访问的攻击，所以在进行业务设计时，往往要限制单个IP的访问次数，当达到了指定的访问量后，则会认为认为该IP存在有问题，需要短暂进行限流处理

RedisTempate与Lua脚本执行RedisTemplate模版类提供了execute()方法，该方法可以执行Lua脚本程序，而为了便于脚本程序的管理，SpringDataRedis又提供了一个RedisScript接口，在实际使用时，需要通过该接口实例描述脚本程序

在src/main/resources中创建“lua/limit.lua”脚本文件localkey=KEYS[1]--模块限流KEY

locallimit=tonumber(ARGV[1])--流量配置

localcurrent=tonumber(redis.call(GET,key)or0)--当前访问量

ifcurrent+1limitthen--超出限流要求

returnfalse--访问失败

else--修改限流值

redis.call(INCRBY,key,1)--数据增加

redis.call(EXPIRE,key,2)--2秒后过期

end

returntrue--允许访问

RedisLuaConfigurationpackagecom.yootk.config;

@Configuration

publicclassRedisScriptConfiguration{

@Bean

publicDefaultRedisScriptBooleanlimitScript(){//限流脚本

DefaultRedisScriptBooleanscript=newDefaultRedisScript();

script.setLocation(newClassPathResource(lua/limit.lua));//Lua脚本路径

script.setResultType(Boolean.class);//脚本返回类型

returnscript;

}

}

创建限流注解packagecom.yootk.annotation;

@Target(value=ElementType.METHOD)//方法上使用

@Retention(RetentionPolicy.RUNTIME)//运行时生效

public@interfaceAccessLimit{

intlimit()default1000;//请求限制，每秒最多允许1000个

Stringmodule()default;//模块名称

}

ILimitService业务接口packagecom.yootk.service;

publicinterfaceILimitService{

@AccessLimit(limit=5,module=limit)//服务限流

publicStringget();//获取消息内容

}

创建LimitServiceImpl业务接口实现子类packagecom.yootk.service.impl;

@Service

publicclassLimitServiceImplimplementsILimitService{

@Override

publicStringget(){

return沐言科技：;

}

}

创建限流访问切面类packagecom.yootk.aop;

@Aspect

@Component

publicclassAccessLimitAspect{

privatestaticfinalStringPREFIX=Yootk-Limit;//前缀

privatestaticfinalStringSPLIT=:;//分隔符

@Autowired

privateRedisScriptBooleanlimitScript;//限流脚本

@Autowired

privateRedisTemplateString,ObjectredisTemplate;//Redis模版

@Pointcut(execution(public*com.y