审核附件修改

chenhj

8 天以前 3f2d4ba07ee9e1e0ddba97b43dc70af09f0e39a2

 src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java

@@ -1,8 +1,10 @@
package com.ruoyi.framework.aspectj;



import java.lang.reflect.Method;

import java.util.Collections;

import java.util.List;

import com.ruoyi.common.exception.ServiceException;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.common.utils.ip.IpUtils;

import com.ruoyi.framework.aspectj.lang.annotation.RateLimiter;

import com.ruoyi.framework.aspectj.lang.enums.LimitType;

import org.aspectj.lang.JoinPoint;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Before;

@@ -13,12 +15,10 @@
import org.springframework.data.redis.core.RedisTemplate;

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

import org.springframework.stereotype.Component;

import com.ruoyi.common.exception.ServiceException;

import com.ruoyi.common.utils.ServletUtils;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.common.utils.ip.IpUtils;

import com.ruoyi.framework.aspectj.lang.annotation.RateLimiter;

import com.ruoyi.framework.aspectj.lang.enums.LimitType;



import java.lang.reflect.Method;

import java.util.Collections;

import java.util.List;



/**

 * 限流处理

@@ -79,7 +79,7 @@
        StringBuffer stringBuffer = new StringBuffer(rateLimiter.key());

        if (rateLimiter.limitType() == LimitType.IP)

        {

            stringBuffer.append(IpUtils.getIpAddr(ServletUtils.getRequest())).append("-");

            stringBuffer.append(IpUtils.getIpAddr()).append("-");

        }

        MethodSignature signature = (MethodSignature) point.getSignature();

        Method method = signature.getMethod();