防止表单重复提交(面向切面)

自定义注解

@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)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值