This is an automated email from the ASF dual-hosted git repository.
sunnianjun pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git
The following commit(s) were added to refs/heads/master by this push:
new a584c52296c Rename StorageUnitNodeMapCreator (#28651)
a584c52296c is described below
commit a584c52296c72839b858a446b18ef92e2721ab33
Author: Liang Zhang <[email protected]>
AuthorDate: Fri Oct 6 23:06:35 2023 +0800
Rename StorageUnitNodeMapCreator (#28651)
* Rename StorageUnitNodeMapCreator
* Rename StorageUnitNodeMapCreator
* Rename StorageUnitNodeMapCreator
* Rename StorageUnitNodeMapCreator
---
.../DataSourceGeneratedDatabaseConfiguration.java | 4 +--
.../DataSourceProvidedDatabaseConfiguration.java | 7 ++--
.../database/resource/ResourceMetaData.java | 6 ++--
...apUtils.java => StorageUnitNodeMapCreator.java} | 42 +++++++---------------
.../manager/switcher/NewResourceSwitchManager.java | 6 ++--
.../manager/switcher/ResourceSwitchManager.java | 8 ++---
.../mode/manager/ContextManagerTest.java | 5 +--
.../YamlDatabaseConfigurationImportExecutor.java | 4 +--
8 files changed, 33 insertions(+), 49 deletions(-)
diff --git
a/infra/common/src/main/java/org/apache/shardingsphere/infra/config/database/impl/DataSourceGeneratedDatabaseConfiguration.java
b/infra/common/src/main/java/org/apache/shardingsphere/infra/config/database/impl/DataSourceGeneratedDatabaseConfiguration.java
index 715ad0bfdc7..8f1dd9fb209 100644
---
a/infra/common/src/main/java/org/apache/shardingsphere/infra/config/database/impl/DataSourceGeneratedDatabaseConfiguration.java
+++
b/infra/common/src/main/java/org/apache/shardingsphere/infra/config/database/impl/DataSourceGeneratedDatabaseConfiguration.java
@@ -27,7 +27,7 @@ import
org.apache.shardingsphere.infra.datasource.pool.props.domain.DataSourcePo
import
org.apache.shardingsphere.infra.metadata.database.resource.StorageResource;
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNode;
import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnit;
-import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapUtils;
+import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapCreator;
import javax.sql.DataSource;
import java.util.Collection;
@@ -52,7 +52,7 @@ public final class DataSourceGeneratedDatabaseConfiguration
implements DatabaseC
ruleConfigurations = ruleConfigs;
Map<String, DataSourcePoolProperties> dataSourcePoolPropertiesMap =
dataSourceConfigs.entrySet().stream()
.collect(Collectors.toMap(Entry::getKey, entry ->
DataSourcePoolPropertiesCreator.create(entry.getValue()), (oldValue,
currentValue) -> oldValue, LinkedHashMap::new));
- Map<String, StorageNode> storageUnitNodeMap =
StorageUnitNodeMapUtils.fromDataSourcePoolProperties(dataSourcePoolPropertiesMap);
+ Map<String, StorageNode> storageUnitNodeMap =
StorageUnitNodeMapCreator.create(dataSourcePoolPropertiesMap);
Map<StorageNode, DataSource> storageNodeDataSources =
getStorageNodeDataSourceMap(dataSourcePoolPropertiesMap, storageUnitNodeMap);
storageUnits = new LinkedHashMap<>(dataSourceConfigs.size(), 1F);
for (Entry<String, DataSourceConfiguration> entry :
dataSourceConfigs.entrySet()) {
diff --git
a/infra/common/src/main/java/org/apache/shardingsphere/infra/config/database/impl/DataSourceProvidedDatabaseConfiguration.java
b/infra/common/src/main/java/org/apache/shardingsphere/infra/config/database/impl/DataSourceProvidedDatabaseConfiguration.java
index bc450dabfd5..1f1fd121fb5 100644
---
a/infra/common/src/main/java/org/apache/shardingsphere/infra/config/database/impl/DataSourceProvidedDatabaseConfiguration.java
+++
b/infra/common/src/main/java/org/apache/shardingsphere/infra/config/database/impl/DataSourceProvidedDatabaseConfiguration.java
@@ -26,7 +26,7 @@ import
org.apache.shardingsphere.infra.metadata.database.resource.StorageResourc
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNode;
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNodeAggregator;
import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnit;
-import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapUtils;
+import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapCreator;
import javax.sql.DataSource;
import java.util.Collection;
@@ -49,7 +49,8 @@ public final class DataSourceProvidedDatabaseConfiguration
implements DatabaseCo
public DataSourceProvidedDatabaseConfiguration(final Map<String,
DataSource> dataSources, final Collection<RuleConfiguration> ruleConfigs) {
this.ruleConfigurations = ruleConfigs;
- Map<String, StorageNode> storageUnitNodeMap =
StorageUnitNodeMapUtils.fromDataSources(dataSources);
+ Map<String, StorageNode> storageUnitNodeMap =
dataSources.keySet().stream()
+ .collect(Collectors.toMap(each -> each, StorageNode::new,
(oldValue, currentValue) -> oldValue, LinkedHashMap::new));
Map<StorageNode, DataSource> storageNodeDataSources =
StorageNodeAggregator.aggregateDataSources(dataSources);
Map<String, DataSourcePoolProperties> dataSourcePoolPropertiesMap =
createDataSourcePoolPropertiesMap(dataSources);
storageUnits = new LinkedHashMap<>(dataSourcePoolPropertiesMap.size(),
1F);
@@ -66,7 +67,7 @@ public final class DataSourceProvidedDatabaseConfiguration
implements DatabaseCo
final
Collection<RuleConfiguration> ruleConfigs, final Map<String,
DataSourcePoolProperties> dataSourcePoolPropertiesMap) {
this.storageResource = storageResource;
this.ruleConfigurations = ruleConfigs;
- Map<String, StorageNode> storageUnitNodeMap =
StorageUnitNodeMapUtils.fromDataSourcePoolProperties(dataSourcePoolPropertiesMap);
+ Map<String, StorageNode> storageUnitNodeMap =
StorageUnitNodeMapCreator.create(dataSourcePoolPropertiesMap);
Map<StorageNode, DataSource> storageNodeDataSources =
storageResource.getDataSources();
storageUnits = new LinkedHashMap<>(dataSourcePoolPropertiesMap.size(),
1F);
for (Entry<String, DataSourcePoolProperties> entry :
dataSourcePoolPropertiesMap.entrySet()) {
diff --git
a/infra/common/src/main/java/org/apache/shardingsphere/infra/metadata/database/resource/ResourceMetaData.java
b/infra/common/src/main/java/org/apache/shardingsphere/infra/metadata/database/resource/ResourceMetaData.java
index ba0432523b2..97962e66f60 100644
---
a/infra/common/src/main/java/org/apache/shardingsphere/infra/metadata/database/resource/ResourceMetaData.java
+++
b/infra/common/src/main/java/org/apache/shardingsphere/infra/metadata/database/resource/ResourceMetaData.java
@@ -24,7 +24,6 @@ import
org.apache.shardingsphere.infra.datasource.pool.props.domain.DataSourcePo
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNode;
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNodeAggregator;
import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnit;
-import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapUtils;
import javax.sql.DataSource;
import java.util.Collection;
@@ -47,11 +46,12 @@ public final class ResourceMetaData {
public ResourceMetaData(final Map<String, DataSource> dataSources) {
this.dataSources =
StorageNodeAggregator.aggregateDataSources(dataSources);
- Map<String, StorageNode> storageNodes =
StorageUnitNodeMapUtils.fromDataSources(dataSources);
+ Map<String, StorageNode> storageUnitNodeMap =
dataSources.keySet().stream()
+ .collect(Collectors.toMap(each -> each, StorageNode::new,
(oldValue, currentValue) -> oldValue, LinkedHashMap::new));
Map<String, DataSourcePoolProperties> dataSourcePoolPropsMap =
dataSources.entrySet().stream().collect(
Collectors.toMap(Entry::getKey, entry ->
DataSourcePoolPropertiesCreator.create(entry.getValue()), (oldValue,
currentValue) -> oldValue, LinkedHashMap::new));
storageUnits = new LinkedHashMap<>();
- for (Entry<String, StorageNode> entry : storageNodes.entrySet()) {
+ for (Entry<String, StorageNode> entry : storageUnitNodeMap.entrySet())
{
storageUnits.put(entry.getKey(), new StorageUnit(entry.getValue(),
dataSourcePoolPropsMap.get(entry.getKey()),
dataSources.get(entry.getValue().getName())));
}
}
diff --git
a/infra/common/src/main/java/org/apache/shardingsphere/infra/metadata/database/resource/unit/StorageUnitNodeMapUtils.java
b/infra/common/src/main/java/org/apache/shardingsphere/infra/metadata/database/resource/unit/StorageUnitNodeMapCreator.java
similarity index 58%
rename from
infra/common/src/main/java/org/apache/shardingsphere/infra/metadata/database/resource/unit/StorageUnitNodeMapUtils.java
rename to
infra/common/src/main/java/org/apache/shardingsphere/infra/metadata/database/resource/unit/StorageUnitNodeMapCreator.java
index 2be8758a117..199a573a2f2 100644
---
a/infra/common/src/main/java/org/apache/shardingsphere/infra/metadata/database/resource/unit/StorageUnitNodeMapUtils.java
+++
b/infra/common/src/main/java/org/apache/shardingsphere/infra/metadata/database/resource/unit/StorageUnitNodeMapCreator.java
@@ -27,57 +27,39 @@ import
org.apache.shardingsphere.infra.database.core.type.DatabaseTypeRegistry;
import
org.apache.shardingsphere.infra.datasource.pool.props.domain.DataSourcePoolProperties;
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNode;
-import javax.sql.DataSource;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.stream.Collectors;
/**
- * Storage unit node map utility class.
+ * Storage unit node map creator.
*/
@NoArgsConstructor(access = AccessLevel.PRIVATE)
-public final class StorageUnitNodeMapUtils {
+public final class StorageUnitNodeMapCreator {
/**
- * Get storage unit node map from data sources.
- *
- * @param dataSources data sources
- * @return storage unit node mappers
- */
- public static Map<String, StorageNode> fromDataSources(final Map<String,
DataSource> dataSources) {
- return dataSources.entrySet().stream()
- .collect(Collectors.toMap(Entry::getKey, entry -> new
StorageNode(entry.getKey()), (oldValue, currentValue) -> currentValue,
LinkedHashMap::new));
- }
-
- /**
- * Get storage unit node map from data source pool properties.
+ * Create storage unit node map.
*
* @param propsMap data source pool properties map
- * @return storage unit node mappers
+ * @return storage unit node map
*/
- public static Map<String, StorageNode> fromDataSourcePoolProperties(final
Map<String, DataSourcePoolProperties> propsMap) {
- Map<String, StorageNode> result = new LinkedHashMap<>();
- for (Entry<String, DataSourcePoolProperties> entry :
propsMap.entrySet()) {
- String storageUnitName = entry.getKey();
- result.put(storageUnitName,
fromDataSourcePoolProperties(storageUnitName, entry.getValue()));
- }
- return result;
+ public static Map<String, StorageNode> create(final Map<String,
DataSourcePoolProperties> propsMap) {
+ return
propsMap.entrySet().stream().collect(Collectors.toMap(Entry::getKey, entry ->
create(entry.getKey(), entry.getValue()), (oldValue, currentValue) -> oldValue,
LinkedHashMap::new));
}
- private static StorageNode fromDataSourcePoolProperties(final String
storageUnitName, final DataSourcePoolProperties props) {
+ private static StorageNode create(final String storageUnitName, final
DataSourcePoolProperties props) {
Map<String, Object> standardProps =
props.getConnectionPropertySynonyms().getStandardProperties();
- String url = standardProps.get("url").toString();
- boolean isInstanceConnectionAvailable = new
DatabaseTypeRegistry(DatabaseTypeFactory.get(url)).getDialectDatabaseMetaData().isInstanceConnectionAvailable();
- return getStorageNode(storageUnitName, url,
standardProps.get("username").toString(), isInstanceConnectionAvailable);
+ return create(storageUnitName, standardProps.get("url").toString(),
standardProps.get("username").toString());
}
- private static StorageNode getStorageNode(final String dataSourceName,
final String url, final String username, final boolean
isInstanceConnectionAvailable) {
+ private static StorageNode create(final String storageUnitName, final
String url, final String username) {
+ boolean isInstanceConnectionAvailable = new
DatabaseTypeRegistry(DatabaseTypeFactory.get(url)).getDialectDatabaseMetaData().isInstanceConnectionAvailable();
try {
JdbcUrl jdbcUrl = new StandardJdbcUrlParser().parse(url);
- return isInstanceConnectionAvailable ? new
StorageNode(jdbcUrl.getHostname(), jdbcUrl.getPort(), username) : new
StorageNode(dataSourceName);
+ return isInstanceConnectionAvailable ? new
StorageNode(jdbcUrl.getHostname(), jdbcUrl.getPort(), username) : new
StorageNode(storageUnitName);
} catch (final UnrecognizedDatabaseURLException ex) {
- return new StorageNode(dataSourceName);
+ return new StorageNode(storageUnitName);
}
}
}
diff --git
a/mode/core/src/main/java/org/apache/shardingsphere/mode/manager/switcher/NewResourceSwitchManager.java
b/mode/core/src/main/java/org/apache/shardingsphere/mode/manager/switcher/NewResourceSwitchManager.java
index 347affcf80f..7f34a2e3ae4 100644
---
a/mode/core/src/main/java/org/apache/shardingsphere/mode/manager/switcher/NewResourceSwitchManager.java
+++
b/mode/core/src/main/java/org/apache/shardingsphere/mode/manager/switcher/NewResourceSwitchManager.java
@@ -23,7 +23,7 @@ import
org.apache.shardingsphere.infra.metadata.database.resource.ResourceMetaDa
import
org.apache.shardingsphere.infra.metadata.database.resource.StorageResource;
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNode;
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNodeAggregator;
-import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapUtils;
+import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapCreator;
import javax.sql.DataSource;
import java.util.Collections;
@@ -49,7 +49,7 @@ public final class NewResourceSwitchManager {
Map<String, DataSourcePoolProperties> mergedPropsMap = new
LinkedHashMap<>(resourceMetaData.getStorageUnits().entrySet().stream()
.collect(Collectors.toMap(Entry::getKey, entry ->
entry.getValue().getDataSourcePoolProperties(), (oldValue, currentValue) ->
oldValue, LinkedHashMap::new)));
mergedPropsMap.putAll(storageUnitDataSourcePoolPropsMap);
- Map<String, StorageNode> toBeCreatedStorageUintNodeMap =
StorageUnitNodeMapUtils.fromDataSourcePoolProperties(storageUnitDataSourcePoolPropsMap);
+ Map<String, StorageNode> toBeCreatedStorageUintNodeMap =
StorageUnitNodeMapCreator.create(storageUnitDataSourcePoolPropsMap);
Map<StorageNode, DataSourcePoolProperties> dataSourcePoolPropsMap =
StorageNodeAggregator.aggregateDataSourcePoolProperties(storageUnitDataSourcePoolPropsMap);
return new SwitchingResource(resourceMetaData,
getRegisterNewStorageResource(resourceMetaData, toBeCreatedStorageUintNodeMap,
dataSourcePoolPropsMap),
new StorageResource(Collections.emptyMap(),
Collections.emptyMap()), mergedPropsMap);
@@ -77,7 +77,7 @@ public final class NewResourceSwitchManager {
Map<String, DataSourcePoolProperties> mergedDataSourcePoolPropsMap =
new LinkedHashMap<>(resourceMetaData.getStorageUnits().entrySet().stream()
.collect(Collectors.toMap(Entry::getKey, entry ->
entry.getValue().getDataSourcePoolProperties(), (oldValue, currentValue) ->
oldValue, LinkedHashMap::new)));
mergedDataSourcePoolPropsMap.putAll(propsMap);
- Map<String, StorageNode> toBeAlteredStorageUintNodeMap =
StorageUnitNodeMapUtils.fromDataSourcePoolProperties(mergedDataSourcePoolPropsMap);
+ Map<String, StorageNode> toBeAlteredStorageUintNodeMap =
StorageUnitNodeMapCreator.create(mergedDataSourcePoolPropsMap);
Map<StorageNode, DataSourcePoolProperties> dataSourcePoolPropsMap =
StorageNodeAggregator.aggregateDataSourcePoolProperties(mergedDataSourcePoolPropsMap);
return new SwitchingResource(resourceMetaData,
getAlterNewStorageResource(toBeAlteredStorageUintNodeMap,
dataSourcePoolPropsMap),
getStaleStorageResource(resourceMetaData,
toBeAlteredStorageUintNodeMap), mergedDataSourcePoolPropsMap);
diff --git
a/mode/core/src/main/java/org/apache/shardingsphere/mode/manager/switcher/ResourceSwitchManager.java
b/mode/core/src/main/java/org/apache/shardingsphere/mode/manager/switcher/ResourceSwitchManager.java
index 1120384feba..f979395fa59 100644
---
a/mode/core/src/main/java/org/apache/shardingsphere/mode/manager/switcher/ResourceSwitchManager.java
+++
b/mode/core/src/main/java/org/apache/shardingsphere/mode/manager/switcher/ResourceSwitchManager.java
@@ -25,7 +25,7 @@ import
org.apache.shardingsphere.infra.metadata.database.resource.StorageResourc
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNode;
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNodeAggregator;
import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnit;
-import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapUtils;
+import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapCreator;
import javax.sql.DataSource;
import java.util.Collection;
@@ -51,7 +51,7 @@ public final class ResourceSwitchManager {
Map<String, DataSourcePoolProperties> mergedPropsMap = new
LinkedHashMap<>(resourceMetaData.getStorageUnits().entrySet().stream()
.collect(Collectors.toMap(Entry::getKey, entry ->
entry.getValue().getDataSourcePoolProperties(), (oldValue, currentValue) ->
oldValue, LinkedHashMap::new)));
mergedPropsMap.putAll(toBeChangedPropsMap);
- Map<String, StorageNode> toBeChangedStorageUnitNodeMap =
StorageUnitNodeMapUtils.fromDataSourcePoolProperties(toBeChangedPropsMap);
+ Map<String, StorageNode> toBeChangedStorageUnitNodeMap =
StorageUnitNodeMapCreator.create(toBeChangedPropsMap);
Map<StorageNode, DataSourcePoolProperties> dataSourcePoolPropsMap =
StorageNodeAggregator.aggregateDataSourcePoolProperties(toBeChangedPropsMap);
StorageResource newStorageResource =
createNewStorageResource(resourceMetaData, toBeChangedStorageUnitNodeMap,
dataSourcePoolPropsMap);
StorageResource staleDataSources =
getStaleDataSources(resourceMetaData, toBeChangedStorageUnitNodeMap,
mergedPropsMap);
@@ -69,7 +69,7 @@ public final class ResourceSwitchManager {
Map<String, DataSourcePoolProperties>
mergedDataSourcePoolPropertiesMap = new
LinkedHashMap<>(resourceMetaData.getStorageUnits().entrySet().stream()
.collect(Collectors.toMap(Entry::getKey, entry ->
entry.getValue().getDataSourcePoolProperties(), (oldValue, currentValue) ->
oldValue, LinkedHashMap::new)));
mergedDataSourcePoolPropertiesMap.keySet().removeIf(toBeDeletedPropsMap::containsKey);
- Map<String, StorageNode> toRemovedStorageUnitNodeMap =
StorageUnitNodeMapUtils.fromDataSourcePoolProperties(toBeDeletedPropsMap);
+ Map<String, StorageNode> toRemovedStorageUnitNodeMap =
StorageUnitNodeMapCreator.create(toBeDeletedPropsMap);
return new SwitchingResource(resourceMetaData, new
StorageResource(Collections.emptyMap(), Collections.emptyMap()),
getToBeRemovedStaleDataSources(resourceMetaData,
toRemovedStorageUnitNodeMap), mergedDataSourcePoolPropertiesMap);
}
@@ -86,7 +86,7 @@ public final class ResourceSwitchManager {
.collect(Collectors.toMap(Entry::getKey, entry ->
entry.getValue().getDataSourcePoolProperties(), (oldValue, currentValue) ->
oldValue, LinkedHashMap::new)));
mergedDataSourcePoolPropertiesMap.keySet().removeIf(each ->
!toBeChangedPropsMap.containsKey(each));
mergedDataSourcePoolPropertiesMap.putAll(toBeChangedPropsMap);
- Map<String, StorageNode> toBeChangedStorageUnitNodeMap =
StorageUnitNodeMapUtils.fromDataSourcePoolProperties(toBeChangedPropsMap);
+ Map<String, StorageNode> toBeChangedStorageUnitNodeMap =
StorageUnitNodeMapCreator.create(toBeChangedPropsMap);
StorageResource staleStorageResource =
getStaleDataSources(resourceMetaData, toBeChangedStorageUnitNodeMap,
toBeChangedPropsMap);
staleStorageResource.getDataSources().putAll(getToBeDeletedDataSources(resourceMetaData.getDataSources(),
toBeChangedStorageUnitNodeMap.values()));
staleStorageResource.getStorageUnitNodeMap().putAll(
diff --git
a/mode/core/src/test/java/org/apache/shardingsphere/mode/manager/ContextManagerTest.java
b/mode/core/src/test/java/org/apache/shardingsphere/mode/manager/ContextManagerTest.java
index c8372f0f1fa..381362e62fc 100644
---
a/mode/core/src/test/java/org/apache/shardingsphere/mode/manager/ContextManagerTest.java
+++
b/mode/core/src/test/java/org/apache/shardingsphere/mode/manager/ContextManagerTest.java
@@ -31,7 +31,6 @@ import
org.apache.shardingsphere.infra.metadata.database.resource.ResourceMetaDa
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNode;
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNodeAggregator;
import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnit;
-import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapUtils;
import org.apache.shardingsphere.infra.metadata.database.rule.RuleMetaData;
import
org.apache.shardingsphere.infra.metadata.database.schema.model.ShardingSphereColumn;
import
org.apache.shardingsphere.infra.metadata.database.schema.model.ShardingSphereSchema;
@@ -60,6 +59,7 @@ import java.util.Map;
import java.util.Map.Entry;
import java.util.Optional;
import java.util.Properties;
+import java.util.stream.Collectors;
import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.MatcherAssert.assertThat;
@@ -253,7 +253,8 @@ class ContextManagerTest {
originalDataSources.put("ds_1", new MockedDataSource());
originalDataSources.put("ds_2", new MockedDataSource());
Map<StorageNode, DataSource> storageNodeDataSourceMap =
StorageNodeAggregator.aggregateDataSources(originalDataSources);
- Map<String, StorageNode> storageUnitNodeMap =
StorageUnitNodeMapUtils.fromDataSources(originalDataSources);
+ Map<String, StorageNode> storageUnitNodeMap =
originalDataSources.keySet().stream()
+ .collect(Collectors.toMap(each -> each, StorageNode::new,
(oldValue, currentValue) -> oldValue, LinkedHashMap::new));
Map<String, StorageUnit> storageUnits = new LinkedHashMap<>(2, 1F);
for (Entry<String, StorageNode> entry : storageUnitNodeMap.entrySet())
{
DataSourcePoolProperties dataSourcePoolProps =
mock(DataSourcePoolProperties.class, RETURNS_DEEP_STUBS);
diff --git
a/proxy/backend/core/src/main/java/org/apache/shardingsphere/proxy/backend/util/YamlDatabaseConfigurationImportExecutor.java
b/proxy/backend/core/src/main/java/org/apache/shardingsphere/proxy/backend/util/YamlDatabaseConfigurationImportExecutor.java
index a704756b65a..cdc0fa74557 100644
---
a/proxy/backend/core/src/main/java/org/apache/shardingsphere/proxy/backend/util/YamlDatabaseConfigurationImportExecutor.java
+++
b/proxy/backend/core/src/main/java/org/apache/shardingsphere/proxy/backend/util/YamlDatabaseConfigurationImportExecutor.java
@@ -45,7 +45,7 @@ import
org.apache.shardingsphere.infra.instance.InstanceContext;
import
org.apache.shardingsphere.infra.metadata.database.ShardingSphereDatabase;
import
org.apache.shardingsphere.infra.metadata.database.resource.node.StorageNode;
import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnit;
-import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapUtils;
+import
org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnitNodeMapCreator;
import org.apache.shardingsphere.infra.rule.ShardingSphereRule;
import
org.apache.shardingsphere.infra.yaml.config.pojo.rule.YamlRuleConfiguration;
import org.apache.shardingsphere.mask.api.config.MaskRuleConfiguration;
@@ -162,7 +162,7 @@ public final class YamlDatabaseConfigurationImportExecutor {
}
Map<String, StorageUnit> storageUnits =
ProxyContext.getInstance().getContextManager()
.getMetaDataContexts().getMetaData().getDatabase(databaseName).getResourceMetaData().getStorageUnits();
- Map<String, StorageNode> toBeAddedStorageNode =
StorageUnitNodeMapUtils.fromDataSourcePoolProperties(propsMap);
+ Map<String, StorageNode> toBeAddedStorageNode =
StorageUnitNodeMapCreator.create(propsMap);
for (Entry<String, DataSourcePoolProperties> entry :
propsMap.entrySet()) {
storageUnits.put(entry.getKey(), new
StorageUnit(toBeAddedStorageNode.get(entry.getKey()), entry.getValue(),
DataSourcePoolCreator.create(entry.getValue())));
}
