Skip to content

Commit

Permalink
refactor TableMetaDataLoader
Browse files Browse the repository at this point in the history
  • Loading branch information
terrymanu committed Aug 1, 2018
1 parent f162921 commit 2d54305
Show file tree
Hide file tree
Showing 6 changed files with 13 additions and 13 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@
import io.shardingsphere.core.jdbc.adapter.AbstractDataSourceAdapter;
import io.shardingsphere.core.jdbc.core.ShardingContext;
import io.shardingsphere.core.jdbc.core.connection.ShardingConnection;
import io.shardingsphere.core.jdbc.metadata.AutoClosedJDBCTableMetaDataExecutorAdapter;
import io.shardingsphere.core.jdbc.metadata.DataSourceMapTableMetaDataExecutorAdapter;
import io.shardingsphere.core.metadata.table.ShardingTableMetaData;
import io.shardingsphere.core.rule.MasterSlaveRule;
import io.shardingsphere.core.rule.ShardingRule;
Expand Down Expand Up @@ -68,7 +68,7 @@ public ShardingDataSource(final Map<String, DataSource> dataSourceMap, final Sha
shardingProperties = new ShardingProperties(null == props ? new Properties() : props);
int executorSize = shardingProperties.getValue(ShardingPropertiesConstant.EXECUTOR_SIZE);
executorEngine = new ExecutorEngine(executorSize);
ShardingTableMetaData shardingTableMetaData = new ShardingTableMetaData(executorEngine.getExecutorService(), new AutoClosedJDBCTableMetaDataExecutorAdapter(dataSourceMap));
ShardingTableMetaData shardingTableMetaData = new ShardingTableMetaData(executorEngine.getExecutorService(), new DataSourceMapTableMetaDataExecutorAdapter(dataSourceMap));
shardingTableMetaData.init(shardingRule);
boolean showSQL = shardingProperties.getValue(ShardingPropertiesConstant.SQL_SHOW);
shardingContext = new ShardingContext(dataSourceMap, shardingRule, getDatabaseType(), executorEngine, shardingTableMetaData, showSQL);
Expand All @@ -91,7 +91,7 @@ public void renew(final Map<String, DataSource> newDataSourceMap, final Sharding
originalExecutorEngine.close();
}
boolean newShowSQL = newShardingProperties.getValue(ShardingPropertiesConstant.SQL_SHOW);
ShardingTableMetaData shardingMetaData = new ShardingTableMetaData(executorEngine.getExecutorService(), new AutoClosedJDBCTableMetaDataExecutorAdapter(newDataSourceMap));
ShardingTableMetaData shardingMetaData = new ShardingTableMetaData(executorEngine.getExecutorService(), new DataSourceMapTableMetaDataExecutorAdapter(newDataSourceMap));
shardingMetaData.init(newShardingRule);
shardingProperties = newShardingProperties;
shardingContext = new ShardingContext(newDataSourceMap, newShardingRule, getDatabaseType(), executorEngine, shardingMetaData, newShowSQL);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@
import io.shardingsphere.core.jdbc.core.connection.ShardingConnection;
import io.shardingsphere.core.jdbc.core.resultset.GeneratedKeysResultSet;
import io.shardingsphere.core.jdbc.core.resultset.ShardingResultSet;
import io.shardingsphere.core.jdbc.metadata.ConnectionHoldJDBCTableMetaDataExecutorAdapter;
import io.shardingsphere.core.jdbc.metadata.ShardingConnectionTableMetaDataExecutorAdapter;
import io.shardingsphere.core.merger.JDBCQueryResult;
import io.shardingsphere.core.merger.MergeEngine;
import io.shardingsphere.core.merger.MergeEngineFactory;
Expand Down Expand Up @@ -217,7 +217,7 @@ private void refreshTableMetaData() {
String logicTableName = routeResult.getSqlStatement().getTables().getSingleTableName();
ShardingTableMetaData shardingTableMetaData = connection.getShardingContext().getMetaData().getTable();
TableMetaDataLoader tableMetaDataLoader = new TableMetaDataLoader(
connection.getShardingContext().getExecutorEngine().getExecutorService(), new ConnectionHoldJDBCTableMetaDataExecutorAdapter(logicTableName, connection));
connection.getShardingContext().getExecutorEngine().getExecutorService(), new ShardingConnectionTableMetaDataExecutorAdapter(logicTableName, connection));
shardingTableMetaData.put(logicTableName, tableMetaDataLoader.loadTableMetaData(logicTableName, connection.getShardingContext().getShardingRule()));
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@
import io.shardingsphere.core.jdbc.core.connection.ShardingConnection;
import io.shardingsphere.core.jdbc.core.resultset.GeneratedKeysResultSet;
import io.shardingsphere.core.jdbc.core.resultset.ShardingResultSet;
import io.shardingsphere.core.jdbc.metadata.ConnectionHoldJDBCTableMetaDataExecutorAdapter;
import io.shardingsphere.core.jdbc.metadata.ShardingConnectionTableMetaDataExecutorAdapter;
import io.shardingsphere.core.merger.JDBCQueryResult;
import io.shardingsphere.core.merger.MergeEngine;
import io.shardingsphere.core.merger.MergeEngineFactory;
Expand Down Expand Up @@ -251,7 +251,7 @@ private void refreshTableMetaData() {
String logicTableName = routeResult.getSqlStatement().getTables().getSingleTableName();
ShardingTableMetaData shardingTableMetaData = connection.getShardingContext().getMetaData().getTable();
TableMetaDataLoader tableMetaDataLoader = new TableMetaDataLoader(
connection.getShardingContext().getExecutorEngine().getExecutorService(), new ConnectionHoldJDBCTableMetaDataExecutorAdapter(logicTableName, connection));
connection.getShardingContext().getExecutorEngine().getExecutorService(), new ShardingConnectionTableMetaDataExecutorAdapter(logicTableName, connection));
shardingTableMetaData.put(logicTableName, tableMetaDataLoader.loadTableMetaData(logicTableName, connection.getShardingContext().getShardingRule()));
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,12 +26,12 @@
import java.util.Map;

/**
* Table meta data executor adapter for JDBC and auto closed.
* Table meta data executor adapter for JDBC with data source map.
*
* @author zhangliang
*/
@RequiredArgsConstructor
public final class AutoClosedJDBCTableMetaDataExecutorAdapter implements TableMetaDataExecutorAdapter {
public final class DataSourceMapTableMetaDataExecutorAdapter implements TableMetaDataExecutorAdapter {

private final Map<String, DataSource> dataSourceMap;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,12 +25,12 @@
import java.sql.SQLException;

/**
* Table meta data executor adapter for JDBC and connection hold by creator.
* Table meta data executor adapter for JDBC with sharding connection.
*
* @author zhangliang
*/
@RequiredArgsConstructor
public final class ConnectionHoldJDBCTableMetaDataExecutorAdapter implements TableMetaDataExecutorAdapter {
public final class ShardingConnectionTableMetaDataExecutorAdapter implements TableMetaDataExecutorAdapter {

private final String logicTableName;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
import io.shardingsphere.core.fixture.TestDataSource;
import io.shardingsphere.core.jdbc.core.ShardingContext;
import io.shardingsphere.core.jdbc.core.datasource.MasterSlaveDataSource;
import io.shardingsphere.core.jdbc.metadata.AutoClosedJDBCTableMetaDataExecutorAdapter;
import io.shardingsphere.core.jdbc.metadata.DataSourceMapTableMetaDataExecutorAdapter;
import io.shardingsphere.core.metadata.table.ShardingTableMetaData;
import io.shardingsphere.core.rule.ShardingRule;
import org.junit.After;
Expand Down Expand Up @@ -75,7 +75,7 @@ public void setUp() {
dataSourceMap.put(DS_NAME, masterSlaveDataSource);
ShardingRule shardingRule = new ShardingRule(shardingRuleConfig, dataSourceMap.keySet());
ListeningExecutorService executorService = Mockito.mock(ListeningExecutorService.class);
ShardingTableMetaData shardingTableMetaData = new ShardingTableMetaData(executorService, new AutoClosedJDBCTableMetaDataExecutorAdapter(dataSourceMap));
ShardingTableMetaData shardingTableMetaData = new ShardingTableMetaData(executorService, new DataSourceMapTableMetaDataExecutorAdapter(dataSourceMap));
ShardingContext shardingContext = new ShardingContext(dataSourceMap, shardingRule, DatabaseType.H2, null, shardingTableMetaData, false);
connection = new ShardingConnection(shardingContext);
}
Expand Down

0 comments on commit 2d54305

Please sign in to comment.