文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

一个注解,两种实现方式完美解决重复提交问题

2024-11-30 10:05

关注

什么是接口防重

接口防重是指在一定时间内只允许执行一次接口请求。这是为了防止由于重复提交和重复处理产生重复数据或相应错误。实现接口防重可以采用以下方法:

  1. 使用唯一标识符:在请求中包含一个唯一标识符(例如请求token),然后在对应接口判断该唯一值在一定时间内是否被消费过,如果已被消费,则拒绝该请求。
  2. 使用时间戳、计数器等机制:记录请求的时间或次数,并在一定范围内拒绝重复请求。
  3. 采用Spring AOP理念:实现请求的切割,在请求执行到某个方法或某层时,开始拦截并进行防重处理。

这些方法有助于确保系统的一致性和稳定性,防止数据的重复提交和处理。

幂等与防重

API接口的幂等性和防重性是两个不同的概念,尽管它们在某些方面有重叠之处。

在实现幂等性时,通常采用以下方法:

技术实现

方式1:通过AOP方式

自定义注解

@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface PreventDuplicate {
  
  
  String header() default "token" ;
  
  
  String param() default "token" ;
}

自定义AOP切面

@Component
@Aspect
public class PreventDuplicateAspect {


  public static final String PREVENT_PREFIX_KEY = "prevent:" ;
  
  private final StringRedisTemplate stringRedisTemplate ;
  private final HttpServletRequest request ;
  
  public PreventDuplicateAspect(StringRedisTemplate stringRedisTemplate, HttpServletRequest request) {
    this.stringRedisTemplate = stringRedisTemplate ;
    this.request = request ;
  }
  
  @Around("@annotation(prevent)")
  public Object preventDuplicate(ProceedingJoinPoint pjp, PreventDuplicate prevent) throws Throwable {
    
    String key = prevent.header() ;
    String value = null ;
    if (key != null && key.length() > 0) {
      value = this.request.getHeader(key) ;
    } else {
      key = prevent.param() ;
      if (key != null && key.length() > 0) {
        value = this.request.getParameter(key) ;
      }
    }
    
    if (value == null || "".equals(value.trim())) {
      return "非法请求" ;
    }


    // 拼接rediskey
    String prevent_key = PREVENT_PREFIX_KEY + value ;
    // 判断redis中是否存在当前请求中携带的唯一标识数据, 删除成功则存在
    Boolean result = this.stringRedisTemplate.delete(prevent_key) ;
    if (result != null && result.booleanValue()) {
      return pjp.proceed() ;
    } else {
      return "请不要重复提交" ;
    }
  }
  
}

生成唯一标识接口

@RestController
@RequestMapping("/generate")
public class GenerateController {


  private final StringRedisTemplate stringRedisTemplate ;
  public GenerateController(StringRedisTemplate stringRedisTemplate) {
    this.stringRedisTemplate = stringRedisTemplate ;
  }
  
  @GetMapping("/token")
  public String token() {
    String token = UUID.randomUUID().toString().replace("-", "") ;
    // 将生成的token存入redis中,设置有效期5分钟
    this.stringRedisTemplate.opsForValue().setIfAbsent(PreventDuplicateAspect.PREVENT_PREFIX_KEY + token, token, 5 * 60, TimeUnit.SECONDS) ;
    return token ;
  }
  
}

业务接口

@RestController
@RequestMapping("/prevent")
public class PreventController {


  @PreventDuplicate
  @GetMapping("/index")
  public Object index() {
    return "index success" ;
  }
  
}

测试

先调用生成唯一接口获取token值

图片

调用业务接口,携带token值

第一次访问, 正常

再次访问

方式2:通过拦截器实现

自定义拦截器

@Component
public class PreventDuplicateInterceptor implements HandlerInterceptor {


  private final StringRedisTemplate stringRedisTemplate ;
  public PreventDuplicateInterceptor(StringRedisTemplate stringRedisTemplate) {
    this.stringRedisTemplate = stringRedisTemplate ;
  }
  
  @Override
  public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
    if (handler instanceof HandlerMethod hm) {
      if (hm.hasMethodAnnotation(PreventDuplicate.class)) {
        PreventDuplicate pd = hm.getMethodAnnotation(PreventDuplicate.class) ;
        
        String key = pd.header() ;
        String value = null ;
        if (key != null && key.length() > 0) {
          value = request.getHeader(key) ;
        } else {
          key = pd.param() ;
          if (key != null && key.length() > 0) {
            value = request.getParameter(key) ;
          }
        }
        
        if (value == null || "".equals(value.trim())) {
          response.setContentType("text/plain;charset=utf-8") ;
          response.getWriter().println("非法请求") ;
          return false ;
        }
        
        // 拼接rediskey
        String prevent_key = PreventDuplicateAspect.PREVENT_PREFIX_KEY + value ;
        // 判断redis中是否存在当前请求中携带的唯一标识数据, 删除成功则存在
        Boolean result = this.stringRedisTemplate.delete(prevent_key) ;
        if (result != null && result.booleanValue()) {
          return true ;
        } else {
          response.setContentType("text/plain;charset=utf-8") ;
          response.getWriter().println("请不要重复提交") ;
          return false ;
        }
      }
    }
    return true ;
  }
  
}

配置拦截器

@Component
public class PreventWebConfig implements WebMvcConfigurer {


  private final PreventDuplicateInterceptor duplicateInterceptor ;
  public PreventWebConfig(PreventDuplicateInterceptor duplicateInterceptor) {
    this.duplicateInterceptor = duplicateInterceptor ;
  }
  
  @Override
  public void addInterceptors(InterceptorRegistry registry) {
    registry.addInterceptor(this.duplicateInterceptor).addPathPatterns("/**") ;
  }
  
}

测试

获取token

第一次请求

再次请求

完毕!!!

来源:Spring全家桶实战案例源码内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯