來源：https://www.jianshu.com/p/697f1c5eaa3f?utm_campaign=haruki&utm_content=note&utm_medium=reader_share&utm_source=qq

本文介紹一種極簡潔、靈活通用介面防刷實現方式、通過在需要防刷的方法加上@Prevent 註解即可實現簡訊防刷；
使用方式大致如下：

/**
     * 測試防刷
     *
     * @param request
     * @return
     */
    @ResponseBody
    @GetMapping(value = "/testPrevent")
    @Prevent //加上該註解即可實現簡訊防刷(預設一分鐘內不允許重複呼叫，支援擴充套件、配置）
    public Response testPrevent(TestRequest request) {
        return Response.success("呼叫成功");
    }
 

目錄

• 1、實現防刷切面PreventAop.java
• 2、使用防刷切面
• 3、演示

1、實現防刷切面PreventAop.java

大致邏輯為：定義一切面，通過@Prevent註解作為切入點、在該切面的前置通知獲取該方法的所有入參並將其Base64編碼，將入參Base64編碼+完整方法名作為redis的key，入參作為reids的value，@Prevent的value作為redis的expire，存入redis；每次進來這個切面根據入參Base64編碼+完整方法名判斷redis值是否存在，存在則攔截防刷，不存在則允許呼叫；

1.1 定義註解Prevent

package com.zetting.aop;

import java.lang.annotation.*;

/**
 * 介面防刷註解
 * 使用：
 * 在相應需要防刷的方法上加上
 * 該註解，即可
 *
 * @author: zetting
 * @date:2018/12/29
 */
@Documented
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface Prevent {

    /**
     * 限制的時間值（秒）
     *
     * @return
     */
    String value() default "60";

    /**
     * 提示
     */
    String message() default "";

    /**
     * 策略
     *
     * @return
     */
    PreventStrategy strategy() default PreventStrategy.DEFAULT;
}
 

1.2 實現防刷切面PreventAop

package com.zetting.aop;

import com.alibaba.fastjson.JSON;
import com.zetting.common.BusinessException;
import com.zetting.util.RedisUtil;
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.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.lang.reflect.Method;
import java.util.Base64;

/**
 * 防刷切面實現類
 *
 * @author: zetting
 * @date: 2018/12/29 20:27
 */
@Aspect
@Component
public class PreventAop {
    private static Logger log = LoggerFactory.getLogger(PreventAop.class);

    @Autowired
    private RedisUtil redisUtil;


    /**
     * 切入點
     */
    @Pointcut("@annotation(com.zetting.aop.Prevent)")
    public void pointcut() {
    }


    /**
     * 處理前
     *
     * @return
     */
    @Before("pointcut()")
    public void joinPoint(JoinPoint joinPoint) throws Exception {
        String requestStr = JSON.toJSONString(joinPoint.getArgs()[0]);
        if (StringUtils.isEmpty(requestStr) || requestStr.equalsIgnoreCase("{}")) {
            throw new BusinessException("[防刷]入參不允許為空");
        }

        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
        Method method = joinPoint.getTarget().getClass().getMethod(methodSignature.getName(),
                methodSignature.getParameterTypes());

        Prevent preventAnnotation = method.getAnnotation(Prevent.class);
        String methodFullName = method.getDeclaringClass().getName() + method.getName();

        entrance(preventAnnotation, requestStr,methodFullName);
        return;
    }


    /**
     * 入口
     *
     * @param prevent
     * @param requestStr
     */
    private void entrance(Prevent prevent, String requestStr,String methodFullName) throws Exception {
        PreventStrategy strategy = prevent.strategy();
        switch (strategy) {
            case DEFAULT:
                defaultHandle(requestStr, prevent,methodFullName);
                break;
            default:
                throw new BusinessException("無效的策略");
        }
    }


    /**
     * 預設處理方式
     *
     * @param requestStr
     * @param prevent
     */
    private void defaultHandle(String requestStr, Prevent prevent,String methodFullName) throws Exception {
        String base64Str = toBase64String(requestStr);
        long expire = Long.parseLong(prevent.value());

        String resp = redisUtil.get(methodFullName+base64Str);
        if (StringUtils.isEmpty(resp)) {
            redisUtil.set(methodFullName+base64Str, requestStr, expire);
        } else {
            String message = !StringUtils.isEmpty(prevent.message()) ? prevent.message() :
                    expire + "秒內不允許重複請求";
            throw new BusinessException(message);
        }
    }


    /**
     * 物件轉換為base64字串
     *
     * @param obj 物件值
     * @return base64字串
     */
    private String toBase64String(String obj) throws Exception {
        if (StringUtils.isEmpty(obj)) {
            return null;
        }
        Base64.Encoder encoder = Base64.getEncoder();
        byte[] bytes = obj.getBytes("UTF-8");
        return encoder.encodeToString(bytes);
    }
}

注：
以上只展示核心程式碼、其他次要程式碼（例如redis配置、redis工具類等）可下載原始碼查閱

2、使用防刷切面

在MyController 使用防刷

package com.zetting.modules.controller;

import com.zetting.aop.Prevent;
import com.zetting.common.Response;
import com.zetting.modules.dto.TestRequest;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

/**
 * 切面實現入參校驗
 */
@RestController
public class MyController {

    /**
     * 測試防刷
     *
     * @param request
     * @return
     */
    @ResponseBody
    @GetMapping(value = "/testPrevent")
    @Prevent
    public Response testPrevent(TestRequest request) {
        return Response.success("呼叫成功");
    }


    /**
     * 測試防刷
     *
     * @param request
     * @return
     */
    @ResponseBody
    @GetMapping(value = "/testPreventIncludeMessage")
    @Prevent(message = "10秒內不允許重複調多次", value = "10")//value 表示10表示10秒
    public Response testPreventIncludeMessage(TestRequest request) {
        return Response.success("呼叫成功");
    }
}

3、演示

GIF.gif

gitee 原始碼：https://gitee.com/Zetting/my-gather/tree/master/springboot-aop-prevent

推薦閱讀：

• 【Springboot之切面程式設計】註解實現敏感欄位加解密
• 【Springboot之搜尋日誌妙招】在日誌上列印請求唯一log標識
• 【Springboot之切面程式設計】通過切面AOP實現入參校驗
• 【Springboot之切面程式設計】自定義註解實現入參指定列舉值校驗

作者：zetting
連結：https://www.jianshu.com/p/697f1c5eaa3f
來源：簡書
簡書著作權歸作者所有，任何形式的轉載都請聯絡作者獲得授權並註明出處。