一种轻量级单体springboot防重复提交的解决方案

一种轻量级单体springboot防重复提交的解决方案

物料准备

pom添加依赖

<!-- 接口重复提交控制 -->
			<dependency>
				<groupId>net.jodah</groupId>
				<artifactId>expiringmap</artifactId>
				<version>0.5.10</version>
			</dependency>

定义1个java注解

package com.xxx.xxx.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface AvoidRepeatableCommit {

    /**
     * 指定时间内不可重复提交，单位毫秒
     */
    long timeout() default 3000;
}

定义注解对应的切面

package com.xxx.xxx.annotation;

import cn.hutool.core.util.StrUtil;
import com.xxx.xxx.exception.BusinessException;
import net.jodah.expiringmap.ExpirationPolicy;
import net.jodah.expiringmap.ExpiringMap;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.springframework.web.util.WebUtils;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

@Aspect
@Component
public class AvoidRepeatableCommitAspect {

    private static final long TIMEOUT = 3000L;

    private static final ExpiringMap<String, String> expiringMap = ExpiringMap.builder()
            .variableExpiration() // 可变有效期，即单独为每个entity设置过期时间和策略
            .expirationPolicy(ExpirationPolicy.CREATED) // 过期策略：CREATED：在每次更新元素时，过期倒计时清零
            .expiration(TIMEOUT, TimeUnit.MILLISECONDS) // 过期时间
            .maxSize(1000) // Map中映射数目超过最大值的大小时，先过期第一个要过期的entity
            .build();

    @Around("@annotation(com.xxx.xxx.annotation.AvoidRepeatableCommit)")
    public Object around(ProceedingJoinPoint point) throws Throwable {

        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();

        Long account = (Long) WebUtils.getSessionAttribute(request, "cur_userId");
        if (account == null) {
            return point.proceed();
        }
        StringBuffer requestURL = request.getRequestURL();
        String reqMethod = request.getMethod();
        String key = String.format("%s_%s_%s", account, reqMethod, requestURL);

        // 获取注解
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        AvoidRepeatableCommit avoidRepeatableCommit = method.getAnnotation(AvoidRepeatableCommit.class);
        long timeout = avoidRepeatableCommit.timeout();
        if (timeout < 0) {
            timeout = TIMEOUT;
        }
        if (StrUtil.isNotBlank(expiringMap.get(key))) {
            throw new BusinessException("请勿重复提交");
        }
        expiringMap.put(key, UUID.randomUUID().toString(), ExpirationPolicy.CREATED, timeout, TimeUnit.MILLISECONDS);

        //执行方法
        return point.proceed();
    }

}

使用案例

可以在你自定义的Controller类或RestController类里的@RequestMapping 对应的方法上加这个注解

@AvoidRepeatableCommit(timeout = 1000L)

这样被注解标注的api接口就会在请求时被切面处理到，并判断前端是否重复提交了。