背景#
同一条数据被用户点击了多次，导致数据冗余，需要防止弱网络等环境下的重复点击
目标#
通过在指定的接口处添加注解，实现根据指定的接口参数来防重复点击
说明#
这里的重复点击是指在指定的时间段内多次点击按钮
技术方案#
springboot + redis锁 + 注解
使用 feign client 进行请求测试
最终的使用实例#
1、根据接口收到 PathVariable 参数判断唯一
Copy/*** 根据请求参数里的 PathVariable 里获取的变量进行接口级别防重复点击** @param testId 测试id* @param requestVo 请求参数* @return* @author daleyzou*/ @PostMapping("/test/{testId}") @NoRepeatSubmit(location = "thisIsTestLocation", seconds = 6) public RsVo thisIsTestLocation(@PathVariable Integer testId, @RequestBody RequestVo requestVo) throws Throwable { // 睡眠 5 秒，模拟业务逻辑 Thread.sleep(5); return RsVo.success("test is return success"); }
2、根据接口收到的 RequestBody 中指定变量名的值判断为一
Copy/*** 根据请求参数里的 RequestBody 里获取指定名称的变量param5的值进行接口级别防重复点击** @param testId 测试id* @param requestVo 请求参数* @return* @author daleyzou*/ @PostMapping("/test/{testId}") @NoRepeatSubmit(location = "thisIsTestBody", seconds = 6, argIndex = 1, name = "param5") public RsVo thisIsTestBody(@PathVariable Integer testId, @RequestBody RequestVo requestVo) throws Throwable { // 睡眠 5 秒，模拟业务逻辑 Thread.sleep(5); return RsVo.success("test is return success"); }
ps: jedis 2.9 和 springboot有各种兼容问题，无奈只有降低springboot的版本了
运行结果#
Copy收到响应：{"succeeded":true,"code":500,"msg":"操作过于频繁，请稍后重试","data":null}收到响应：{"succeeded":true,"code":500,"msg":"操作过于频繁，请稍后重试","data":null}收到响应：{"succeeded":true,"code":500,"msg":"操作过于频繁，请稍后重试","data":null}收到响应：{"succeeded":true,"code":200,"msg":"success","data":"test is return success"}
测试用例#
Copypackage com.dalelyzou.preventrepeatsubmit.controller;import com.dalelyzou.preventrepeatsubmit.PreventrepeatsubmitApplicationTests;import com.dalelyzou.preventrepeatsubmit.service.AsyncFeginService;import com.dalelyzou.preventrepeatsubmit.vo.RequestVo;import org.junit.jupiter.api.Test;import org.springframework.beans.factory.annotation.Autowired;import java.io.IOException;import java.util.concurrent.ExecutorService;import java.util.concurrent.Executors;/** * TestControllerTest * @description 防重复点击测试类 * @author daleyzou * @date 09月28日 17:13 * @version 1.3.1 */class TestControllerTest extends PreventrepeatsubmitApplicationTests { @Autowired AsyncFeginService asyncFeginService; @Test public void thisIsTestLocation() throws IOException { RequestVo requestVo = new RequestVo(); requestVo.setParam5("random"); ExecutorService executorService = Executors.newFixedThreadPool(4); for (int i = 0; i <= 3; i++) { executorService.execute(() -> {String kl = asyncFeginService.thisIsTestLocation(requestVo);System.err.println("收到响应：" + kl); }); } System.in.read(); } @Test public void thisIsTestBody() throws IOException { RequestVo requestVo = new RequestVo(); requestVo.setParam5("special"); ExecutorService executorService = Executors.newFixedThreadPool(4); for (int i = 0; i <= 3; i++) { executorService.execute(() -> {String kl = asyncFeginService.thisIsTestBody(requestVo);System.err.println("收到响应：" + kl); }); } System.in.read(); }}
定义一个注解#
Copypackage com.dalelyzou.preventrepeatsubmit.aspect;import java.lang.annotation.ElementType;import java.lang.annotation.Retention;import java.lang.annotation.RetentionPolicy;import java.lang.annotation.Target;/** * NoRepeatSubmit * @description 重复点击的切面 * @author daleyzou * @date 09月23日 14:35 * @version 1.4.8 */@Target(ElementType.METHOD)@Retention(RetentionPolicy.RUNTIME)public @interface NoRepeatSubmit { /*** 锁过期的时间* */ int seconds() default 5; /*** 锁的位置* */ String location() default "NoRepeatSubmit"; /*** 要扫描的参数位置* */ int argIndex() default 0; /*** 参数名称* */ String name() default "";}
根据指定的注解定义一个切面，根据参数中的指定值来判断请求是否重复#
Copypackage com.dalelyzou.preventrepeatsubmit.aspect;import com.dalelyzou.preventrepeatsubmit.constant.RedisKey;import com.dalelyzou.preventrepeatsubmit.service.LockService;import com.dalelyzou.preventrepeatsubmit.vo.RsVo;import mon.collect.Maps;import com.google.gson.Gson;import org.aspectj.lang.ProceedingJoinPoint;import org.aspectj.lang.annotation.Around;import org.aspectj.lang.annotation.Aspect;import org.aspectj.lang.annotation.Pointcut;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.beans.factory.annotation.Autowired;import org.ponent;import org.springframework.util.StringUtils;import java.lang.reflect.Field;import java.util.Map;@Aspect@Componentpublic class NoRepeatSubmitAspect { private static final Logger logger = LoggerFactory.getLogger(NoRepeatSubmitAspect.class); private static Gson gson = new Gson(); private static final String SUFFIX = "SUFFIX"; @Autowired LockService lockService; /*** 横切点*/ @Pointcut("@annotation(noRepeatSubmit)") public void repeatPoint(NoRepeatSubmit noRepeatSubmit) { } /*** 接收请求，并记录数据*/ @Around(value = "repeatPoint(noRepeatSubmit)") public Object doBefore(ProceedingJoinPoint joinPoint, NoRepeatSubmit noRepeatSubmit) { String key = RedisKey.NO_REPEAT_LOCK_PREFIX + noRepeatSubmit.location(); Object[] args = joinPoint.getArgs(); String name = noRepeatSubmit.name(); int argIndex = noRepeatSubmit.argIndex(); String suffix; if (StringUtils.isEmpty(name)) { suffix = String.valueOf(args[argIndex]); } else { Map keyAndValue = getKeyAndValue(args[argIndex]); Object valueObj = keyAndValue.get(name); if (valueObj == null) {suffix = SUFFIX; } else {suffix = String.valueOf(valueObj); } } key = key + ":" + suffix; logger.info("=================================================="); for (Object arg : args) { logger.info(gson.toJson(arg)); } logger.info("=================================================="); int seconds = noRepeatSubmit.seconds(); logger.info("lock key : " + key); if (!lockService.isLock(key, seconds)) { return RsVo.fail("操作过于频繁，请稍后重试"); } try { Object proceed = joinPoint.proceed(); return proceed; } catch (Throwable throwable) { logger.error("运行业务代码出错", throwable); throw new RuntimeException(throwable.getMessage()); } finally { lockService.unLock(key); } } public static Map getKeyAndValue(Object obj) { Map map = Maps.newHashMap(); // 得到类对象 Class userCla = (Class) obj.getClass(); /* 得到类中的所有属性集合 */ Field[] fs = userCla.getDeclaredFields(); for (int i = 0; i < fs.length; i++) { Field f = fs[i]; // 设置些属性是可以访问的 f.setAccessible(true); Object val = new Object(); try {val = f.get(obj);// 得到此属性的值// 设置键值map.put(f.getName(), val); } catch (IllegalArgumentException e) {logger.error("getKeyAndValue IllegalArgumentException", e); } catch (IllegalAccessException e) {logger.error("getKeyAndValue IllegalAccessException", e); } } logger.info("扫描结果：" + gson.toJson(map)); return map; }}
完整地址：/daleyzou/PreventRepeatSubmit
原文链接：/daleyzou/p/noSubmitRepeat.html
如果觉得本文对你有帮助，可以转发关注支持一下
