czg_team
/
video_server
7
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

频率限制2

This commit is contained in:
GYJ 2025-03-23 16:52:08 +08:00
parent 7a4a57c4ef
commit 35bf6e479e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/sqx/common/aspect/RateLimitAspect.java
View File

@ -2,6 +2,7 @@ package com.sqx.common.aspect;
import com.google.common.util.concurrent.RateLimiter; import com.google.common.util.concurrent.RateLimiter;
import com.sqx.common.annotation.Limiting; import com.sqx.common.annotation.Limiting;
import com.sqx.common.utils.IPUtils;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint; import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.Signature; import org.aspectj.lang.Signature;
@ -37,7 +38,7 @@ public class RateLimitAspect {
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes(); ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
HttpServletRequest request = attributes.getRequest(); HttpServletRequest request = attributes.getRequest();
// 获取请求的 IP 地址 // 获取请求的 IP 地址
String ip = request.getRemoteAddr(); String ip = IPUtils.getIpAddr(request);
// 获取方法上的 @AccessLimit 注解 // 获取方法上的 @AccessLimit 注解
MethodSignature signature = (MethodSignature) point.getSignature(); MethodSignature signature = (MethodSignature) point.getSignature();