自定义注解+Redis防止用户重复访问接口 代码如下:

第一步:
@Target({ElementType.METHOD})//设置在方法上使用此注解
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Starter {

    /**
     * 默认3秒内算重复提交
     */
    long timeOut() default 3;
}
第二步:
package com.example.pdf_to_word.test.aop;

import com.example.pdf_to_word.test.common.Starter;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.Assert;
import org.springframework.util.StringUtils;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.net.*;
import java.util.concurrent.TimeUnit;

@Component
@Slf4j
@Aspect
public class StarterAop {

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    @Pointcut("@annotation(com.example.pdf_to_word.test.common.Starter)")
    public void serviceNoRepeat() {
        System.out.println("这是一个切入点!");
    }

    @Before("serviceNoRepeat()")
    public synchronized void Before(JoinPoint point) throws Exception {
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        Assert.notNull(attributes, "无请求!");
        HttpServletRequest request = attributes.getRequest();
        Assert.notNull(request, "request can not null");

        //获取用户ip
        String userIp = getIpAddr(request);
        String requestURI = request.getRequestURI();//请求接口
        //这里获取一下用户是否在过期时间内是否请求过相对应的接口!
        String userUrl = redisTemplate.opsForValue().get(userIp + requestURI);
        Assert.isNull(userUrl, "重复请求!");

        //获取注解内的重复时间
        MethodSignature signature = (MethodSignature) point.getSignature();
        Assert.notNull(signature, "无法解析注解!");
        Method method = signature.getMethod();
        Assert.notNull(method, "解析注解错误!");
        Starter annotation = method.getAnnotation(Starter.class);
        Assert.notNull(annotation, "获取注解失败!");
        long time = annotation.timeOut();
        Assert.isTrue(time > 0, "时间解析失败,请设置正确的过期时间!");

        //设置过期时间 条件:用户ip加用户请求接口做限制!
        redisTemplate.opsForValue().set(userIp + requestURI, requestURI, time, TimeUnit.SECONDS);
    }

    //获取用户ip地址
    public static String getIpAddr(HttpServletRequest request) {
        if (request == null) {
            return "unknown";
        }
        String ip = request.getHeader("x-forwarded-for");
        if (ip == null || ip.isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("Proxy-Client-IP");
        }
        if (ip == null || ip.isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("X-Forwarded-For");
        }
        if (ip == null || ip.isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("WL-Proxy-Client-IP");
        }
        if (ip == null || ip.isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("X-Real-IP");
        }

        if (ip == null || ip.isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getRemoteAddr();
        }

        return "0:0:0:0:0:0:0:1".equals(ip) ? "127.0.0.1" : ip;
    }
    
}
最后将注解写到需要的接口上实现成功。

  • 20
    点赞
  • 40
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值