SpringBoot中防止接口重复提交多种方法（代码示例）

最新推荐文章于 2024-06-04 15:38:03 发布

Tank_Chen

最新推荐文章于 2024-06-04 15:38:03 发布

阅读量1.4k

点赞数 36

分类专栏： Spring 文章标签： spring boot 后端 java

本文链接：https://blog.csdn.net/qq_35759769/article/details/139407284

版权

Spring 专栏收录该内容

24 篇文章 0 订阅

订阅专栏

在Spring Boot应用中，防止接口重复提交是一个常见的需求，尤其是在处理订单、支付等关键业务时。

这可以通过在单机环境和分布式环境下采用不同的策略来实现。

单机环境

在单机环境下，可以使用ThreadLocal或者Spring的@SessionAttributes来防止接口重复提交。

1. 使用ThreadLocal

ThreadLocal为每个线程提供了一个单独的变量副本，因此可以用来存储请求级别的数据，以此来判断是否是重复提交。

import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class OrderController {

    private ThreadLocal<String> orderSubmitFlag = new ThreadLocal<>();

    @PostMapping("/order")
    public String submitOrder() {
        // 检查是否已经提交过
        if (orderSubmitFlag.get() != null) {
            return "请勿重复提交";
        }
        
        orderSubmitFlag.set("提交中");
        
        try {
            // 执行业务逻辑...
            
            // 提交成功，清理标志
            orderSubmitFlag.remove();
            return "提交成功";
        } catch (Exception e) {
            // 处理异常，可根据需要重置标志或不处理
            return "提交失败";
        }
    }
}

2. 使用@SessionAttributes

如果你的应用启用了session，并且希望在用户会话级别防止重复提交，可以使用@SessionAttributes。

import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.SessionAttributes;
import org.springframework.web.bind.support.SessionStatus;

@Controller
@SessionAttributes("submitFlag")
public class OrderController {

    @PostMapping("/order")
    public String submitOrder(Model model, SessionStatus sessionStatus) {
        if (model.getAttribute("submitFlag") != null) {
            return "请勿重复提交";
        }
        
        model.addAttribute("submitFlag", true);
        
        try {
            // 执行业务逻辑...
            
            sessionStatus.setComplete(); // 成功后清除session中的标志
            return "提交成功";
        } catch (Exception e) {
            // 处理异常
            return "提交失败";
        }
    }
}

分布式环境

在分布式环境中，由于请求可能被负载均衡到不同的服务器上，因此需要一个全局的共享存储来记录请求状态，常用的方法有Redis、数据库等。

使用Redis

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class OrderController {

    @Autowired
    private StringRedisTemplate redisTemplate;

    private static final String ORDER_SUBMIT_LOCK_PREFIX = "order:submit:";

    @PostMapping("/order")
    public String submitOrder(String orderId) {
        String key = ORDER_SUBMIT_LOCK_PREFIX + orderId;
        Boolean hasKey = redisTemplate.hasKey(key);
        
        if (Boolean.TRUE.equals(hasKey)) {
            return "请勿重复提交";
        }
        
        redisTemplate.opsForValue().set(key, "processing", 5, TimeUnit.MINUTES); // 设置5分钟过期
        
        try {
            // 执行业务逻辑...
            
            redisTemplate.delete(key); // 成功后删除key
            return "提交成功";
        } catch (Exception e) {
            // 处理异常，根据需要决定是否删除key
            return "提交失败";
        }
    }
}

除了上述提到的方法，还有其他一些策略和技术可以用来防止接口重复提交，特别是在处理高并发和分布式场景下。

更多方式

1. 使用Token机制

前端生成Token: 在表单中嵌入一个由前端生成的唯一Token（可以是UUID），每次提交时携带此Token。服务器接收到请求后验证Token的有效性（例如，检查Token是否已使用或是否在有效期内），并处理完请求后标记此Token为已使用，后续相同的Token请求则被拒绝。
后端生成Token: 类似地，服务器可以在首次请求时生成Token返回给客户端，客户端在后续的提交中携带此Token。服务器验证Token并执行幂等性检查。

2. 分布式锁

基于Redis的分布式锁: 已在上述答案中提及，但更进一步，可以利用RedLock算法或者Lua脚本来实现更安全的分布式锁，确保在高并发下的锁的正确获取与释放。
Zookeeper分布式锁: 利用Zookeeper的临时节点特性，实现分布式锁。客户端在尝试执行操作前先获取锁，操作完成后释放锁。