快捷導(dǎo)航

SpringBoot攔截器實現(xiàn)項目防止接口重復(fù)提交

更新時間：2023年09月03日 15:26:06 作者：草帽夫卡

基于SpringBoot框架來開發(fā)業(yè)務(wù)后臺項目時,接口重復(fù)提交是一個常見的問題,本文主要介紹了SpringBoot攔截器實現(xiàn)項目防止接口重復(fù)提交,具有一定的參考價值,感興趣的可以了解一下

基于SpringBoot框架來開發(fā)業(yè)務(wù)后臺項目時，接口重復(fù)提交是一個常見的問題。為了避免這個問題，我們可以通過自定義攔截器實現(xiàn)一個后臺攔截接口重復(fù)提交的功能，本文將介紹如何使用基于SpringBoot實現(xiàn)這個功能。

首先，我們需要引入一個Aop依賴。在pom.xml文件中添加如下依賴：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

創(chuàng)建一個自定義注解@NoRepeatSubmit，用于標(biāo)記需要攔截的接口：

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 {
    /**
     * 設(shè)置請求鎖定時間，默認(rèn)為5秒
     */
    int lockTime() default 5;
}

創(chuàng)建一個攔截器類NoRepeatSubmitInterceptor，實現(xiàn)HandlerInterceptor接口，并在其中實現(xiàn)攔截邏輯：

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.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
@Aspect
@Component
public class NoRepeatSubmitInterceptor implements HandlerInterceptor {
    @Pointcut("@annotation(com.example.demo.annotation.NoRepeatSubmit)")
    public void noRepeatSubmitPointcut() {
    }
    @Around("noRepeatSubmitPointcut()")
    public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
        // 獲取請求參數(shù)中的token值
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        HttpSession session = request.getSession();
        String token = session.getAttribute("token").toString();
        // 判斷是否已經(jīng)提交過請求，如果已經(jīng)提交過并且時間間隔小于鎖時間，則直接返回成功結(jié)果，否則認(rèn)為是重復(fù)提交，拋出異常并鎖定session
        if (isSubmitted(token)) {
            String lockKey = UUID.randomUUID().toString();
            session.setAttribute(lockKey, System.currentTimeMillis());
            throw new RuntimeException("請勿重復(fù)提交");
        } else {
            session.setAttribute("token", token);
            return joinPoint.proceed();
        }
    }
    private boolean isSubmitted(String token) {
        HttpSession session = request.getSession();
        Object lockKey = session.getAttribute(token);
        if (lockKey == null) {
            return false;
        } else {
            long lockTime = (Long) session.getAttribute(token);
            if (lockTime > 0 && System.currentTimeMillis() - lockTime < TimeUnit.SECONDS.toMillis(5)) {
                return true;
            } else {
                session.removeAttribute(token);
                return false;
            }
        }
    }
}

注冊攔截器

實現(xiàn)HandlerInterceptor接口的重寫，重寫preHandle、postHandle、afterCompletion方法。攔截器中的方法執(zhí)行流程為 preHandle → controlle → postHandle → afterCompletion。然后需要將攔截器注冊到容器中，可以通過實現(xiàn)WebMvcConfigurer的addInterceptors方法來實現(xiàn)。下面是一個簡單的例子：

@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new NoRepeatSubmitInterceptor()).addPathPatterns("/**").excludePathPatterns("/login", "/register");
    }
}

在需要進行攔截的接口上添加@NoRepeatSubmit注解，例如：

@RestController
public class UserController {
    @NoRepeatSubmit // 添加此注解表示該接口需要攔截重復(fù)提交請求
    @GetMapping("/submit")
    public String submit() {
        // 處理業(yè)務(wù)邏輯...
        return "success";
    }
}

通過以上步驟，我們實現(xiàn)了一個簡單的后臺攔截接口重復(fù)提交的功能。在實際項目中，還需要考慮更多的細節(jié)，例如如何保證鎖的釋放、如何處理并發(fā)請求等。但這個示例應(yīng)該能幫助你入門SpringBoot攔截器的使用。

到此這篇關(guān)于SpringBoot攔截器實現(xiàn)項目防止接口重復(fù)提交的文章就介紹到這了,更多相關(guān)SpringBoot 防止接口重復(fù)提交內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: