提交 73faf950 编写于 作者: T tristaZero

use getKeyGeneratorType()

上级 3dfa639c
......@@ -79,7 +79,7 @@ public class YamlShardingRuleConfiguration {
private YamlKeyGeneratorConfiguration getYamlKeyGeneratorConfiguration(final ShardingRuleConfiguration shardingRuleConfiguration) {
YamlKeyGeneratorConfiguration result = new YamlKeyGeneratorConfiguration();
String keyGeneratorClassName = shardingRuleConfiguration.getDefaultKeyGenerator().getClass().getName();
KeyGeneratorType keyGeneratorType = KeyGeneratorType.getBuiltinKeyGeneratorType(keyGeneratorClassName);
KeyGeneratorType keyGeneratorType = KeyGeneratorType.getKeyGeneratorType(keyGeneratorClassName);
if (null == keyGeneratorType) {
result.setClassName(keyGeneratorClassName);
} else {
......
......@@ -58,7 +58,7 @@ public class YamlTableRuleConfiguration {
private YamlKeyGeneratorConfiguration getYamlKeyGeneratorConfiguration(final TableRuleConfiguration tableRuleConfiguration) {
YamlKeyGeneratorConfiguration result = new YamlKeyGeneratorConfiguration();
String keyGeneratorClassName = tableRuleConfiguration.getKeyGenerator().getClass().getName();
KeyGeneratorType keyGeneratorType = KeyGeneratorType.getBuiltinKeyGeneratorType(keyGeneratorClassName);
KeyGeneratorType keyGeneratorType = KeyGeneratorType.getKeyGeneratorType(keyGeneratorClassName);
if (null == keyGeneratorType) {
result.setClassName(keyGeneratorClassName);
} else {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册