study/Spring-Cloud-Alibaba
1
0
Fork 0
mirror of https://gitee.com/mirrors/Spring-Cloud-Alibaba.git synced 2021-06-26 13:25:11 +08:00
Code Issues Releases Wiki Activity

check ParamFlowRule

Browse Source
This commit is contained in:
fangjian0423 2018-12-14 18:14:32 +08:00
parent 8e0239dae1
commit 2bb73f4524
1 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
spring-cloud-alibaba-sentinel/src/main/java/org/springframework/cloud/alibaba/sentinel/custom/SentinelDataSourceHandler.java
View File

@ -51,6 +51,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.degrade.DegradeRuleManager;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRule; 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.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.SystemRule;
import com.alibaba.csp.sentinel.slots.system.SystemRuleManager; import com.alibaba.csp.sentinel.slots.system.SystemRuleManager;
@ -71,7 +73,8 @@ public class SentinelDataSourceHandler {
private List<String> dataTypeList = Arrays.asList("json", "xml"); private List<String> dataTypeList = Arrays.asList("json", "xml");
private List<Class<? extends AbstractRule>> rulesList = Arrays.asList(FlowRule.class, private List<Class<? extends AbstractRule>> rulesList = Arrays.asList(FlowRule.class,
DegradeRule.class, SystemRule.class, AuthorityRule.class); DegradeRule.class, SystemRule.class, AuthorityRule.class,
ParamFlowRule.class);
private List<String> dataSourceBeanNameList = Collections private List<String> dataSourceBeanNameList = Collections
.synchronizedList(new ArrayList<String>()); .synchronizedList(new ArrayList<String>());
@ -163,9 +166,12 @@ public class SentinelDataSourceHandler {
else if (ruleType == SystemRule.class) { else if (ruleType == SystemRule.class) {
SystemRuleManager.register2Property(sentinelProperty); SystemRuleManager.register2Property(sentinelProperty);
} }
else { else if (ruleType == AuthorityRule.class) {
AuthorityRuleManager.register2Property(sentinelProperty); AuthorityRuleManager.register2Property(sentinelProperty);
} }
else {
ParamFlowRuleManager.register2Property(sentinelProperty);
}
} }
} }
} }
@ -338,9 +344,10 @@ public class SentinelDataSourceHandler {
private boolean checkAllRuleTypeSame(List convertedRuleList) { private boolean checkAllRuleTypeSame(List convertedRuleList) {
int matchCount = 0; int matchCount = 0;
for(Object rule : convertedRuleList) { for (Object rule : convertedRuleList) {
if(rulesList.contains(rule.getClass()) && rule.getClass() == convertedRuleList.get(0).getClass()) { if (rulesList.contains(rule.getClass())
matchCount ++; && rule.getClass() == convertedRuleList.get(0).getClass()) {
matchCount++;
} }
} }
return matchCount == convertedRuleList.size(); return matchCount == convertedRuleList.size();