基于注解實現(xiàn) SpringBoot 接口防刷的方法
更新時間：2021年03月02日 11:52:38 作者：殺人償命百歲
這篇文章主要介紹了基于注解實現(xiàn) SpringBoot 接口防刷的方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
該示例項目通過自定義注解，實現(xiàn)接口訪問次數(shù)控制，從而實現(xiàn)接口防刷功能，項目結(jié)構(gòu)如下：
一、編寫注解類 AccessLimit
package cn.mygweb.annotation;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* 訪問控制注解（實現(xiàn)接口防刷功能）
*/
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface AccessLimit {
/**
* 限制周期（單位為秒）
*
* @return
*/
int seconds();
/**
* 規(guī)定周期內(nèi)限制次數(shù)
*
* @return
*/
int maxCount();
/**
* 是否需要登錄
*
* @return
*/
boolean needLogin() default false;
}
二、在Interceptor攔截器中實現(xiàn)攔截邏輯
package cn.mygweb.interceptor;
import cn.mygweb.annotation.AccessLimit;
import cn.mygweb.entity.Result;
import cn.mygweb.entity.StatusCode;
import com.alibaba.fastjson.JSON;
import org.springframework.stereotype.Component;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;
/**
* 訪問控制攔截器
*/
@Component
public class AccessLimitInterceptor extends HandlerInterceptorAdapter {
//模擬數(shù)據(jù)存儲，實際業(yè)務(wù)中可以自定義實現(xiàn)方式
private static Map<String, AccessInfo> accessInfoMap = new HashMap<>();
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response,
Object handler) throws Exception {
//判斷請求是否屬于方法的請求
if (handler instanceof HandlerMethod) {
HandlerMethod hm = (HandlerMethod) handler;
//獲取方法中的注解，看是否有該注解
AccessLimit accessLimit = hm.getMethodAnnotation(AccessLimit.class);
if (accessLimit == null) {
return true;
}
int seconds = accessLimit.seconds();
int maxCount = accessLimit.maxCount();
boolean needLogin = accessLimit.needLogin();
String key = request.getRequestURI();
//如果需要登錄
if (needLogin) {
//獲取登錄的session進行判斷
//……
key += " " + "userA";//這里假設(shè)用戶是userA，實際項目中可以改為userId
}
//模擬從redis中獲取數(shù)據(jù)
AccessInfo accessInfo = accessInfoMap.get(key);
if (accessInfo == null) {
//第一次訪問
accessInfo = new AccessInfo();
accessInfo.setFirstVisitTimestamp(System.currentTimeMillis());
accessInfo.setAccessCount(1);
accessInfoMap.put(key, accessInfo);
} else if (accessInfo.getAccessCount() < maxCount) {
//訪問次數(shù)加1
accessInfo.setAccessCount(accessInfo.getAccessCount() + 1);
accessInfoMap.put(key, accessInfo);
} else {
//超出訪問次數(shù)，判斷時間是否超出設(shè)定時間
if ((System.currentTimeMillis() - accessInfo.getFirstVisitTimestamp()) <= seconds * 1000) {
//如果還在設(shè)定時間內(nèi)，則為不合法請求，返回錯誤信息
render(response, "達到訪問限制次數(shù)，請稍后重試！");
return false;
} else {
//如果超出設(shè)定時間，則為合理的請求，將之前的請求清空，重新計數(shù)
accessInfo.setFirstVisitTimestamp(System.currentTimeMillis());
accessInfo.setAccessCount(1);
accessInfoMap.put(key, accessInfo);
}
}
}
return true;
}
/**
* 向頁面發(fā)送消息
*
* @param response
* @param msg
* @throws Exception
*/
private void render(HttpServletResponse response, String msg) throws Exception {
response.setContentType("application/json;charset=UTF-8");
OutputStream out = response.getOutputStream();
String str = JSON.toJSONString(new Result(true, StatusCode.ACCESSERROR, msg));
out.write(str.getBytes("UTF-8"));
out.flush();
out.close();
}
/**
* 封裝的訪問信息對象
*/
class AccessInfo {
/**
* 一個計數(shù)周期內(nèi)第一次訪問的時間戳
*/
private long firstVisitTimestamp;
/**
* 訪問次數(shù)統(tǒng)計
*/
private int accessCount;
public long getFirstVisitTimestamp() {
return firstVisitTimestamp;
}
public void setFirstVisitTimestamp(long firstVisitTimestamp) {
this.firstVisitTimestamp = firstVisitTimestamp;
}
public int getAccessCount() {
return accessCount;
}
public void setAccessCount(int accessCount) {
this.accessCount = accessCount;
}
@Override
public String toString() {
return "AccessInfo{" +
"firstVisitTimestamp=" + firstVisitTimestamp +
", accessCount=" + accessCount +
'}';
}
}
}
三、把Interceptor注冊到springboot中
package cn.mygweb.config;
import cn.mygweb.interceptor.AccessLimitInterceptor;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
/**
* 攔截器注冊配置
*/
@Configuration
public class WebConfig implements WebMvcConfigurer {
@Override
public void addInterceptors(InterceptorRegistry registry) {
//注冊攔截器
registry.addInterceptor(new AccessLimitInterceptor());
}
}
四、在Controller中加入注解實現(xiàn)接口防刷
package cn.mygweb.controller;
import cn.mygweb.annotation.AccessLimit;
import cn.mygweb.entity.Result;
import cn.mygweb.entity.StatusCode;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
@RequestMapping("/access")
public class AccessController {
@AccessLimit(seconds = 5, maxCount = 2)//訪問控制，5秒內(nèi)只能訪問2次
@GetMapping
public Result access() {
return new Result(true, StatusCode.OK, "訪問成功！");
}
}
五、測試訪問
附：StatusCode.java、Result.java、application.yml
StatusCode類
package cn.mygweb.entity;
/**
* 返回狀態(tài)碼
*/
public class StatusCode {
public static final int OK = 20000;//成功
public static final int ERROR = 20001;//失敗
public static final int LOGINERROR = 20002;//用戶名或密碼錯誤
public static final int ACCESSERROR = 20003;//權(quán)限不足
public static final int REMOTEERROR = 20004;//遠(yuǎn)程調(diào)用失敗
public static final int REPERROR = 20005;//重復(fù)操作
public static final int NOTFOUNDERROR = 20006;//沒有對應(yīng)的搶購數(shù)據(jù)
}
Result類：
package cn.mygweb.entity;
import java.io.Serializable;
/**
* 響應(yīng)結(jié)果
*/
public class Result<T> implements Serializable {
private boolean flag;//是否成功
private Integer code;//返回碼
private String message;//返回消息
private T data;//返回數(shù)據(jù)
public Result(boolean flag, Integer code, String message, Object data) {
this.flag = flag;
this.code = code;
this.message = message;
this.data = (T) data;
}
public Result(boolean flag, Integer code, String message) {
this.flag = flag;
this.code = code;
this.message = message;
}
public Result() {
this.flag = true;
this.code = StatusCode.OK;
this.message = "操作成功!";
}
public boolean isFlag() {
return flag;
}
public void setFlag(boolean flag) {
this.flag = flag;
}
public Integer getCode() {
return code;
}
public void setCode(Integer code) {
this.code = code;
}
public String getMessage() {
return message;
}
public void setMessage(String message) {
this.message = message;
}
public T getData() {
return data;
}
public void setData(T data) {
this.data = data;
}
}
applications.yml:
server: port: 8080
到此這篇關(guān)于基于注解實現(xiàn) SpringBoot 接口防刷的方法的文章就介紹到這了,更多相關(guān)SpringBoot 接口防刷內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
python中文分詞+詞頻統(tǒng)計的實現(xiàn)步驟
詞頻統(tǒng)計就是輸入一段句子或者一篇文章,然后統(tǒng)計句子中每個單詞出現(xiàn)的次數(shù),下面這篇文章主要給大家介紹了關(guān)于python中文分詞+詞頻統(tǒng)計的相關(guān)資料,需要的朋友可以參考下2022-06-06
Python實現(xiàn)讀取SQLServer數(shù)據(jù)并插入到MongoDB數(shù)據(jù)庫的方法示例
這篇文章主要介紹了Python實現(xiàn)讀取SQLServer數(shù)據(jù)并插入到MongoDB數(shù)據(jù)庫的方法,涉及Python同時進行SQLServer與MongoDB數(shù)據(jù)庫的連接、查詢、讀取、寫入等相關(guān)操作實現(xiàn)技巧,需要的朋友可以參考下2018-06-06
本文將介紹處理大量數(shù)據(jù)時非常方便的工具，例如tqdm與 multiprocessing?imap??一起使用、并行處理檔案、繪制和處理3D數(shù)據(jù)等，感興趣的小伙伴可以了解一下2021-12-12
在Python的標(biāo)準(zhǔn)庫中,sys模塊是一個常用而強大的工具,它提供了與Python解釋器交互的函數(shù)和變量,本文將介紹sys模塊的一些常用函數(shù)和方法,并通過實際的代碼實例來解析它們的用法,希望對大家有所幫助2024-03-03
Python?中?Selenium?的?send_keys()?函數(shù)用法小結(jié)
send_keys() 是將數(shù)字、文本和符號等鍵盤輸入發(fā)送到應(yīng)用程序的文本框的過程， send_keys() 是 WebDriver 的一部分，每個鍵盤輸入都會發(fā)送到此元素，這篇文章主要介紹了Python?中?Selenium?的?send_keys()?函數(shù),需要的朋友可以參考下2023-11-11
Python多任務(wù)版靜態(tài)Web服務(wù)器實現(xiàn)示例
這篇文章主要為大家介紹了Python靜態(tài)Web服務(wù)器多任務(wù)版實現(xiàn)示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-06-06
今天小編就為大家分享一篇python3 對list中每個元素進行處理的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-06-06
Python構(gòu)造函數(shù)與析構(gòu)函數(shù)超詳細(xì)分析
在python之中定義一個類的時候會在類中創(chuàng)建一個名為__init__的函數(shù)，這個函數(shù)就叫做構(gòu)造函數(shù)。它的作用就是在實例化類的時候去自動的定義一些屬性和方法的值，而析構(gòu)函數(shù)恰恰是一個和它相反的函數(shù)，這篇文章主要介紹了Python構(gòu)造函數(shù)與析構(gòu)函數(shù)2022-11-11
python實戰(zhàn)之利用pygame實現(xiàn)貪吃蛇游戲（二）
這篇文章主要介紹了python實戰(zhàn)之利用pygame實現(xiàn)貪吃蛇游戲（二）,文中有非常詳細(xì)的代碼示例,對正在學(xué)習(xí)python的小伙伴們有很好的幫助,需要的朋友可以參考下2021-05-05
今天小編就為大家分享一篇python 把列表轉(zhuǎn)化為字符串的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-10-10