Parker
5 years ago
parent
62d212a9e9
commit
2ec4ef2d67
@ -0,0 +1,24 @@
|
|||||||
|
package org.opsli.common.annotation.limiter;
|
||||||
|
|
||||||
|
|
||||||
|
import org.opsli.common.enums.AlertType;
|
||||||
|
import org.opsli.common.utils.RateLimiterUtil;
|
||||||
|
|
||||||
|
import java.lang.annotation.*;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Java 限流器
|
||||||
|
* @author Parker
|
||||||
|
*/
|
||||||
|
@Target(ElementType.METHOD)
|
||||||
|
@Retention(RetentionPolicy.RUNTIME)
|
||||||
|
@Documented
|
||||||
|
public @interface Limiter {
|
||||||
|
|
||||||
|
/** 标题 */
|
||||||
|
double qps() default RateLimiterUtil.DEFAULT_QPS;
|
||||||
|
|
||||||
|
/** 提醒方式 */
|
||||||
|
AlertType alertType() default AlertType.JSON;
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,17 @@
|
|||||||
|
package org.opsli.common.enums;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @Author: Parker
|
||||||
|
* @CreateTime: 2020-09-17 23:40
|
||||||
|
* @Description: 字典
|
||||||
|
*/
|
||||||
|
public enum AlertType {
|
||||||
|
|
||||||
|
/** alert 弹出 */
|
||||||
|
ALERT(),
|
||||||
|
|
||||||
|
/** JSON 回推*/
|
||||||
|
JSON,
|
||||||
|
;
|
||||||
|
}
|
||||||
@ -0,0 +1,62 @@
|
|||||||
|
package org.opsli.common.utils;
|
||||||
|
|
||||||
|
import cn.hutool.core.io.IoUtil;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.opsli.common.exception.ServiceException;
|
||||||
|
import org.opsli.common.msg.CommonMsg;
|
||||||
|
|
||||||
|
import javax.servlet.http.HttpServletResponse;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.OutputStream;
|
||||||
|
import java.io.PrintWriter;
|
||||||
|
import java.nio.charset.StandardCharsets;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @BelongsProject: think-bboss-parent
|
||||||
|
* @BelongsPackage: com.think.bboss.common.utils
|
||||||
|
* @Author: Parker
|
||||||
|
* @CreateTime: 2021-01-05 14:26
|
||||||
|
* @Description: 周鹏程
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
public final class OutputStreamUtil {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 导出文件时为Writer生成OutputStream
|
||||||
|
*/
|
||||||
|
public static OutputStream getOutputStream(String fileName, HttpServletResponse response)
|
||||||
|
throws ServiceException {
|
||||||
|
try {
|
||||||
|
fileName = new String(fileName.getBytes(), StandardCharsets.ISO_8859_1);
|
||||||
|
response.addHeader("Content-Disposition", "attachment; filename=" + fileName);
|
||||||
|
return response.getOutputStream();
|
||||||
|
} catch (IOException e) {
|
||||||
|
// 创建文件失败
|
||||||
|
throw new ServiceException(CommonMsg.EXCEPTION_CREATE_FILE_ERROR);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 返回异常值
|
||||||
|
*/
|
||||||
|
public static void exceptionResponse(String msg, HttpServletResponse response){
|
||||||
|
try {
|
||||||
|
response.setCharacterEncoding("utf-8");
|
||||||
|
response.setContentType("text/html;charset=utf-8;");
|
||||||
|
PrintWriter writer = response.getWriter();
|
||||||
|
writer.write(
|
||||||
|
"<script type=\"text/javascript\">alert('"+msg+"');</script>");
|
||||||
|
writer.flush();
|
||||||
|
// 关闭流
|
||||||
|
IoUtil.close(writer);
|
||||||
|
}catch (Exception e){
|
||||||
|
log.error(e.getMessage(), e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// ==========================
|
||||||
|
|
||||||
|
private OutputStreamUtil(){}
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,101 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2020 OPSLI 快速开发平台 https://www.opsli.com
|
||||||
|
* <p>
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
||||||
|
* use this file except in compliance with the License. You may obtain a copy of
|
||||||
|
* the License at
|
||||||
|
* <p>
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* <p>
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
* License for the specific language governing permissions and limitations under
|
||||||
|
* the License.
|
||||||
|
*/
|
||||||
|
package org.opsli.core.aspect;
|
||||||
|
|
||||||
|
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.aspectj.lang.JoinPoint;
|
||||||
|
import org.aspectj.lang.annotation.Aspect;
|
||||||
|
import org.aspectj.lang.annotation.Before;
|
||||||
|
import org.aspectj.lang.annotation.Pointcut;
|
||||||
|
import org.aspectj.lang.reflect.MethodSignature;
|
||||||
|
import org.opsli.common.annotation.limiter.Limiter;
|
||||||
|
import org.opsli.common.enums.AlertType;
|
||||||
|
import org.opsli.common.exception.ServiceException;
|
||||||
|
import org.opsli.common.utils.OutputStreamUtil;
|
||||||
|
import org.opsli.common.utils.RateLimiterUtil;
|
||||||
|
import org.opsli.core.msg.CoreMsg;
|
||||||
|
import org.springframework.core.annotation.Order;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
import org.springframework.web.context.request.RequestAttributes;
|
||||||
|
import org.springframework.web.context.request.RequestContextHolder;
|
||||||
|
import org.springframework.web.context.request.ServletRequestAttributes;
|
||||||
|
|
||||||
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
import javax.servlet.http.HttpServletResponse;
|
||||||
|
import java.lang.reflect.Method;
|
||||||
|
|
||||||
|
import static org.opsli.common.constants.OrderConstants.LIMITER_AOP_SORT;
|
||||||
|
/**
|
||||||
|
* 限流器
|
||||||
|
*
|
||||||
|
* @author 周鹏程
|
||||||
|
* @date 2020-09-16
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
@Order(LIMITER_AOP_SORT)
|
||||||
|
@Aspect
|
||||||
|
@Component
|
||||||
|
public class LimiterAop {
|
||||||
|
|
||||||
|
|
||||||
|
@Pointcut("@annotation(org.opsli.common.annotation.limiter.Limiter)")
|
||||||
|
public void requestMapping() {
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 限流
|
||||||
|
* @param point
|
||||||
|
*/
|
||||||
|
@Before("requestMapping()")
|
||||||
|
public void limiterHandle(JoinPoint point){
|
||||||
|
try {
|
||||||
|
RequestAttributes ra = RequestContextHolder.getRequestAttributes();
|
||||||
|
ServletRequestAttributes sra = (ServletRequestAttributes) ra;
|
||||||
|
MethodSignature signature = (MethodSignature) point.getSignature();
|
||||||
|
Method method = signature.getMethod();
|
||||||
|
if(sra != null) {
|
||||||
|
HttpServletRequest request = sra.getRequest();
|
||||||
|
HttpServletResponse response = sra.getResponse();
|
||||||
|
Limiter limiter = method.getAnnotation(Limiter.class);
|
||||||
|
if(limiter != null){
|
||||||
|
AlertType alertType = limiter.alertType();
|
||||||
|
double qps = limiter.qps();
|
||||||
|
|
||||||
|
// 限流
|
||||||
|
boolean enterFlag = RateLimiterUtil.enter(request, qps);
|
||||||
|
if(!enterFlag){
|
||||||
|
// alert 弹出
|
||||||
|
if(AlertType.ALERT == alertType){
|
||||||
|
OutputStreamUtil.exceptionResponse(
|
||||||
|
CoreMsg.OTHER_EXCEPTION_LIMITER.getMessage(),
|
||||||
|
response
|
||||||
|
);
|
||||||
|
}else {
|
||||||
|
// 异常返回
|
||||||
|
throw new ServiceException(CoreMsg.OTHER_EXCEPTION_LIMITER);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}catch (ServiceException e){
|
||||||
|
throw e;
|
||||||
|
}catch (Exception e){
|
||||||
|
log.error(e.getMessage(),e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Reference in new issue