自定义注解
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface SubmitLock {
String key() default "";
}
设置缓存 可以用rides 代替
@Configuration
public class UrlCache {
@Bean
public Cache<String, String> getCache() {
return CacheBuilder.newBuilder()
// 最大缓存 1000 个,请结合业务需求和内存大小设置
.maximumSize(1000)
// 设置缓存有效期 5 秒钟
.expireAfterWrite(5, TimeUnit.SECONDS)
.build();
}
}
面象切面处理
@Aspect
@Component
public class NoRepeatSubmitAop {
private static Logger logger = LoggerFactory.getLogger(NoRepeatSubmitAop.class);
@Autowired
private Cache<String, String> CACHES;
@Pointcut("@annotation(submitLock)")
public void pointCut(SubmitLock submitLock) {
}
@Around("pointCut(lock)")
public Object submitInterceptor(ProceedingJoinPoint pjp, SubmitLock lock) {
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
HttpServletRequest request = attributes.getRequest();
// String url = request.getRequestURL().toString();// 请求url
String uri = request.getRequestURI();// 请求uri
// String queryString = request.getQueryString();// get请求的查询参数
MethodSignature signature = (MethodSignature) pjp.getSignature();
Method method = signature.getMethod();
String targetName = pjp.getTarget().getClass().getName();// 真实类名字
String methodName = pjp.getSignature().getName();// 真实方式
Object[] arguments = pjp.getArgs();// 所有请求参数
Object[] args = new Object[arguments.length];
SubmitLock localLock = method.getAnnotation(SubmitLock.class);
String key = setKey(localLock.key(), pjp.getArgs());
if (!StringUtils.isEmpty(key)) {
if (CACHES.getIfPresent(key) != null) {
logger.error("请勿重复请求,uri =【{}】", uri);
return Result.fail("请勿重复请求");
}
// 如果是第一次请求,就将 key,即当前对象存入缓存
CACHES.put(key, key);
}
Object result = null;
try {
result = pjp.proceed();
return result;
} catch (Throwable throwable) {
throw new RuntimeException("服务器异常");
} finally {
int order = 0;
for (Object arg : arguments) {
if (arg instanceof ServletRequest || arg instanceof ServletResponse || arg instanceof MultipartFile) {
//ServletRequest不能序列化,从入参里排除,否则报异常:java.lang.IllegalStateException: It is illegal to call this method if the current request is not in asynchronous mode (i.e. isAsyncStarted() returns false)
//ServletResponse亦能序列化 从入参里排除,否则报异常:java.lang.IllegalStateException: getOutputStream() has already been called for this response
continue;
}
args[order] = arg;
order ++;
}
// 使用 AOP 实现统一记录请求方法返回值日志,当然,也可以存入数据库
logger.info("调用Controller方法返回结果,targetName = {}, methodName = {}, args = {}, result = {}",
targetName, methodName, args, result);
}
}
/**
* key 的生成策略,如果想灵活可以写成接口与实现类的方式(TODO 后续讲解)
*
* @param keyExpress 表达式
* @param args 参数
* @return 生成的key
*/
private String setKey(String keyExpress, Object[] args) {
if (null != args && args.length > 0) {
keyExpress = keyExpress.replace("arg[0]", args[0].toString());
}
return keyExpress;
}
}
实例
@SubmitLock(key = "saveOrUpdateBaiDuMapApi:arg[0]")
@ApiOperation(value = "保存或修改百度地图api", notes = "保存或修改百度地图api")
@ApiImplicitParam(name = "BaiDuMapApi", value = "保存或修改百度地图api", required = true, dataType = "baiDuMapApi")
@RequestMapping(value = "/saveOrUpdateBaiDuMapApi", method = RequestMethod.POST)