SpringBoot防止表單重複提交AOP




import java.util.Objects;
import javax.servlet.http.HttpServletRequest;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

@Aspect
@Component
public class NoRepeatSubmitAop {
    Logger logger = LoggerFactory.getLogger(this.getClass());
    int time = 3000;
    @Autowired
    private RedisUtil redisUtil;

    public NoRepeatSubmitAop() {
    }

    @Around("execution(* com.hl..controller.*.*(..))")
    public Object doAround(ProceedingJoinPoint pjp) throws Throwable {
        HttpServletRequest request = ((ServletRequestAttributes)Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
        String method = request.getMethod();
        if (!method.equals("POST") && !method.equals("PUT")) {
            Object proceed = pjp.proceed();
            return proceed;
        } else {
            String ip = BaseUtil.getIpAddress(request);
            String url = request.getRequestURL().toString();
            long now = System.currentTimeMillis();
            String key = "REQUEST_FORM_" + ip;
            if (this.redisUtil.hasKey(key)) {
                long lastTime = ConvertUtil.parseLong(this.redisUtil.get(key));
                if (now - lastTime > (long)this.time) {
                    this.redisUtil.set(key, String.valueOf(now));
                    Object proceed = pjp.proceed();
                    return proceed;
                } else {
                    throw new Exception("你提交的太頻繁了,請稍後再試");
                }
            } else {
                this.redisUtil.set(key, String.valueOf(now));
                Object proceed = pjp.proceed();
                return proceed;
            }
        }
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章