Use provided dataSourceNames instead of usedDataSourceNames.

Ensure return logic datasource name.

Author: iamhucong

infra/route/src/main/java/org/apache/shardingsphere/infra/route/engine/tableless/TablelessRouteEngineFactory.java

@@ -28,7 +28,6 @@
 import org.apache.shardingsphere.infra.route.engine.tableless.type.broadcast.TablelessInstanceBroadcastRouteEngine;
 import org.apache.shardingsphere.infra.route.engine.tableless.type.ignore.TablelessIgnoreRouteEngine;
 import org.apache.shardingsphere.infra.route.engine.tableless.type.unicast.TablelessDataSourceUnicastRouteEngine;
-import org.apache.shardingsphere.infra.session.connection.ConnectionContext;
 import org.apache.shardingsphere.infra.session.query.QueryContext;
 import org.apache.shardingsphere.sql.parser.statement.core.statement.SQLStatement;
 import org.apache.shardingsphere.sql.parser.statement.core.statement.dal.AlterResourceGroupStatement;
@@ -82,7 +81,7 @@ public static TablelessRouteEngine newInstance(final QueryContext queryContext,
             return getDDLRouteEngine(queryContext.getSqlStatementContext(), database);
         }
         if (sqlStatement instanceof DMLStatement) {
-            return getDMLRouteEngine(queryContext.getSqlStatementContext(), queryContext.getConnectionContext());
+            return getDMLRouteEngine(queryContext.getSqlStatementContext());
         }
         return new TablelessIgnoreRouteEngine();
     }
@@ -135,9 +134,9 @@ private static TablelessRouteEngine getCursorRouteEngine(final SQLStatementConte
         return new TablelessIgnoreRouteEngine();
     }
 
-    private static TablelessRouteEngine getDMLRouteEngine(final SQLStatementContext sqlStatementContext, final ConnectionContext connectionContext) {
+    private static TablelessRouteEngine getDMLRouteEngine(final SQLStatementContext sqlStatementContext) {
         if (sqlStatementContext instanceof SelectStatementContext) {
-            return new TablelessDataSourceUnicastRouteEngine(connectionContext);
+            return new TablelessDataSourceUnicastRouteEngine();
         }
         return new TablelessIgnoreRouteEngine();
     }

infra/route/src/main/java/org/apache/shardingsphere/infra/route/engine/tableless/type/unicast/TablelessDataSourceUnicastRouteEngine.java

@@ -36,8 +36,6 @@
 @RequiredArgsConstructor
 public final class TablelessDataSourceUnicastRouteEngine implements TablelessRouteEngine {
 
-    private final ConnectionContext connectionContext;
-    
     @Override
     public RouteContext route(final RuleMetaData globalRuleMetaData, final Collection<String> aggregatedDataSources) {
         RouteContext result = new RouteContext();
@@ -51,7 +49,6 @@ private RouteMapper getDataSourceRouteMapper(final Collection<String> dataSource
     }
 
     private String getRandomDataSourceName(final Collection<String> dataSourceNames) {
-        Collection<String> usedDataSourceNames = connectionContext.getUsedDataSourceNames().isEmpty() ? dataSourceNames : connectionContext.getUsedDataSourceNames();
-        return new ArrayList<>(usedDataSourceNames).get(ThreadLocalRandom.current().nextInt(usedDataSourceNames.size()));
+        return new ArrayList<>(dataSourceNames).get(ThreadLocalRandom.current().nextInt(dataSourceNames.size()));
     }
 }

infra/route/src/test/java/org/apache/shardingsphere/infra/route/engine/tableless/type/unicast/TablelessDataSourceUnicastRouteEngineTest.java

@@ -39,9 +39,8 @@ class TablelessDataSourceUnicastRouteEngineTest {
 
     @Test
     void assertRouteWithoutUsedDataSourceNames() {
-        ConnectionContext connectionContext = new ConnectionContext(Collections::emptyList);
         Collection<String> aggregatedDataSources = Arrays.asList("foo_ds_1", "foo_ds_2");
-        RouteContext actual = new TablelessDataSourceUnicastRouteEngine(connectionContext).route(mock(RuleMetaData.class), aggregatedDataSources);
+        RouteContext actual = new TablelessDataSourceUnicastRouteEngine().route(mock(RuleMetaData.class), aggregatedDataSources);
         assertThat(actual.getRouteUnits().size(), is(1));
         RouteUnit routeUnit = actual.getRouteUnits().iterator().next();
         assertTrue(aggregatedDataSources.contains(routeUnit.getDataSourceMapper().getLogicName()));
@@ -51,9 +50,8 @@ void assertRouteWithoutUsedDataSourceNames() {
 
     @Test
     void assertRouteWithUsedDataSourceNames() {
-        ConnectionContext connectionContext = new ConnectionContext(() -> Collections.singleton("foo_ds_1"));
         Collection<String> aggregatedDataSources = Arrays.asList("foo_ds_1", "foo_ds_2");
-        RouteContext actual = new TablelessDataSourceUnicastRouteEngine(connectionContext).route(mock(RuleMetaData.class), aggregatedDataSources);
+        RouteContext actual = new TablelessDataSourceUnicastRouteEngine().route(mock(RuleMetaData.class), aggregatedDataSources);
         assertThat(actual.getRouteUnits().size(), is(1));
         RouteUnit routeUnit = actual.getRouteUnits().iterator().next();
         assertThat(routeUnit.getDataSourceMapper().getLogicName(), is("foo_ds_1"));