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

using Enum replace of String

Browse Source
This commit is contained in:
fangjian0423
2018-12-28 14:43:56 +08:00
parent 0e75da3b8d
commit 6806cd77ac
3 changed files with 8 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
spring-cloud-alibaba-sentinel-datasource/src/main/java/org/springframework/cloud/alibaba/sentinel/datasource/config/AbstractDataSourceProperties.java
View File

@@ -19,7 +19,7 @@ import com.fasterxml.jackson.annotation.JsonIgnore;
public class AbstractDataSourceProperties {
private String dataType = "json";
private String ruleType;
private RuleType ruleType;
private String converterClass;
@JsonIgnore
protected String factoryBeanName;
@@ -36,11 +36,11 @@ public class AbstractDataSourceProperties {
this.dataType = dataType;
}
public String getRuleType() {
public RuleType getRuleType() {
return ruleType;
}
public void setRuleType(String ruleType) {
public void setRuleType(RuleType ruleType) {
this.ruleType = ruleType;
}
@@ -61,15 +61,10 @@ public class AbstractDataSourceProperties {
}
public void preCheck(String dataSourceName) {
if (!RuleType.getByName(this.getRuleType()).isPresent()) {
throw new IllegalArgumentException(
"[Sentinel Starter] DataSource " + dataSourceName
+ " get error ruleType [" + this.getRuleType() + "]");
}
}
public void postRegister(AbstractDataSource dataSource) {
switch (RuleType.getByName(this.getRuleType()).get()) {
switch (this.getRuleType()) {
case FLOW:
FlowRuleManager.register2Property(dataSource.getProperty());
break;

4
spring-cloud-alibaba-sentinel-datasource/src/main/java/org/springframework/cloud/alibaba/sentinel/datasource/config/NacosDataSourceProperties.java
View File

@@ -128,10 +128,10 @@ public class NacosDataSourceProperties extends AbstractDataSourceProperties {
.getProperty(SentinelDataSourceConstants.PROJECT_NAME) + "-" + type);
result.setGroupId("nacos-sentinel");
if (type.equals(RuleType.FLOW.getName())) {
result.setRuleType(RuleType.FLOW.getName());
result.setRuleType(RuleType.FLOW);
}
else {
result.setRuleType(RuleType.DEGRADE.getName());
result.setRuleType(RuleType.DEGRADE);
}
return result;
}