目录
一、maven依赖
二、控制台
三、基于注解限流
四、热点参数限流
五、使用JMeter验证
一、maven依赖
需要注意,使用的版本需要和你的SpringBoot版本匹配!!
Spring-Cloud直接添加如下依赖即可,baba已经帮你指定好版本了。
当然可以点进去spring-cloud-starter-alibaba-sentinel搜索sentinel-core对应的版本
<dependency><groupId>com.alibaba.cloud</groupId><artifactId>spring-cloud-starter-alibaba-sentinel</artifactId>
</dependency>
如果是SpringBoot的话,需要手动导入包。
我这里使用的是SpringBoot Version 2.2.5.RELEASE,对应的版本是1.7.1 (你要根据自己SpringBoot版本查询对应版本,如果版本不对应的话根本用不了)
<sentinel.version>1.7.1</sentinel.version><!--sentinel核心包-->
<dependency><groupId>com.alibaba.csp</groupId><artifactId>sentinel-core</artifactId>
<version>${sentinel.version}</version>
</dependency>
<!--sentinel注解支持模块-->
<dependency><groupId>com.alibaba.csp</groupId><artifactId>sentinel-annotation-aspectj</artifactId><version>${sentinel.version}</version>
</dependency>
<!--sentinel与控制台交互-->
<dependency><groupId>com.alibaba.csp</groupId><artifactId>sentinel-transport-simple-http</artifactId>
<version>${sentinel.version}</version>
</dependency>
二、控制台
这里也介绍下控制台的使用方法,但我的项目是没运维给我安装这个东西,就没法用它。。
下载地址:Release v1.7.1 · alibaba/Sentinel · GitHub
使用命令启动jar包
java -jar sentinel-dashboard-17.1.jar
前端访问:localhost:8080, 账号密码 默认都是 sentinel
但是但是!!!!!!!界面是空的!!!!!!!!!
启动命令需要指定你的项目的端口,以及名称才行:
java -Dserver.port=8080 -Dcsp.sentinel.dashboard.server=localhost:8080 -Dcsp.sentinel.api.port=8998 -Dproject.name=sentinel-dashboard -jar sentinel-dashboard-1.7.1.jar
以下是各个命令的介绍:
-Dserver.port=9000 运行端口
-Dcsp.sentinel.dashboard.server=localhost:9000 浏览器访问地址
-Dproject.name=sentinel-dashboard 项目名称
-Dcsp.sentinel.api.port=8092 sentinel客户端端口
-Dsentinel.dashboard.auth.username=username,设置用户名
-Dsentinel.dashboard.auth.password=password,设置访问密码
三、基于注解限流
我的项目没法使用控制台进行动态修改规则,只能在数据库表里面写好规则,系统启动的时候进行读取(当然可以动态修改规则无须重启)
1、maven引入依赖:
<!--sentinel核心包-->
<dependency><groupId>com.alibaba.csp</groupId><artifactId>sentinel-core</artifactId><version>${sentinel.version}</version>
</dependency>
<!--sentinel注解支持模块-->
<dependency><groupId>com.alibaba.csp</groupId><artifactId>sentinel-annotation-aspectj</artifactId><version>${sentinel.version}</version>
</dependency>
2、创建SentinelConfig类,启用注解
import com.alibaba.csp.sentinel.annotation.aspectj.SentinelResourceAspect;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;@Slf4j
@Configuration
public class SentinelConfig {@Beanpublic SentinelResourceAspect sentinelResourceAspect() {return new SentinelResourceAspect();}
}
3、创建表 system_sentinel_config 规则配置表
DROP TABLE IF EXISTS `system_sentinel_config`;
CREATE TABLE `system_sentinel_config` (`id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '自增id',`module_name` varchar(100) NOT NULL COMMENT '资源名称',`resource` varchar(100) NOT NULL COMMENT '资源名称(唯一), 对应{@link SentinelResName}',`desc` varchar(100) NOT NULL COMMENT '资源描述',`grade` tinyint(1) NOT NULL DEFAULT 1 COMMENT '阈值类型, 0: 线程数, 1: QPS',`count` bigint(20) NOT NULL COMMENT '如果grade为线程数, 表示每秒最高线程数。如果grade为QPS, 表示每秒最高访问量。',`strategy` tinyint(1) NOT NULL DEFAULT 0 COMMENT '流控模式。0:直接, 1: 关联, 2:链路',`control_behavior` tinyint(1) NOT NULL DEFAULT 0 COMMENT '流控效果。0:默认(直接拒绝), 1: Warm Up(冷启动), 2:匀速排队 3: 冷启动+匀速排队',`max_queueing_time_ms` int DEFAULT NULL COMMENT '流控效果为匀速排队时,队列的排队时间(单位:毫秒),需要注意这里类型是int, 并且单位是毫秒',`warm_up_period_sec` int DEFAULT NULL COMMENT '冷启动到达最大值的时间(单位:秒)',`create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',`update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '更新时间',`create_user` VARCHAR(50) DEFAULT NULL COMMENT '创建人' ,`update_user` VARCHAR(50) DEFAULT NULL COMMENT '更新人' ,PRIMARY KEY (`id`),UNIQUE KEY resource_key(`resource`)
) ENGINE=InnoDB AUTO_INCREMENT = 1 DEFAULT CHARSET=utf8 COMMENT='sentinel限流配置表';-- 标准对接--档案类
INSERT INTO `system_sentinel_config` (`module_name`, `resource`, `desc`, `grade`, `count`, `strategy`, `control_behavior`, `max_queueing_time_ms`, `warm_up_period_sec`) VALUES
('标准对接--档案类', 'standard:structure:add', '新增/编辑建筑类节点(片区/楼栋/单元/住家)', 1, 20, 0, 0, null, null),
('标准对接--档案类', 'standard:structure:delete', '删除建筑类节点(片区/楼栋/单元/住家)', 1, 20, 0, 0, null, null),
('标准对接--档案类', 'standard:file:upload', '上传图片', 1, 20, 0, 0, null, null),
('标准对接--档案类', 'standard:check:face', '人脸质量校验', 1, 20, 0, 0, null, null),
('标准对接--档案类', 'standard:household:add', '新增/编辑住户(业主,租户)', 1, 20, 0, 0, null, null),
('标准对接--档案类', 'standard:household:delete', '删除住户', 1, 20, 0, 0, null, null),
('标准对接--档案类', 'standard:visitor:add', '新增/编辑访客', 1, 20, 0, 0, null, null),
('标准对接--档案类', 'standard:visitor:delete', '删除访客', 1, 20, 0, 0, null, null);-- 标准对接--其它
INSERT INTO `system_sentinel_config` (`module_name`, `resource`, `desc`, `grade`, `count`, `strategy`, `control_behavior`, `max_queueing_time_ms`, `warm_up_period_sec`) VALUES
('标准对接--其它', 'standard:remote:open', '第三方远程开门', 1, 20, 0, 0, null, null),
('标准对接--其它', 'standard:smart:lock:control', '第三方锁门禁权限冻结/解冻', 1, 20, 0, 0, null, null),
('标准对接--其它', 'standard:smart:lock:set:temporary:pwd', '第三方锁临时密码下发', 1, 20, 0, 0, null, null);
4、对应的SystemSentinelConfigDO对象
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
import com.whysu.scd.base.common.constant.sentinel.SentinelResName;
import lombok.Data;
import lombok.EqualsAndHashCode;/*** sentinel限流配置表(SystemSentinelConfig)实体类*/
@Data
public class SystemSentinelConfigDO {/*** 主键id*/private Long id;/*** 资源名称*/private String moduleName;/*** 资源名称(唯一), 对应{@link SentinelResName}*/private String resource;/*** 资源描述*/private String desc;/*** 阈值类型* 0: 线程数 {@link RuleConstant#FLOW_GRADE_THREAD},* 1: QPS {@link RuleConstant#FLOW_GRADE_QPS}*/private Integer grade;/*** 如果grade为线程数, 表示每秒最高线程数。* 如果grade为QPS, 表示每秒最高访问量。*/private Long count;/*** 流控模式。* 0:直接{@link RuleConstant#STRATEGY_DIRECT},* 1:关联{@link RuleConstant#STRATEGY_RELATE},* 2:链路{@link RuleConstant#STRATEGY_CHAIN},*/private Integer strategy;/*** 流控效果。* 0:默认(直接拒绝){@link RuleConstant#CONTROL_BEHAVIOR_DEFAULT},* 1: Warm Up(冷启动){@link RuleConstant#CONTROL_BEHAVIOR_WARM_UP},* 2:匀速排队{@link RuleConstant#CONTROL_BEHAVIOR_RATE_LIMITER},* 3: 冷启动+匀速排队{@link RuleConstant#CONTROL_BEHAVIOR_WARM_UP_RATE_LIMITER},*/private Integer controlBehavior;/*** 流控效果为匀速排队时,队列的排队时间(单位:毫秒), 需要注意这里类型是int, 并且单位是毫秒*/private Integer maxQueueingTimeMs;/*** 冷启动到达最大值的时间(单位:秒)*/private Integer warmUpPeriodSec;/*** 创建时间*/private Date createTime;/*** 更新时间*/private Date updateTime;/*** 创建人(手机号码)*/private String createUser;/*** 更新人(手机号码)*/private String updateUser;}
5、系统启动的时候,进行加载。这里的主要方法是。
FlowRuleManager.loadRules(rules);
public void initFlowRule() {// 查询配置List<SystemSentinelConfigDO> list = systemSentinelConfigService.selectList();if (CollectionUtils.isEmpty(list)) {log.error("sentinel限流配置为空!!!!");return;}List<FlowRule> rules = new ArrayList<>();for (SystemSentinelConfigDO configDO : list) {rules.add(SentinelUtil.getFlowRule(configDO));}// 加载FlowRuleManager.loadRules(rules);}
SentinelUtil工具类:
public class SentinelUtil {public static void update(List<SystemSentinelConfigDO> list) {List<FlowRule> rules = new ArrayList<>();for (SystemSentinelConfigDO configDO : list) {rules.add(SentinelUtil.getFlowRule(configDO));}// 加载FlowRuleManager.loadRules(rules);}public static List<FlowRule> getFlowRuleList() {return FlowRuleManager.getRules();}public static FlowRule getFlowRule(SystemSentinelConfigDO configDO) {// 限流规则FlowRule rateRule = new FlowRule();// 设置资源名,即流量控制规则的作用对象rateRule.setResource(configDO.getResource());// 设置限流阈值rateRule.setCount(configDO.getCount());// 限流阈值类型rateRule.setGrade(configDO.getGrade());// 流控模式rateRule.setStrategy(configDO.getStrategy());// 流控效果rateRule.setControlBehavior(configDO.getControlBehavior());// 流控效果为匀速排队时,队列的排队时间(单位:毫秒), 需要注意这里类型是int, 并且单位是毫秒if (configDO.getMaxQueueingTimeMs() != null && configDO.getMaxQueueingTimeMs() != 0) {rateRule.setMaxQueueingTimeMs(configDO.getMaxQueueingTimeMs());}// 冷启动到达最大值的时间(单位:秒)if (configDO.getWarmUpPeriodSec() != null && configDO.getWarmUpPeriodSec() != 0) {rateRule.setWarmUpPeriodSec(configDO.getWarmUpPeriodSec());}return rateRule;}
}
6、通过update方法动态更新:
@Transactional@Overridepublic void update(SystemSentinelConfigDO configDO) {ValidateUtils.checkBlank(configDO.getResource());SystemSentinelConfigDO dbDO = systemSentinelConfigDao.selectByResource(configDO.getResource());if (dbDO != null) {configDO.setId(dbDO.getId());systemSentinelConfigDao.update(configDO);}// 全部更新SentinelUtil.update(systemSentinelConfigDao.selectList(null));}
7、定义限流通用返回, 这里的名称是standardLimit
@Slf4j
public class StandardBlockHandler {/*** 标准对接, 触发限流时返回*/public static ThirdResponseDTO<Object> standardLimit(BlockException ex) {return ResponseThirdHelper.failResponse(ResultCodeThirdEnum.THIRD_STANDARD_RATE_LIMIT);}
}
8、Controller接口通过@SentinelResource指定资源名称,以及触发限流时的返回消息
@RestController
@RequestMapping("/third/archives")
@Slf4j
@RequiredArgsConstructor
public class StandardThirdArchivesController {private final StandardArchivesService standardArchivesService;/*** 新增/编辑建筑类节点(片区/楼栋/单元/住家)*/@PostMapping("/structure/addEdit")@SentinelResource(value = "standard:structure:add", blockHandler = "standardLimit", blockHandlerClass = {StandardBlockHandler.class})public ThirdResponseDTO<StandardStructureAddRes> addEditStructure(@RequestBody @Validated ThirdRequestDTO<StandardStructureAddReq> req) {return ResponseThirdHelper.successResponse(standardArchivesService.addEditStructure(req.getData(), req.getNeighNo()));}
}
四、热点参数限流
很遗憾,前面说的第三点【基于注解限流】的方式,没法具体到热点参数,因为我的项目的“入参”是有些复杂的对象,但热点参数只支持【基本类型】的入参。。。
1、第三点说的【基于注解限流】直接作废,重新来过
2、maven引入依赖
<!--sentinel核心包-->
<dependency><groupId>com.alibaba.csp</groupId><artifactId>sentinel-core</artifactId><version>${sentinel.version}</version>
</dependency>
<!--sentinel热点参数限流-->
<dependency><groupId>com.alibaba.csp</groupId><artifactId>sentinel-parameter-flow-control</artifactId><version>${sentinel.version}</version>
</dependency>
3、我这里的入参 ThirdRequestDTO 里面有个字段 clientId,是我需要限流的热点参数:
@Data
public class ThirdRequestDTO<T> {/*** 授权分配的clientId*/private String clientId;}
4、因为热点参数只支持QPS,于是可以配置的选项就很少了:
以下是配置表:standard_sentinel_config (需要注意sql查询的时候,desc, count要加单引号。你也可以给这2个参数重命名)
这里表示在duration_in_sec(秒)内,访问resource(资源)超过count (次数) 就进行限流,
如果是热点参数的话,则表示:
在duration_in_sec(秒)内,访问resource(资源)超过client_id_count(次数) 就进行限流,
DROP TABLE IF EXISTS `standard_sentinel_config`;
CREATE TABLE `standard_sentinel_config` (`id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '自增id',`resource` varchar(100) NOT NULL COMMENT '资源名称{@link SentinelResName}',`desc` varchar(100) NOT NULL COMMENT '资源描述',`duration_in_sec` int NOT NULL COMMENT '限流的单位时间(秒)',`count` int NOT NULL COMMENT '限流参数',`client_id_count` int NOT NULL COMMENT '针对clientId限流参数',`create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',`update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '更新时间',create_user VARCHAR(50) DEFAULT NULL COMMENT '创建人' ,update_user VARCHAR(50) DEFAULT NULL COMMENT '更新人' ,PRIMARY KEY (`id`),UNIQUE KEY `resource_key`(`resource`)
) ENGINE=InnoDB AUTO_INCREMENT = 1 DEFAULT CHARSET=utf8 COMMENT='标准对接限流';-- 公共字典
INSERT INTO `standard_sentinel_config` (`resource`, `desc`, `duration_in_sec`, `count`, `client_id_count`) VALUES
('standard:archives', '标准对接--档案类', 1, 60, 10),
('standard:query:archives', '标准对接--查询档案类', 1, 60, 10),
('standard:other', '标准对接--其它', 1, 60, 10);
5、StandardSentinelConfigDO对象
@Data
public class StandardSentinelConfigDO {/*** 主键id*/private Long id;/*** 资源名称{@link SentinelResName}*/private String resource;/*** 资源描述*/private String desc;/*** 限流的单位时间(秒)*/private Integer durationInSec;/*** 限流参数*/private Integer count;/*** 针对clientId限流参数*/private Integer clientIdCount;/*** 创建时间*/private Date createTime;/*** 更新时间*/private Date updateTime;/*** 创建人(手机号码)*/private String createUser;/*** 更新人(手机号码)*/private String updateUser;}
6、系统启动的时候,加载规则:(主要是通过ParamFlowRuleManager.loadRules)
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
import com.alibaba.csp.sentinel.slots.block.flow.param.ParamFlowItem;
import com.alibaba.csp.sentinel.slots.block.flow.param.ParamFlowRule;
import com.alibaba.csp.sentinel.slots.block.flow.param.ParamFlowRuleManager;
import com.whysu.scd.base.common.enums.ResultCodeEnum;
import com.whysu.scd.base.common.exception.BusinessException;
import com.whysu.scd.base.common.util.ValidateUtils;
import com.whysu.scd.module.standard.dao.StandardPlatformInfoDao;
import com.whysu.scd.module.standard.dao.StandardSentinelConfigDao;
import com.whysu.scd.module.standard.entity.StandardSentinelConfigDO;
import com.whysu.scd.module.standard.service.StandardSentinelService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.stereotype.Service;import java.util.ArrayList;
import java.util.List;/*** 标准对接--限流*/
@Slf4j
@Service
@RequiredArgsConstructor
public class StandardSentinelServiceImpl implements StandardSentinelService {private final StandardPlatformInfoDao standardPlatformInfoDao;private final StandardSentinelConfigDao standardSentinelConfigDao;@Overridepublic void init() {// 查询sentinel配置List<StandardSentinelConfigDO> configList = standardSentinelConfigDao.selectList(null);if (CollectionUtils.isEmpty(configList)) {log.error("标准对接sentinel配置为空!");return;}// 查询所有的clientIdList<String> clientIdList = standardPlatformInfoDao.selectClientIdList2();List<ParamFlowRule> ruleList = new ArrayList<>();for (StandardSentinelConfigDO configDO : configList) {// 资源名ParamFlowRule rule = new ParamFlowRule(configDO.getResource());// 指定当前 rule 对应的热点参数索引rule.setParamIdx(0);// 限流的维度,该策略针对 QPS 限流rule.setGrade(RuleConstant.FLOW_GRADE_QPS);// 限流的单位时间(秒)rule.setDurationInSec(configDO.getDurationInSec());// 未使用指定热点参数时,该资源限流大小为50rule.setCount(configDO.getCount());// 热点参数限流if (CollectionUtils.isNotEmpty(clientIdList)) {List<ParamFlowItem> itemList = new ArrayList<>();for (String clientId : clientIdList) {// item1 设置了clientId的限流,单位时间(DurationInSec)内只能访问10次ParamFlowItem item1 = new ParamFlowItem().setObject(clientId) // 热点参数 value.setClassType(String.class.getName()) // 热点参数数据类型.setCount(configDO.getClientIdCount()); // 针对该value的限流值itemList.add(item1);}rule.setParamFlowItemList(itemList);}ruleList.add(rule);}// 加载ParamFlowRuleManager.loadRules(ruleList);}@Overridepublic void update(StandardSentinelConfigDO configDO) {ValidateUtils.checkBlank(configDO.getResource());// 查询StandardSentinelConfigDO dbDO = standardSentinelConfigDao.selectByResource(configDO.getResource());if (dbDO == null) {throw new BusinessException(ResultCodeEnum.PUTIAN_PARAM_ERROR);}// 更新configDO.setId(dbDO.getId());standardSentinelConfigDao.update(configDO);// 重新初始化init();}@Overridepublic List<ParamFlowRule> getRuleList() {return ParamFlowRuleManager.getRules();}@Overridepublic List<ParamFlowRule> getRuleOfResource(String resource) {return ParamFlowRuleManager.getRulesOfResource(resource);}
}
7、定义了一个注解:StandardSignature
@Documented
@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface StandardSignature {/*** 签名算法版本 2.0*/String version() default ThirdConstant.SignatureVersion.TWO;/*** 接口权限*/StandardOpenTypeEnum[] openType();/*** data参数是否不能为空(默认为true表示不能为空)*/boolean dataNotNull() default true;
}
8、针对该注解切面:
这里的核心代码是:
Entry entry = null;
try {// 调用限流entry = SphU.entry(resourceName, EntryType.IN, 1, clientId);// 业务代码...return proceedingJoinPoint.proceed();
} catch (BlockException e) {// 接口限流return ResponseThirdHelper.failResponse(ResultCodeThirdEnum.THIRD_STANDARD_RATE_LIMIT);
} finally {if (entry != null) {entry.exit(1, clientId);}
}
有2个入参:resourceName和clientId
其中clientId是入参ThirdRequestDTO里面的。
resourceName是通过注解指定的StandardSignature里的opernType参数,查询出来的:
/*** key: 标准对接接口,value: sentinel限流资源名称{@link SentinelResName}*/public static final HashMap<StandardOpenTypeEnum, String> SENTINEL_RESOURCE = new HashMap<StandardOpenTypeEnum, String>() {{// 标准对接-档案类put(StandardOpenTypeEnum.ARCHIVES, SentinelResName.STANDARD_ARCHIVES);put(StandardOpenTypeEnum.ARCHIVES_GET_CALL_NO_FIRST4, SentinelResName.STANDARD_ARCHIVES);put(StandardOpenTypeEnum.ARCHIVES_STRUCT_ADD, SentinelResName.STANDARD_ARCHIVES);put(StandardOpenTypeEnum.ARCHIVES_STRUCT_DELETE, SentinelResName.STANDARD_ARCHIVES);put(StandardOpenTypeEnum.ARCHIVES_UPLOAD_FACE_FILE, SentinelResName.STANDARD_ARCHIVES);put(StandardOpenTypeEnum.ARCHIVES_CHECK_FACE, SentinelResName.STANDARD_ARCHIVES);put(StandardOpenTypeEnum.ARCHIVES_HOUSEHOLD_ADD, SentinelResName.STANDARD_ARCHIVES);put(StandardOpenTypeEnum.ARCHIVES_HOUSEHOLD_DELETE, SentinelResName.STANDARD_ARCHIVES);put(StandardOpenTypeEnum.ARCHIVES_VISITOR_ADD, SentinelResName.STANDARD_ARCHIVES);put(StandardOpenTypeEnum.ARCHIVES_VISITOR_DELETE, SentinelResName.STANDARD_ARCHIVES);// 标准对接-查询档案类put(StandardOpenTypeEnum.QUERY_ARCHIVES, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_NEIGH, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_TREE_WHOLE, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_TREE_PAGE, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_AREA, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_BUILDING, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_UNIT, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_HOUSE, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_DEVICE, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_DEVICE_STATUS, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_PARKING_INFO, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_PARKING_ROAD_INFO, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_PARKING_CAR_INFO, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_PERSON_INFO, SentinelResName.STANDARD_QUERY_ARCHIVES);put(StandardOpenTypeEnum.QUERY_ARCHIVES_DEVICE_INFO, SentinelResName.STANDARD_QUERY_ARCHIVES);// 标准对接-其它put(StandardOpenTypeEnum.OTHER, SentinelResName.STANDARD_OTHER);put(StandardOpenTypeEnum.OTHER_REMOTE_OPEN, SentinelResName.STANDARD_OTHER);put(StandardOpenTypeEnum.OTHER_BLOCK_CONTROL, SentinelResName.STANDARD_OTHER);put(StandardOpenTypeEnum.OTHER_SEND_TEMPORARY_PWD, SentinelResName.STANDARD_OTHER);}};
其中SentinelResName (对应standard_sentinel_config表的resources字段)
public class SentinelResName {/*** 标准对接--档案类*/public static final String STANDARD_ARCHIVES = "standard:archives";/*** 标准对接--查询档案类*/public static final String STANDARD_QUERY_ARCHIVES = "standard:query:archives";/*** 标准对接--其它*/public static final String STANDARD_OTHER = "standard:other";}
OpenType是用的枚举:
import com.whysu.scd.module.standard.dto.StandardTreeNode;
import lombok.AllArgsConstructor;
import lombok.Getter;
import lombok.experimental.Accessors;import java.util.ArrayList;
import java.util.List;/*** 允许第三方调用的接口类型集合*/
@Getter
@Accessors(fluent = true)
@AllArgsConstructor
public enum StandardOpenTypeEnum {// 全部接口ALL(1, 0, "全部接口"),// 二级ARCHIVES(2, ALL.type, "档案类"),QUERY_ARCHIVES(3, ALL.type, "查询档案类"),QUERY_RECORD(4, ALL.type, "查询记录"),OTHER(5, ALL.type, "其它接口"),// 档案类,取值范围:100到199ARCHIVES_GET_CALL_NO_FIRST4(100, ARCHIVES.type, "获取呼叫号码前四位"),ARCHIVES_STRUCT_ADD(101, ARCHIVES.type, "新增/编辑片区/楼栋/单元/住家"),ARCHIVES_STRUCT_DELETE(102, ARCHIVES.type, "删除片区/楼栋/单元/住家"),ARCHIVES_UPLOAD_FACE_FILE(103, ARCHIVES.type, "上传人脸"),ARCHIVES_CHECK_FACE(104, ARCHIVES.type, "人脸质量校验"),ARCHIVES_HOUSEHOLD_ADD(105, ARCHIVES.type, "新增/编辑住户(会进行人脸质量校验)"),ARCHIVES_HOUSEHOLD_DELETE(106, ARCHIVES.type, "删除住户"),ARCHIVES_VISITOR_ADD(107, ARCHIVES.type, "新增/编辑访客(会进行人脸质量校验)"),ARCHIVES_VISITOR_DELETE(108, ARCHIVES.type, "删除访客"),// 查询档案类,取值范围:200到299QUERY_ARCHIVES_NEIGH(200, QUERY_ARCHIVES.type, "查询小区档案"),QUERY_ARCHIVES_TREE_WHOLE(201, QUERY_ARCHIVES.type, "查询树结构(全部)"),QUERY_ARCHIVES_TREE_PAGE(202, QUERY_ARCHIVES.type, "查询树结构(分页)(小区片区楼栋单元住家地点)"),QUERY_ARCHIVES_AREA(203, QUERY_ARCHIVES.type, "查询片区档案(分页)"),QUERY_ARCHIVES_BUILDING(204, QUERY_ARCHIVES.type, "查询楼栋档案(分页)"),QUERY_ARCHIVES_UNIT(205, QUERY_ARCHIVES.type, "查询单元档案(分页)"),QUERY_ARCHIVES_HOUSE(206, QUERY_ARCHIVES.type, "查询房屋档案(分页)"),QUERY_ARCHIVES_DEVICE(207, QUERY_ARCHIVES.type, "查询设备档案(分页)"),QUERY_ARCHIVES_DEVICE_STATUS(208, QUERY_ARCHIVES.type, "查询设备在离线状态(分页)"),QUERY_ARCHIVES_PARKING_INFO(209, QUERY_ARCHIVES.type, "查询车场档案"),QUERY_ARCHIVES_PARKING_ROAD_INFO(210, QUERY_ARCHIVES.type, "查询车道档案(分页)"),QUERY_ARCHIVES_PARKING_CAR_INFO(211, QUERY_ARCHIVES.type, "查询车辆档案(分页)"),QUERY_ARCHIVES_PERSON_INFO(212, QUERY_ARCHIVES.type, "查询人员档案(住户/访客/工作人员)(分页)"),QUERY_ARCHIVES_DEVICE_INFO(213, QUERY_ARCHIVES.type, "查询设备详情"),// 查询记录类,取值范围:300到399QUERY_RECORD_ACCESS_PAGE(300, QUERY_RECORD.type, "查询门禁记录(分页,不能跨月查询)"),QUERY_RECORD_TALK_PAGE(301, QUERY_RECORD.type, "查询对讲记录(分页,不能跨月查询)"),QUERY_RECORD_CAR_PAGE(302, QUERY_RECORD.type, "查询车辆出入记录(分页,不能跨月查询)"),// 其它接口,取值范围:400到499OTHER_REMOTE_OPEN(400, OTHER.type, "第三方远程开门"),OTHER_BLOCK_CONTROL(401, OTHER.type, "锁门禁权限冻结/解冻"),OTHER_SEND_TEMPORARY_PWD(402, OTHER.type, "锁临时密码下发"),;/*** 允许第三方调用的接口类型*/private final int type;/*** 父节点接口类型*/private final int parentType;/*** 描述*/private final String desc;}
完整的切面代码如下:
import com.alibaba.csp.sentinel.Entry;
import com.alibaba.csp.sentinel.EntryType;
import com.alibaba.csp.sentinel.SphU;
import com.alibaba.csp.sentinel.slots.block.BlockException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.whysu.scd.base.common.constant.i18n.third.standard.MessageThirdStandard;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.context.annotation.Configuration;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.HashSet;
import java.util.List;
import java.util.Set;/*** 标准对接-数字签名校验切面*/
@Aspect
@Component
@Slf4j
@Configuration
public class StandardSignatureValidateAspect {@Pointcut(value = "@annotation(com.whysu.scd.module.standard.annotation.StandardSignature)")public void signaturePointcut() {}@Around("signaturePointcut()")public Object around(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {MethodSignature signature = (MethodSignature) proceedingJoinPoint.getSignature();StandardSignature annotation = signature.getMethod().getAnnotation(StandardSignature.class);Object[] args = proceedingJoinPoint.getArgs();for (Object arg : args) {if (arg instanceof ThirdRequestDTO) {ThirdRequestDTO requestDTO = (ThirdRequestDTO) arg;String clientId = requestDTO.getClientId();// 根据openType获取资源名称Set<String> resourceSet = new HashSet<>();for (StandardOpenTypeEnum openTypeEnum : annotation.openType()) {String resourceName = StandardMap.SENTINEL_RESOURCE.get(openTypeEnum);if (StringUtils.isNotBlank(resourceName)) {resourceSet.add(resourceName);}}// 限流判断for (String resourceName : resourceSet) {Entry entry = null;try {// 调用限流entry = SphU.entry(resourceName, EntryType.IN, 1, clientId);// 业务代码...return proceedingJoinPoint.proceed();} catch (BlockException e) {// 接口限流return ResponseThirdHelper.failResponse(ResultCodeThirdEnum.THIRD_STANDARD_RATE_LIMIT);} finally {if (entry != null) {entry.exit(1, clientId);}}}}}return null;}@AfterReturning(returning = "ret", pointcut = "signaturePointcut()")public void doAfterReturning(Object ret) {// 处理完请求,返回内容log.info("标准对接请求scd, 响应参数为:{}", JacksonUtils.toJson(ret));}}
9、在Controller接口指定注解:
/*** 新增/编辑建筑类节点(片区/楼栋/单元/住家)*/@StandardSignature(openType = StandardOpenTypeEnum.ARCHIVES_STRUCT_ADD)@PostMapping("/structure/addEdit")public ThirdResponseDTO<StandardStructureAddRes> addEditStructure(@RequestBody @Validated ThirdRequestDTO<StandardStructureAddReq> req) {return ResponseThirdHelper.successResponse(standardArchivesService.addEditStructure(req.getData(), req.getNeighNo()));}
五、使用JMeter验证
首先声明这个工具我不大会用哈哈哈
1、先下载http://jmeter.apache.org/download_jmeter.cgi
2、解压,进入bin目录下执行jmeter.bat即可启动成功。
3、默认语言是英文,进入bin目录修改 jmeter.properties,添加如下配置即可改成中文
language=zh_CN
4、右键添加--线程(用户)-线程组
配置1秒并发100次
5、线程组右键:添加--取样器--HTTP请求
指定请求协议,地址,端口,方式,路径。可以直接在【消息体数据】里面复制json对象。
6、HTTP请求右键:添加--配置元件--HTTP信息头管理器
添加请求头:Content-Type application/json;charset=utf-8
7、HTTP请求右键:添加--断言--响应断言
我这里也不知道咋写,但是我知道我项目的响应码3011是对应限流。
我这里就设置成,响应文本,包含3011,就表示成功
8、线程组右键:添加--监听器--察看结果树
9、线程组右键:启动
查看结果树,被限流了哈哈哈哈^_^、、、。。。