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
Show Stats 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;
}

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

@ -18,7 +18,6 @@ import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.boot.context.event.ApplicationStartedEvent;
import org.springframework.cloud.alibaba.sentinel.SentinelConstants;
import org.springframework.cloud.alibaba.sentinel.SentinelProperties;
import org.springframework.cloud.alibaba.sentinel.datasource.RuleType;
import org.springframework.cloud.alibaba.sentinel.datasource.SentinelDataSourceConstants;
import org.springframework.cloud.alibaba.sentinel.datasource.config.AbstractDataSourceProperties;
import org.springframework.cloud.alibaba.sentinel.datasource.config.DataSourcePropertiesConfiguration;
@ -199,7 +198,7 @@ public class SentinelDataSourceHandler {
// 'sentinel-{converterType}-{ruleType}-converter'
builder.addPropertyReference("converter",
"sentinel-" + propertyValue.toString() + "-"
+ dataSourceProperties.getRuleType()
+ dataSourceProperties.getRuleType().getName()
+ "-converter");
}
}
@ -220,7 +219,7 @@ public class SentinelDataSourceHandler {
.getBean(dataSourceName);
logAndCheckRuleType(newDataSource, dataSourceName,
RuleType.getByName(dataSourceProperties.getRuleType()).get().getClazz());
dataSourceProperties.getRuleType().getClazz());
// register property in RuleManager
dataSourceProperties.postRegister(newDataSource);