package com.ruoyi.common.config;
|
|
import com.ruoyi.common.annotation.NotRepeatCommit;
|
import com.ruoyi.common.filter.RepeatedlyRequestWrapper;
|
import com.ruoyi.common.utils.http.HttpHelper;
|
import org.apache.logging.log4j.util.Strings;
|
import org.aspectj.lang.ProceedingJoinPoint;
|
import org.aspectj.lang.annotation.Around;
|
import org.aspectj.lang.annotation.Aspect;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.redis.core.RedisTemplate;
|
import org.springframework.stereotype.Component;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.util.concurrent.TimeUnit;
|
|
@Aspect
|
@Component
|
public class IsAspectAspect {
|
@Autowired
|
private RedisTemplate redisTemplate;
|
|
@Around("@annotation(notRepeat)")
|
public Object before(ProceedingJoinPoint point, NotRepeatCommit notRepeat) {
|
String nowParams = "";
|
|
String key = getKey(notRepeat.key(), point.getArgs());
|
if (Strings.isNotBlank(key)) {
|
// 不存在 就添加 缓存 设置 时间
|
if (!redisTemplate.opsForValue().setIfAbsent(key, "1", notRepeat.value(), TimeUnit.MILLISECONDS)) {
|
throw new RuntimeException("请勿重复提交");
|
}
|
}
|
|
try {
|
return point.proceed();
|
} catch (Throwable throwable) {
|
throw new RuntimeException("服务器异常");
|
}
|
}
|
|
/**
|
* key 生成策略
|
*
|
* @param key key表达式
|
* @param args 参数
|
* @return 生成的key
|
*/
|
private String getKey(String key, Object[] args) {
|
for (int i = 0; i < args.length; i++) {
|
key = key.replace("arg[" + i + "]", args[i].toString());
|
}
|
return key;
|
}
|
}
|
