Fix default data source lost when create migration job (apache#32735)

terrymanu · Aug 30, 2024 · 6f0d325 · 6f0d325
1 parent 677749f
commit 6f0d325
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/...sphere/data/pipeline/core/datasource/creator/ShardingSpherePipelineDataSourceCreator.java b/...sphere/data/pipeline/core/datasource/creator/ShardingSpherePipelineDataSourceCreator.java
@@ -72,9 +72,12 @@ private void removeAuthorityRuleConfiguration(final YamlRootConfiguration yamlRo
     }
 
     private void updateSingleRuleConfiguration(final YamlRootConfiguration yamlRootConfig) {
+        Optional<YamlSingleRuleConfiguration> originalSingleRuleConfig =
+                yamlRootConfig.getRules().stream().filter(YamlSingleRuleConfiguration.class::isInstance).map(YamlSingleRuleConfiguration.class::cast).findFirst();
         yamlRootConfig.getRules().removeIf(YamlSingleRuleConfiguration.class::isInstance);
         YamlSingleRuleConfiguration singleRuleConfig = new YamlSingleRuleConfiguration();
         singleRuleConfig.setTables(Collections.singletonList(SingleTableConstants.ALL_TABLES));
+        originalSingleRuleConfig.ifPresent(optional -> singleRuleConfig.setDefaultDataSource(optional.getDefaultDataSource()));
         yamlRootConfig.getRules().add(singleRuleConfig);
     }