自定义注解实现拦截前端重复访问接口

最新推荐文章于 2024-10-21 19:38:22 发布

原创最新推荐文章于 2024-10-21 19:38:22 发布 · 769 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#java

Java 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种使用AOP和自定义注解防止Web应用中重复提交请求的方法。通过在控制器方法上添加@NoRepeatSubmit注解，并结合AspectJ切面编程，实现对请求的唯一性和时效性控制，有效避免了短时间内多次提交相同请求的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.自定义注解类NoRepeatSubmit


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 NoRepeatSubmit {

    /**
     * 设置请求锁定时间
     *
     * @return
     */
    long lockTime() default 1000;

2.AOP类 RepeatSubmitAspect

import com.agrhino.framework.common.response.Result;
import com.agrhino.framework.redis.util.RedisValueUtils;
import com.agrhino.miaomiao.auth.web.annotation.NoRepeatSubmit;
import com.agrhino.miaomiao.base.util.IPUtils;
import lombok.extern.slf4j.Slf4j;
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.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import javax.servlet.http.HttpServletRequest;
import java.util.concurrent.TimeUnit;

@Aspect
@Component
@Slf4j
public class RepeatSubmitAspect {

    @Autowired
    private RedisValueUtils redisValueUtils;

    @Pointcut("@annotation(noRepeatSubmit)")
    public void pointOut (NoRepeatSubmit noRepeatSubmit) {

    }

    @Around("pointOut(noRepeatSubmit)")
    public Object around(ProceedingJoinPoint pjp, NoRepeatSubmit noRepeatSubmit) throws Throwable {
        long lockSeconds = noRepeatSubmit.lockTime();
        HttpServletRequest request = getRequest();
        // 此处可以用token或者JSessionId
        String token = request.getHeader("Access-Token");
        String path = request.getServletPath();
        String key = getKey(token, path);
        Object vlue = redisValueUtils.get(key);
        if (null==vlue) {
            // 执行进程
            Object o = pjp.proceed();
            redisValueUtils.set(key,key,lockSeconds,TimeUnit.MILLISECONDS);
            return o;

        } else {
            // 获取锁失败，认为是重复提交的请求
            log.warn("tryLock fail, key = [{}],ip =[{}]", key, IPUtils.getIp(request));
            return Result.failed("请稍后再试");
           // throw new ServiceException("重复提交");


        }

    }

    private String getKey(String token, String path) {
        return "NoRepeatSubmit:" + token + "-" + path;
    }

    private  HttpServletRequest getRequest() {
        ServletRequestAttributes ra = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        return ra.getRequest();
    }

}

3.用例@NoRepeatSubmit(lockTime = 1000)

 @ApiOperation(value = "发布动态评论")
    @ApiImplicitParam(paramType = "header", name = "Access-Token", value = "accessToken", required = true)
    @PostMapping("createDynamicComment")
    @NoRepeatSubmit(lockTime = 1000)
    public Result<UserDynamicComment> addDynamicComment(@RequestBody @ApiParam(name = "form", value = "发布动态评论", required = true)
                                                     UserDynamicCommentForm form) {
        if(!inputFilterService.inputCheck(form.getComment())){
            return Result.failed("您的发言包含敏感信息，请修改后再发送");
        }
        UserDynamicComment userDynamicComment = new UserDynamicComment();
        BeanUtils.copyProperties(form, userDynamicComment);

        UserDynamicComment result = userDynamicCommentService.add(userDynamicComment);
        return Result.success(result);

    }