Kotlin+SpringBoot+Redis+Lua实现限流访问控制详解
liuian 2024-12-25 14:00 32 浏览
1、Redis是简介
- redis官方网
- Redis是一个开源的使用ANSI C语言编写、支持网络、可基于内存亦可持久化的日志型、Key-Value数据库,并提供多种语言的API。从2010年3月15日起,Redis的开发工作由VMware主持。从2013年5月开始,Redis的开发由Pivotal赞助
- 工程基于基础架构 Kotlin+SpringBoot+MyBatisPlus搭建最简洁的前后端分离框架美搭建最简洁最酷的前后端分离框架进行完善
2、Redis开发者
- redis 的作者,叫Salvatore Sanfilippo,来自意大利的西西里岛,现在居住在卡塔尼亚。目前供职于Pivotal公司。他使用的网名是antirez。
3、Redis安装
- Redis安装与其他知识点请参考几年前我编写文档 Redis Detailed operating instruction.pdf,这里不做太多的描述,主要讲解在kotlin+SpringBoot然后搭建Redis与遇到的问题
Redis详细使用说明书.pdf
https://github.com/jilongliang/kotlin/blob/dev-redis/src/main/resource/Redis%20Detailed%20operating%20instruction.pdf
4、Redis应该学习哪些?
- 列举一些常见的内容
5、Redis有哪些命令
Redis官方命令清单 https://redis.io/commands
- Redis常用命令
6、 Redis常见应用场景
7、 Redis常见的几种特征
- Redis的哨兵机制
- Redis的原子性
- Redis持久化有RDB与AOF方式
8、工程结构
9、Kotlin与Redis+Lua的代码实现
- Redis 依赖的Jar配置
<!-- Spring Boot Redis 依赖 -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>
<dependency>
<groupId>redis.clients</groupId>
<artifactId>jedis</artifactId>
</dependency>
- LuaConfiguration
- 设置加载限流lua脚本
@Configuration
class LuaConfiguration {
@Bean
fun redisScript(): DefaultRedisScript<Number> {
val redisScript = DefaultRedisScript<Number>()
//设置限流lua脚本
redisScript.setScriptSource(ResourceScriptSource(ClassPathResource("limitrate.lua")))
//第1种写法反射转换Number类型
//redisScript.setResultType(Number::class.java)
//第2种写法反射转换Number类型
redisScript.resultType = Number::class.java
return redisScript
}
}
- Lua限流脚本
local key = "request:limit:rate:" .. KEYS[1] --限流KEY
local limitCount = tonumber(ARGV[1]) --限流大小
local limitTime = tonumber(ARGV[2]) --限流时间
local current = tonumber(redis.call('get', key) or "0")
if current + 1 > limitCount then --如果超出限流大小
return 0
else --请求数+1,并设置1秒过期
redis.call("INCRBY", key,"1")
redis.call("expire", key,limitTime)
return current + 1
end
- RateLimiter自定义注解
- 1、Java自定义注解Target使用@Target(ElementType.TYPE, ElementType.METHOD)
- 2、Java自定义注解Retention使用@Retention(RetentionPolicy.RUNTIME)
- 3、Kotlin自定义注解Target使用@Target(AnnotationTarget.TYPE, AnnotationTarget.FUNCTION)
- 4、Kotlin自定义注解Target使用@Retention(AnnotationTarget.TYPE, AnnotationTarget.FUNCTION)
@Target(AnnotationTarget.TYPE, AnnotationTarget.FUNCTION)
@Retention(AnnotationRetention.RUNTIME)
annotation class RateLimiter(
/**
* 限流唯一标识
* @return
*/
val key: String = "",
/**
* 限流时间
* @return
*/
val time: Int,
/**
* 限流次数
* @return
*/
val count: Int
)
限流AOP的Aspect切面实现
import com.flong.kotlin.core.annotation.RateLimiter
import com.flong.kotlin.core.exception.BaseException
import com.flong.kotlin.core.exception.CommMsgCode
import com.flong.kotlin.core.vo.ErrorResp
import com.flong.kotlin.utils.WebUtils
import org.aspectj.lang.ProceedingJoinPoint
import org.aspectj.lang.annotation.Around
import org.aspectj.lang.annotation.Aspect
import org.aspectj.lang.reflect.MethodSignature
import org.slf4j.Logger
import org.slf4j.LoggerFactory
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.context.annotation.Configuration
import org.springframework.data.redis.core.RedisTemplate
import org.springframework.data.redis.core.script.DefaultRedisScript
import org.springframework.web.context.request.RequestContextHolder
import org.springframework.web.context.request.ServletRequestAttributes
import java.util.*
@Suppress("SpringKotlinAutowiring")
@Aspect
@Configuration
class RateLimiterAspect {
@Autowired lateinit var redisTemplate: RedisTemplate<String, Any>
@Autowired var redisScript: DefaultRedisScript<Number>? = null
/**
* 半生对象
*/
companion object {
private val log: Logger = LoggerFactory.getLogger(RateLimiterAspect::class.java)
}
@Around("execution(* com.flong.kotlin.modules.controller ..*(..) )")
@Throws(Throwable::class)
fun interceptor(joinPoint: ProceedingJoinPoint): Any {
val signature = joinPoint.signature as MethodSignature
val method = signature.method
val targetClass = method.declaringClass
val rateLimit = method.getAnnotation(RateLimiter::class.java)
if (rateLimit != null) {
val request = (RequestContextHolder.getRequestAttributes() as ServletRequestAttributes).request
val ipAddress = WebUtils.getIpAddr(request = request)
val stringBuffer = StringBuffer()
stringBuffer.append(ipAddress).append("-")
.append(targetClass.name).append("- ")
.append(method.name).append("-")
.append(rateLimit!!.key)
val keys = Collections.singletonList(stringBuffer.toString())
print(keys + rateLimit!!.count + rateLimit!!.time)
val number = redisTemplate!!.execute<Number>(redisScript, keys, rateLimit!!.count, rateLimit!!.time)
if (number != null && number!!.toInt() != 0 && number!!.toInt() <= rateLimit!!.count) {
log.info("限流时间段内访问第:{} 次", number!!.toString())
return joinPoint.proceed()
}
} else {
var proceed: Any? = joinPoint.proceed() ?: return ErrorResp(CommMsgCode.SUCCESS.code!!, CommMsgCode.SUCCESS.message!!)
return joinPoint.proceed()
}
throw BaseException(CommMsgCode.RATE_LIMIT.code!!, CommMsgCode.RATE_LIMIT.message!!)
}
}
WebUtils代码
import javax.servlet.http.HttpServletRequest
/**
* User: liangjl
* Date: 2020/7/28
* Time: 10:01
*/
object WebUtils {
fun getIpAddr(request: HttpServletRequest): String? {
var ipAddress: String? = null
try {
ipAddress = request.getHeader("x-forwarded-for")
if (ipAddress == null || ipAddress!!.length == 0 || "unknown".equals(ipAddress!!, ignoreCase = true)) {
ipAddress = request.getHeader("Proxy-Client-IP")
}
if (ipAddress == null || ipAddress!!.length == 0 || "unknown".equals(ipAddress!!, ignoreCase = true)) {
ipAddress = request.getHeader("WL-Proxy-Client-IP")
}
if (ipAddress == null || ipAddress!!.length == 0 || "unknown".equals(ipAddress!!, ignoreCase = true)) {
ipAddress = request.getRemoteAddr()
}
// 对于通过多个代理的情况,第一个IP为客户端真实IP,多个IP按照','分割
if (ipAddress != null && ipAddress!!.length > 15) {
// "***.***.***.***".length()= 15
if (ipAddress!!.indexOf(",") > 0) {
ipAddress = ipAddress!!.substring(0, ipAddress.indexOf(","))
}
}
} catch (e: Exception) {
ipAddress = ""
}
return ipAddress
}
}
- Controller代码
@RestController
@RequestMapping("rest")
class RateLimiterController {
companion object {
private val log: Logger = LoggerFactory.getLogger(RateLimiterController::class.java)
}
@Autowired
private val redisTemplate: RedisTemplate<*, *>? = null
@GetMapping(value = "/limit")
@RateLimiter(key = "limit", time = 10, count = 1)
fun limit(): ResponseEntity<Any> {
val date = DateFormatUtils.format(Date(), "yyyy-MM-dd HH:mm:ss.SSS")
val limitCounter = RedisAtomicInteger("limit:rate:counter", redisTemplate!!.connectionFactory!!)
val str = date + " 累计访问次数:" + limitCounter.andIncrement
log.info(str)
return ResponseEntity.ok<Any>(str)
}
}
注意:RedisTemplateK, V>这个类由于有K与V,下面的做法是必须要指定Key-Value 2 type arguments expected for class RedisTemplate
- 运行结果
- 访问地址:http://localhost:8080/rest/limit
10、参考文章
参考分布式限流之Redis+Lua实现参考springboot + aop + Lua分布式限流的最佳实践
11、工程架构源代码
Kotlin+SpringBoot+Redis+Lua实现限流访问控制详解工程源代码
https://github.com/jilongliang/kotlin/tree/dev-lua
12 、总结与建议
- 1 、以上问题根据搭建 kotlin与Redis实际情况进行总结整理,除了技术问题查很多网上资料,通过自身进行学习之后梳理与分享。
- 2、 在学习过程中也遇到很多困难和疑点,如有问题或误点,望各位老司机多多指出或者提出建议。本人会采纳各种好建议和正确方式不断完善现况,人在成长过程中的需要优质的养料。
- 3、 希望此文章能帮助各位老铁们更好去了解如何在 kotlin上搭建RabbitMQ,也希望您看了此文档或者通过找资料进行手动安装效果会更好。
备注:此文章属于本人原创,欢迎转载和收藏.
相关推荐
- 10种常见的MySQL错误,你可中招?
-
【51CTO.com快译】如果未能对MySQL8进行恰当的配置,您非但可能遇到无法顺利访问、或调用MySQL的窘境,而且还可能给真实的应用生产环境带来巨大的影响。本文列举了十种MySQL...
- MySQL主从如何保证数据一致性
-
MySQL主从(主备)搭建请点击基于Spring的数据库读写分离。MySQL主备基本原理假设主备切换前,我们的主库是节点A,节点B是节点A的备库,客户端的读写都是直接访问节点A,节点B只是将A的更新同...
- MySQL低版本升级操作流程
-
(关注“数据库架构师”公众号,提升数据库技能,助力职业发展)0-升级背景MySQL5.5发布于2010年,至今已有十年历史,官方已经停止更新。2008年发布的MySQL5.1版本,在2018年...
- MySQL数据库知识
-
MySQL是一种关系型数据库管理系统;那废话不多说,直接上自己以前学习整理文档:查看数据库命令:(1).查看存储过程状态:showprocedurestatus;(2).显示系统变量:show...
- Mysql 8.4数据库安装、新建用户和数据库、表单
-
1、下载MySQL数据库yuminstall-ywgetperlnet-toolslibtirpc#安装wget和perl、net-tools、libtirpcwgethtt...
- mysql8.0新功能介绍
-
MySQL8.0新特性集锦一、默认字符集由latin1变为utf8mb4在8.0版本之前,默认字符集为latin1,utf8指向的是utf8mb3,8.0版本默认字符集为utf8mb4,utf8默...
- 全网最详细解决Windows下Mysql数据库安装后忘记初始root 密码方法
-
一、准备重置root的初始化密码Win+R键启动命令输入窗口;输入cmd打开命令执行窗口;##界面如下##输入命令:netstopmysqld#此操作会停止当前运行的...
- 互联网大厂面试:MySQL使用grant授权后必须flush privilege吗
-
从我上大学时,数据库概论老师就告诉我,MySQL使用grant对用户授权之后,一定记得要用flushprivilege命令刷新缓存,这样才能使赋权命令生效。毕业工作以后,在很多的技术文档上,仍然可以...
- # mysql 8.0 版本无法使用 sqlyog 等图形界面 登录 的解决方法
-
30万以下的理想L6来了##mysql8.0版本无法使用sqlyog等图形界面登录的解决方法当我们在cmd下登录mysql时正常时,用sqlyog等图形界面连接数据库时却...
- MySQL触发器介绍
-
前言:在学习MySQL的过程中,可能你了解过触发器的概念,不清楚各位是否有详细的去学习过触发器,最近看了几篇关于触发器的文档,分享下MySQL触发器相关知识。1.触发器简介触发器即trigg...
- 管理员常用的MySQL命令汇总(一)
-
以下是管理员常用的MySQL命令:以管理员身份连接到MySQL:mysql-uroot-p创建新的MySQL用户:CREATEUSER'username'@'...
- Linux(CentOS) 在线安装MySQL8.0和其他版本,修改root密码
-
一:安装MySQL数据库1),下载并安装MySQL官方的YumRepositorymysql官方仓库地址:https://dev.mysql.com/downloads/repo/yum/选择自...
- 解决 MySQL 8.0 一直拒绝 root 登录问题
-
Accessdeniedforuser'root'@'localhost'(usingpassword:YES)这个错误在网上搜一下,能看到非常多的此类...
- 大模型MCP之MYSQL安装
-
前言学习大模型的时候需要一个mysql,原因还是在公司使用电脑的时候不允许按照Docker-Desktop,我的宿主机其实是MAC,我习惯上还是在centsos上面安装,就发现这件过去很简单的事情居然...
- MySQL ERROR 1396
-
ERROR1396(HY000):OperationCREATEUSERfailedfor'usera'@'%'问题描述mysql>create...
- 一周热门
-
-
Python实现人事自动打卡,再也不会被批评
-
【验证码逆向专栏】vaptcha 手势验证码逆向分析
-
Psutil + Flask + Pyecharts + Bootstrap 开发动态可视化系统监控
-
一个解决支持HTML/CSS/JS网页转PDF(高质量)的终极解决方案
-
再见Swagger UI 国人开源了一款超好用的 API 文档生成框架,真香
-
网页转成pdf文件的经验分享 网页转成pdf文件的经验分享怎么弄
-
C++ std::vector 简介
-
系统C盘清理:微信PC端文件清理,扩大C盘可用空间步骤
-
飞牛OS入门安装遇到问题,如何解决?
-
10款高性能NAS丨双十一必看,轻松搞定虚拟机、Docker、软路由
-
- 最近发表
- 标签列表
-
- python判断字典是否为空 (50)
- crontab每周一执行 (48)
- aes和des区别 (43)
- bash脚本和shell脚本的区别 (35)
- canvas库 (33)
- dataframe筛选满足条件的行 (35)
- gitlab日志 (33)
- lua xpcall (36)
- blob转json (33)
- python判断是否在列表中 (34)
- python html转pdf (36)
- 安装指定版本npm (37)
- idea搜索jar包内容 (33)
- css鼠标悬停出现隐藏的文字 (34)
- linux nacos启动命令 (33)
- gitlab 日志 (36)
- adb pull (37)
- python判断元素在不在列表里 (34)
- python 字典删除元素 (34)
- vscode切换git分支 (35)
- python bytes转16进制 (35)
- grep前后几行 (34)
- hashmap转list (35)
- c++ 字符串查找 (35)
- mysql刷新权限 (34)