java对同一个请求防重

原创

mob649e815e258d 2024-08-01 13:40:45 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e815e258d的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java对同一个请求防重的实现

在Web应用中，重复提交请求是一个常见的问题。这不仅会导致数据的不一致性，还可能引发一系列的性能问题和安全隐患。为了防止重复请求，Java提供了多种策略。在本文中，我们将探讨如何通过不同的方法在Java中实现请求防重，重点展示一种简单有效的实现。

为什么需要防止重复请求？

重复请求通常是由以下原因引起的：

用户的操作：用户不小心点击了“提交”按钮多次。
网络问题：网络延迟可能导致用户认为请求没有被发送，于是再次提交。
脚本或Bots：恶意用户可能会通过脚本重复提交请求。

这些问题会导致不必要的资源消耗，甚至破坏数据的完整性。

防重方法概述

Java中常用的防重方法包括：

客户端防重：通过前端逻辑控制。
Token机制：使用token来标识每次请求。
Session机制：在Session中存储请求状态。
数据库查重：在数据库中记录请求状态。

本文将重点介绍Token机制和数据库查重。

Token机制实现

这种机制的基本思想是：每当用户发起请求时，生成一个唯一的token，并在一定时间内只允许该token的请求达到处理逻辑。下面是一个简单的示例：

import org.springframework.web.bind.annotation.*;
import org.springframework.http.ResponseEntity;

import java.util.concurrent.ConcurrentHashMap;
import java.util.UUID;

@RestController
@RequestMapping("/api")
public class RequestController {

    private static final ConcurrentHashMap<String, String> tokenStore = new ConcurrentHashMap<>();

    @PostMapping("/submit")
    public ResponseEntity<String> handleRequest(@RequestParam String token) {
        if (tokenStore.containsKey(token)) {
            return ResponseEntity.badRequest().body("Duplicate request");
        }
        tokenStore.put(token, "processed");  // 标记该token已处理
        // 处理请求逻辑
        return ResponseEntity.ok("Request processed");
    }

    // 清理过期token，可以定时调用
    public void cleanupTokens() {
        // 实现token过期清理逻辑
    }
}

状态图

请求的状态可以通过以下状态图表示：

stateDiagram
    [*] --> Idle
    Idle --> Processing : 请求到达
    Processing --> Processed : 请求处理完成
    Processing --> Duplicate : 重复请求
    Duplicate --> Idle : 等待新的请求
    Processed --> Idle : 等待新的请求

数据库查重实现

另一种防重方式是利用数据库。在处理请求时，先查询数据库，判断先前是否有相同的请求。如果有，则拒绝；如果没有，就插入一条记录并处理请求。这种方法的实现如下：

import org.springframework.web.bind.annotation.*;
import org.springframework.http.ResponseEntity;
import org.springframework.beans.factory.annotation.Autowired;

@RestController
@RequestMapping("/api")
public class DatabaseRequestController {

    @Autowired
    private RequestRepository requestRepository;

    @PostMapping("/submit")
    public ResponseEntity<String> handleRequest(@RequestParam String uniqueId) {
        if (requestRepository.existsById(uniqueId)) {
            return ResponseEntity.badRequest().body("Duplicate request");
        }
        requestRepository.save(new Request(uniqueId)); // 保存新请求
        // 处理请求逻辑
        return ResponseEntity.ok("Request processed");
    }
}

数据库表结构示例

以下是建议的数据库表结构：

| 字段        | 类型      | 描述                  |
|------------|-----------|-----------------------|
| id         | VARCHAR   | 请求唯一标识          |
| created_at | TIMESTAMP | 请求创建时间          |