From 20741db009f319ab808c761680272fe6e41445ee Mon Sep 17 00:00:00 2001 From: fangjian0423 Date: Fri, 14 Dec 2018 18:10:49 +0800 Subject: [PATCH] Polish #115 check ParamFlowRule --- .../sentinel/custom/SentinelDataSourceHandler.java | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/spring-cloud-alibaba-sentinel/src/main/java/org/springframework/cloud/alibaba/sentinel/custom/SentinelDataSourceHandler.java b/spring-cloud-alibaba-sentinel/src/main/java/org/springframework/cloud/alibaba/sentinel/custom/SentinelDataSourceHandler.java index 66c44e4e..f93d889d 100644 --- a/spring-cloud-alibaba-sentinel/src/main/java/org/springframework/cloud/alibaba/sentinel/custom/SentinelDataSourceHandler.java +++ b/spring-cloud-alibaba-sentinel/src/main/java/org/springframework/cloud/alibaba/sentinel/custom/SentinelDataSourceHandler.java @@ -35,6 +35,8 @@ import com.alibaba.csp.sentinel.slots.block.degrade.DegradeRule; import com.alibaba.csp.sentinel.slots.block.degrade.DegradeRuleManager; import com.alibaba.csp.sentinel.slots.block.flow.FlowRule; import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager; +import com.alibaba.csp.sentinel.slots.block.flow.param.ParamFlowRule; +import com.alibaba.csp.sentinel.slots.block.flow.param.ParamFlowRuleManager; import com.alibaba.csp.sentinel.slots.system.SystemRule; import com.alibaba.csp.sentinel.slots.system.SystemRuleManager; @@ -55,7 +57,7 @@ public class SentinelDataSourceHandler { private List dataTypeList = Arrays.asList("json", "xml"); private List rulesList = Arrays.asList(FlowRule.class, DegradeRule.class, - SystemRule.class, AuthorityRule.class); + SystemRule.class, AuthorityRule.class, ParamFlowRule.class); private List dataSourceBeanNameList = Collections .synchronizedList(new ArrayList<>()); @@ -146,9 +148,12 @@ public class SentinelDataSourceHandler { else if (ruleType == SystemRule.class) { SystemRuleManager.register2Property(sentinelProperty); } - else { + else if (ruleType == AuthorityRule.class) { AuthorityRuleManager.register2Property(sentinelProperty); } + else { + ParamFlowRuleManager.register2Property(sentinelProperty); + } } }); }