1. 首页
  2. 软件开发
  3. Spring Boot中实现接口的防抖效果(禁止用户连续访问接口)

Spring Boot中实现接口的防抖效果(禁止用户连续访问接口)

  • 发布于 2024-08-17
  • 0 次阅读

本文由 简悦 SimpRead 转码, 原文地址 blog.csdn.net

在 Spring Boot 中,可以使用一些库或框架来实现接口的防抖效果。以下是一种常见的实现方式:

  1. 使用 Spring 的 AOP(面向切面编程)机制:通过定义一个自定义的注解,结合 AOP,实现对接口方法的防抖控制。

首先,你可以定义一个 @Debounce 注解,用于标记需要进行防抖处理的接口方法。例如:

package com.example.demo.utils;

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 Debounce {
    long value() default 1000; // 设置默认的防抖时间间隔,单位为毫秒
}

接下来,在你的 AOP 处理类中,使用 @Around 注解创建一个环绕通知,对使用了 @Debounce 注解的方法进行防抖处理。例如:

package com.example.demo.aspect;

import com.example.demo.utils.Debounce;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.reflect.MethodSignature;

import java.lang.reflect.Method;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

/**
 * DebounceAop
 * 防抖
 */
public class DebounceAop {
    private final ConcurrentMap<Method, Lock> lockMap = new ConcurrentHashMap<>();

    @Around("@annotation(debounce)")
    public Object debounce(ProceedingJoinPoint pjp, Debounce debounce) throws Throwable {
        MethodSignature signature = (MethodSignature) pjp.getSignature();
        Method method = signature.getMethod();
        Lock lock = lockMap.computeIfAbsent(method, k -> new ReentrantLock());
        try {
            if (lock.tryLock(debounce.value(), TimeUnit.MILLISECONDS)) {
                return pjp.proceed();
            } else {
                // 防抖时间内的请求不执行方法,可以根据需要进行处理
                return null;
            }
        } finally {
            lock.unlock();
        }
    }
}

在上述实现中,我们使用 ConcurrentHashMap 来维护不同的方法对应的锁,防止并发访问冲突。在防抖逻辑中,使用 tryLock() 方法尝试获取锁,如果成功获取到锁则执行目标方法,否则阻止方法执行。

最后,在需要进行防抖处理的接口方法上,使用 @Debounce 注解标记,并可以传入自定义的防抖时间间隔。例如:

@CrossOrigin
    @PostMapping("/test")
    @Debounce(1000) // 设置防抖时间间隔为1秒
    public void test(@RequestBody LoginPO po) {
        System.out.println("我进来了");
    }

这样,被 @Debounce 注解标记的接口方法在设定的防抖时间间隔内只会执行一次,后续的请求会被防抖处理。