SpringBoot集成Sentinel 实现QPS限流

Spring Cloud Alibaba 的 Sentinel 组件提供了丰富的"流量控制"规则,

单体SpringBoot应用中也可以集成 Sentinel 来实现流量控制,本文主要讲 QPS流量控制。

SpringBoot集成Sentinel有两种方式:

一种是 dashboard控制面板的方式,需要下载sentinel-dashboard-1.6.3.jar

另一种是不需要dashboard,直接代码集成的方式。

这里采用的是第二种

注意:版本非常重要

有博文说 SpringBoot版本在 2.2.0 到 2.3.1之间才行,其他版本流量控制不生效。

我这里SpringBoot版本为 2.2.5.RELEASE,Sentinel版本为 2.2.1RELEASE(之前 sentinel版本为2.1.2.RELEASE,限流死活不生效)

1.添加Sentinel的Maven依赖包

XML 复制代码
<dependency>
    <groupId>com.alibaba.cloud</groupId>
    <artifactId>spring-cloud-starter-alibaba-sentinel</artifactId>
	<version>2.2.1.RELEASE</version>
</dependency>

2.定义限流规则

java 复制代码
import java.util.ArrayList;
import java.util.List;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.stereotype.Component;
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager;
@Component
public class SentinelConfiguration implements InitializingBean{
    public final static String QPS_LIMIT_GET_USER= "getUser";
    @Override
    public void afterPropertiesSet() {
        initFlowQpsRule(QPS_LIMIT_GET_USER);
    }
    private void initFlowQpsRule(String resource) {
        List<FlowRule> rules = new ArrayList<>();
        FlowRule rule = new FlowRule();
        rule.setResource(resource);
        //设置限流规则为  QPS限流(每秒请求量)
        rule.setGrade(RuleConstant.FLOW_GRADE_QPS);
        rule.setCount(1); //1秒内允许请求1次
        rules.add(rule);
        FlowRuleManager.loadRules(rules);
    }
}

3.限流触发时的处理逻辑

java 复制代码
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import com.alibaba.csp.sentinel.slots.block.flow.FlowException;
@RestControllerAdvice
public class GlobalExceptionHandler{
    /**
     * 限流异常
     */
    @ExceptionHandler(FlowException.class)
    public String flowExceptionHandler(FlowException ex) {
        return "---接口被限流了---";
    }
}

注意:如果没有限流触发后的处理逻辑,会报以下异常

java 复制代码
com.alibaba.csp.sentinel.slots.block.flow.FlowException: null

4.编写接口测试限流是否生效

java 复制代码
import java.util.HashMap;
import java.util.Map;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import com.alibaba.csp.sentinel.annotation.SentinelResource;
import com.qyp.hpa.config.SentinelConfiguration;
@RestController
public class UserController {
	@RequestMapping("/getUser")
	@SentinelResource(value = SentinelConfiguration.QPS_LIMIT_GET_USER)
    public Map<String,Object> tempList() {
		Map<String,Object> user = new HashMap<String,Object>();
		user.put("name", "张三");
		user.put("age", 30);
        return user;
    }
}

浏览器请求getUser接口,正常返回

浏览器连续快速点击刷新(1秒内点2次以上),触发接口限流规则,返回预设的限流触发消息。

相关推荐
后端小张10 小时前
基于飞算AI的图书管理系统设计与实现
spring boot
考虑考虑1 天前
Jpa使用union all
java·spring boot·后端
阿杆2 天前
同事嫌参数校验太丑,我直接掏出了更优雅的 SpEL Validator
java·spring boot·后端
昵称为空C2 天前
SpringBoot3 http接口调用新方式RestClient + @HttpExchange像使用Feign一样调用
spring boot·后端
麦兜*3 天前
MongoDB Atlas 云数据库实战:从零搭建全球多节点集群
java·数据库·spring boot·mongodb·spring·spring cloud
麦兜*3 天前
MongoDB 在物联网(IoT)中的应用:海量时序数据处理方案
java·数据库·spring boot·物联网·mongodb·spring
汤姆yu3 天前
基于springboot的毕业旅游一站式定制系统
spring boot·后端·旅游
计算机毕业设计木哥3 天前
计算机毕设选题推荐:基于Java+SpringBoot物品租赁管理系统【源码+文档+调试】
java·vue.js·spring boot·mysql·spark·毕业设计·课程设计
波波烤鸭3 天前
Redis 高可用实战源码解析(Sentinel + Cluster 整合应用)
数据库·redis·sentinel
hdsoft_huge3 天前
Java & Spring Boot常见异常全解析:原因、危害、处理与防范
java·开发语言·spring boot