今天用AI搞一個防止接口重復提交注解

作者：了不起 2023-06-05 08:22:20

人工智能

接口冪等性是指無論調用接口的次數是一次還是多次，對于同一資源的操作都只會產生一次結果。換句話說，多次重復調用相同的接口請求應該具有與單次請求相同的效果，不會導致不一致或副作用的發生。

哈嘍，大家好，我是了不起。

今天我們使用AI幫我們去創建一個自定義注解，可以防止接口30秒內的重復請求，并采用Redis作為緩存。

提問

話不多說，直接提問：

等待數分鐘后。。。

1.創建自定義注解其中包括接口保護時長，開啟防止重復提交保護等。

2.然后創建攔截器

這里我們貼出攔截器的核心代碼：

@Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        if (handler instanceof HandlerMethod) {
            HandlerMethod handlerMethod = (HandlerMethod) handler;
            RepeatSubmit annotation = handlerMethod.getMethodAnnotation(RepeatSubmit.class);
            if (annotation != null && annotation.enable()) {
                String key = buildKey(request);
                if (StringUtils.hasText(redisTemplate.opsForValue().get(key))) {
                    response.getWriter().write("repeat request, please try again later！");
                    return false;
                } else {
                    redisTemplate.opsForValue().set(key, Arrays.toString(request.getInputStream().readAllBytes()), annotation.timeout(), TimeUnit.SECONDS);
                }
            }
        }
        return true;
    }
 //創建redis 緩存key
    private String buildKey(HttpServletRequest request) throws IOException, NoSuchAlgorithmException {
        String key = useRequestMD5 ? hashRequest(request) : request.getRequestURI();
        return "repeat-submit:" + key;
    }

 //對請求做hash運算
    private String hashRequest(HttpServletRequest request) throws IOException, NoSuchAlgorithmException {
        byte[] hashBytes = MessageDigest.getInstance("MD5").digest(request.getInputStream().readAllBytes());
        StringBuilder sb = new StringBuilder();
        for (byte b : hashBytes) {
            sb.append(String.format("%02x", b));
        }
        return sb.toString();
    }

3.注冊攔截器

最后給出的解釋與使用方法。

上面就是最關鍵的代碼了。

接入Redis

下面我們接入Redis。最精簡的配置版本：

spring:
  data:
    redis:
      host: 127.0.0.1 
      port: 6379

接口使用注解

@RestController
public class RepeatTestController {

    @RepeatSubmit
    @GetMapping("/hello/mono1")
    public Mono<String> mono(){
        return Mono.just("Hello Mono -  Java North");
    }

    @RepeatSubmit
    @PostMapping ("/hello/mono1")
    public Mono<String> mono1(@RequestBody User user){
        return Mono.just("Hello Mono -  Java North-"+user.getName());
    }
}

本地起一個Redis，然后啟動本地的SpringBoot項目進行測試。