限流

1.限流的原理嚼黔,根據(jù)ip(也可以對(duì)總的訪問頻次做限制)和固定時(shí)間內(nèi)的訪問次數(shù)细层,做限制

2.springboot實(shí)現(xiàn)限流:

? ? 1)新建一個(gè)自定義注解:AccessLimit 用于接口上(注解實(shí)現(xiàn)更簡(jiǎn)單方便)

? ? ? ??@Inherited

????????@Documented

????????@Target({ElementType.FIELD, ElementType.TYPE, ElementType.METHOD})

????????@Retention(RetentionPolicy.RUNTIME)

????????public @interface AccessLimit {

???????????? /**

? ????????? * 指定second 時(shí)間內(nèi),API最多的請(qǐng)求次數(shù)

????????????*/

????????????int times() default 3;

????????????/**

???????????? * 指定時(shí)間second唬涧,redis數(shù)據(jù)過期時(shí)間

???????????? */

????????????int second() default 10;

????????}

? ? 2)新建一個(gè)攔截器疫赎,獲取自定義注解上的參數(shù),然后緩存到redis中做校驗(yàn)

????????@Component

????????public class AccessLimitInterceptimplements HandlerInterceptor {

????????private static final Loggerlogger = LoggerFactory.getLogger(AccessLimitIntercept.class);

? ? ????????@Autowired

? ????????? private RedisUtilredisUtil;

????????????@Override

????????????public boolean preHandle(HttpServletRequest request, HttpServletResponse response,Object handler)throws Exception{

????????????????try{

????????????????????// handler 是否為 HandlerMethod 實(shí)例

????????????????????if(handlerinstanceof HandlerMethod){

????????????????????????// 強(qiáng)轉(zhuǎn)

????????????????????????HandlerMethod handlerMethod = (HandlerMethod) handler;

????????????????????????// 獲取方法

????????????????????????Method method = handlerMethod.getMethod();

????????????????????????// 判斷方法是否有AccessLimit注解碎节,有的才需要做限流

????????????????????????if(!method.isAnnotationPresent(AccessLimit.class)){

????????????????????????????return true;

????????????????????????}

????????????????????????// 獲取注釋上的內(nèi)容

????????????????????????AccessLimit accessLimit = method.getAnnotation(AccessLimit.class);

????????????????????????????if(accessLimit ==null){

????????????????????????????return true;

????????????????????????}

????????????????????????// 獲取方法上的請(qǐng)求次數(shù)

????????????????????????int times = accessLimit.times();

????????????????????????// 獲取方法注釋上的請(qǐng)求時(shí)間

????????????????????????Integer second = accessLimit.second();

????????????????????????// 拼接redis key = ip + api限流

????????????????????????String key = IpUtils.getIpAddr(request) + request.getRequestURI();

????????????????????????Integer maxTimes =null;

????????????????????????// 獲取 redis 的 value

????????????????????????String value =redisUtil.get(key) ==null?"":redisUtil.get(key).toString();

????????????????????????if(!"".equals(value)){

????????????????????????????maxTimes = Integer.valueOf(value);

????????????????????????}

????????????????????????if(maxTimes ==null){

????????????????????????????// 如果redis中沒有該ip對(duì)應(yīng)的時(shí)間則表示第一次調(diào)用捧搞,保存key到redis

????????????????????????????redisUtil.set(key,"1",second, TimeUnit.SECONDS);

????????????????????????}else if (maxTimes < times){

????????????????????????????// 如果 redis 中的時(shí)間比注解上的時(shí)間小,則表示可以允許訪問狮荔,這是修改redis的value時(shí)間

????????????????????????????redisUtil.set(key,maxTimes +1 +"",second,TimeUnit.SECONDS);

????????????????????????}else {

????????????????????????????logger.info(key +"請(qǐng)求過于頻繁");

????????????????????????????return setResponse(Result.error(ResultStatus.REQUEST_SEND_FREQUENTLY_ERROR.getStatus(),"請(qǐng)求過于頻繁胎撇,請(qǐng)稍后重試"),response);

????????????????????????}

????????????????????}

????????????????}catch (Exception ex){

????????????????????logger.error("API請(qǐng)求限流攔截異常,異常原因:" + ex);

????????????????????throw new ParameterException(ex.getMessage());

????????????????}

????????????????return true;

????????????}

????????????@Override

????????????public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView)throws Exception{

????????????}

????????????@Override

????????????public void afterCompletion(HttpServletRequest request,HttpServletResponse response,Object handler,Exception ex)throws Exception{

????????????}

????????????private boolean setResponse(Result result,HttpServletResponse response)throws IOException{

????????????????ServletOutputStream outputStream =null;

????????????????try{

????????????????????response.setHeader("Content-Type","application/json;charset=utf-8");

????????????????????outputStream = response.getOutputStream();

????????????????????outputStream.write(JSONUtil.toJsonStr(result).getBytes("UTF-8"));

????????????????}catch (Exception ex){

????????????????????logger.error("setResponse方法報(bào)錯(cuò)",ex);

????????????????????return false;

? ? ? ? ? ? ? ? }finally {

????????????????????if(outputStream !=null){

????????????????????????outputStream.flush();

????????????????????????outputStream.close();

????????????????????}

? ? ? ? ? ? ? ? }

????????????????return false;

????????????}

????????}

? ? 3)配置攔截器

? ??????@Configuration

????????public class WebFilterConfig implements WebMvcConfigurer {

????????????/**

????????????* 這里需要先將限流攔截器入住殖氏,不然無法獲取到攔截器中的redistemplate

????????????* @return

????????????*/

????????????@Bean

????????????public AccessLimitIntercept getAccessLimitIntercept() {

????????????????return new AccessLimitIntercept();

????????????}

????????????/**

????????????* 多個(gè)攔截器組成一個(gè)攔截器鏈

????????????* @param registry

????????????*/

????????????@Override

????????????public void addInterceptors(InterceptorRegistry registry) {

????????????????registry.addInterceptor(getAccessLimitIntercept()).addPathPatterns("/**");

????????????}

????????}

? ? 4)在接口上添加注解创坞,設(shè)置固定秒數(shù)內(nèi),限制的訪問次數(shù)

? ??????@RestController

????????@RequestMapping("/")

????????public class PingController extends BaseController {

????????????@AccessLimit(times = 5, second = 10)

????????????@GetMapping(value = "/ping")

????????????public Results ping() {

????????????????return succeed("pong", "");

????????????}

? ? ? ? }

? ? 5)注意:該限流加完后受葛,接口依然可以被訪問,如果需要修改為接口不可被訪問,需要修改2)中setResponse()方法的放回值為false总滩,preHandle()方法的返回值纲堵,可以決定方法是否可被執(zhí)行,當(dāng)為false時(shí)闰渔,方法不會(huì)被執(zhí)行

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末席函,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子冈涧,更是在濱河造成了極大的恐慌茂附,老刑警劉巖,帶你破解...
    沈念sama閱讀 219,270評(píng)論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件督弓,死亡現(xiàn)場(chǎng)離奇詭異营曼,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)愚隧,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,489評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門蒂阱,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人狂塘,你說我怎么就攤上這事录煤。” “怎么了荞胡?”我有些...
    開封第一講書人閱讀 165,630評(píng)論 0 356
  • 文/不壞的土叔 我叫張陵妈踊,是天一觀的道長(zhǎng)。 經(jīng)常有香客問我泪漂,道長(zhǎng)廊营,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,906評(píng)論 1 295
  • 正文 為了忘掉前任窖梁,我火速辦了婚禮赘风,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘纵刘。我一直安慰自己邀窃,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,928評(píng)論 6 392
  • 文/花漫 我一把揭開白布假哎。 她就那樣靜靜地躺著瞬捕,像睡著了一般。 火紅的嫁衣襯著肌膚如雪舵抹。 梳的紋絲不亂的頭發(fā)上肪虎,一...
    開封第一講書人閱讀 51,718評(píng)論 1 305
  • 那天,我揣著相機(jī)與錄音惧蛹,去河邊找鬼扇救。 笑死刑枝,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的迅腔。 我是一名探鬼主播装畅,決...
    沈念sama閱讀 40,442評(píng)論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼沧烈!你這毒婦竟也來了掠兄?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,345評(píng)論 0 276
  • 序言:老撾萬榮一對(duì)情侶失蹤锌雀,失蹤者是張志新(化名)和其女友劉穎蚂夕,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體腋逆,經(jīng)...
    沈念sama閱讀 45,802評(píng)論 1 317
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡婿牍,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,984評(píng)論 3 337
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了闲礼。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片牍汹。...
    茶點(diǎn)故事閱讀 40,117評(píng)論 1 351
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖柬泽,靈堂內(nèi)的尸體忽然破棺而出慎菲,到底是詐尸還是另有隱情,我是刑警寧澤锨并,帶...
    沈念sama閱讀 35,810評(píng)論 5 346
  • 正文 年R本政府宣布露该,位于F島的核電站,受9級(jí)特大地震影響第煮,放射性物質(zhì)發(fā)生泄漏解幼。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,462評(píng)論 3 331
  • 文/蒙蒙 一包警、第九天 我趴在偏房一處隱蔽的房頂上張望撵摆。 院中可真熱鬧,春花似錦害晦、人聲如沸特铝。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,011評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)鲫剿。三九已至,卻和暖如春稻轨,著一層夾襖步出監(jiān)牢的瞬間灵莲,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,139評(píng)論 1 272
  • 我被黑心中介騙來泰國(guó)打工殴俱, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留政冻,地道東北人枚抵。 一個(gè)月前我還...
    沈念sama閱讀 48,377評(píng)論 3 373
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像赠幕,于是被迫代替她去往敵國(guó)和親俄精。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,060評(píng)論 2 355