From ee58dc26ada00ec0b191f546a73cb807ec048a78 Mon Sep 17 00:00:00 2001 From: dushitaoyuan Date: Sat, 31 Aug 2019 21:51:11 +0800 Subject: [PATCH] =?UTF-8?q?=E9=99=90=E6=B5=81=E5=BC=82=E5=B8=B8=E6=94=B9?= =?UTF-8?q?=E5=90=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../{RadioLimitException.java => LimitException.java} | 4 ++-- .../securitydemo/exception/SystemExceptionHandler.java | 3 +-- .../com/taoyuanx/securitydemo/security/RateLimitAspect.java | 4 ++-- 3 files changed, 5 insertions(+), 6 deletions(-) rename src/main/java/com/taoyuanx/securitydemo/exception/{RadioLimitException.java => LimitException.java} (61%) diff --git a/src/main/java/com/taoyuanx/securitydemo/exception/RadioLimitException.java b/src/main/java/com/taoyuanx/securitydemo/exception/LimitException.java similarity index 61% rename from src/main/java/com/taoyuanx/securitydemo/exception/RadioLimitException.java rename to src/main/java/com/taoyuanx/securitydemo/exception/LimitException.java index a245a77..cfb6fee 100644 --- a/src/main/java/com/taoyuanx/securitydemo/exception/RadioLimitException.java +++ b/src/main/java/com/taoyuanx/securitydemo/exception/LimitException.java @@ -5,8 +5,8 @@ package com.taoyuanx.securitydemo.exception; * @desc 限制访问异常 http状态码 429 * @date 2019/8/26 */ -public class RadioLimitException extends RuntimeException { - public RadioLimitException(String message) { +public class LimitException extends RuntimeException { + public LimitException(String message) { super(message); } } diff --git a/src/main/java/com/taoyuanx/securitydemo/exception/SystemExceptionHandler.java b/src/main/java/com/taoyuanx/securitydemo/exception/SystemExceptionHandler.java index bfdee17..9ad1979 100644 --- a/src/main/java/com/taoyuanx/securitydemo/exception/SystemExceptionHandler.java +++ b/src/main/java/com/taoyuanx/securitydemo/exception/SystemExceptionHandler.java @@ -8,7 +8,6 @@ import com.taoyuanx.securitydemo.utils.ResponseUtil; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.http.HttpStatus; -import org.springframework.stereotype.Component; import org.springframework.web.HttpMediaTypeNotSupportedException; import org.springframework.web.method.HandlerMethod; import org.springframework.web.servlet.HandlerExceptionResolver; @@ -54,7 +53,7 @@ public class SystemExceptionHandler implements HandlerExceptionResolver { } else if (e instanceof HttpMediaTypeNotSupportedException) { HttpMediaTypeNotSupportedException mediaEx = (HttpMediaTypeNotSupportedException) e; result = ResultBuilder.failed(ResultCode.UN_SUPPORT_MEDIATYPE.code, "不支持该媒体类型:" + mediaEx.getContentType()); - } else if (e instanceof RadioLimitException) { + } else if (e instanceof LimitException) { httpStatus = HttpStatus.TOO_MANY_REQUESTS; result = ResultBuilder.failed(ResultCode.TOO_MANY_REQUESTS.code, e.getMessage()); } else if (e instanceof JSONException) { diff --git a/src/main/java/com/taoyuanx/securitydemo/security/RateLimitAspect.java b/src/main/java/com/taoyuanx/securitydemo/security/RateLimitAspect.java index 3635f15..055665b 100644 --- a/src/main/java/com/taoyuanx/securitydemo/security/RateLimitAspect.java +++ b/src/main/java/com/taoyuanx/securitydemo/security/RateLimitAspect.java @@ -3,7 +3,7 @@ package com.taoyuanx.securitydemo.security; import com.google.common.collect.Maps; import com.google.common.util.concurrent.RateLimiter; -import com.taoyuanx.securitydemo.exception.RadioLimitException; +import com.taoyuanx.securitydemo.exception.LimitException; import com.taoyuanx.securitydemo.utils.RequestUtil; import org.aspectj.lang.ProceedingJoinPoint; import org.aspectj.lang.Signature; @@ -63,7 +63,7 @@ public class RateLimitAspect { private void handleRateLimit(RateLimit rateLimit, String methodName) throws Throwable { RateLimiter rateLimiter = doGetRateLimiter(rateLimit, methodName); if (!rateLimiter.tryAcquire()) { - throw new RadioLimitException("请求过于频繁,请稍后再试"); + throw new LimitException("请求过于频繁,请稍后再试"); } }