0
点赞
收藏
分享

微信扫一扫

【Springboot之切面编程】注解实现接口防刷

/**
* 测试防刷
*
* @param request
* @return
*/

@ResponseBody
@GetMapping(value = "/testPrevent")
@Prevent //加上该注解即可实现短信防刷(默认一分钟内不允许重复调用,支持扩展、配置)
public Response testPrevent(TestRequest request) {
return Response.success("调用成功");
}

目录

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

1、实现防刷切面PreventAop.java

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、演示

gitee 源码:https://gitee.com/Zetting/my-gather/tree/master/springboot-aop-prevent

推荐阅读:

  • 【Springboot之切面编程】注解实现敏感字段加解密
  • 【Springboot之搜索日志妙招】在日志上打印请求唯一log标识
  • 【Springboot之切面编程】通过切面AOP实现入参校验
  • 【Springboot之切面编程】自定义注解实现入参指定枚举值校验

其他推荐:

light4admin一款及简单管理后台

举报

相关推荐

0 条评论