Commit e0f4245a authored by terrymanu's avatar terrymanu
Browse files

refactor database routing

parent 0002f4bf

sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/routing/type/database/DatabaseRouter.java

+9 −9
Original line number Original line Diff line number Diff line
@@ -38,7 +38,7 @@ import java.util.Collections; 
 */

 */

@RequiredArgsConstructor

@RequiredArgsConstructor

@Slf4j

@Slf4j

public class DatabaseRouter {

public final class DatabaseRouter {

    

    

    private final DataSourceRule dataSourceRule;

    private final DataSourceRule dataSourceRule;
@@ -47,17 +47,17 @@ public class DatabaseRouter { 
    private final SQLType sqlType;

    private final SQLType sqlType;

    

    

    /**

    /**

     * 根据Hint路由到库.

     * 根据线索路由到库.

     * 

     * 

     * @return 库路由结果

     * @return 库路由结果

     */

     */

    public DatabaseRoutingResult route() {

    public DatabaseRoutingResult route() {

        Optional<ShardingValue<?>> shardingValueOptional = HintManagerHolder.getDatabaseShardingValue(new ShardingKey(HintManagerHolder.DB_TABLE_NAME, HintManagerHolder.DB_COLUMN_NAME));

        Optional<ShardingValue<?>> shardingValue = HintManagerHolder.getDatabaseShardingValue(new ShardingKey(HintManagerHolder.DB_TABLE_NAME, HintManagerHolder.DB_COLUMN_NAME));

        Preconditions.checkState(shardingValueOptional.isPresent());

        Preconditions.checkState(shardingValue.isPresent());

        log.debug("Before database sharding only db:{} sharding values: {}", dataSourceRule.getDataSourceNames(), shardingValueOptional.get());

        log.debug("Before database sharding only db:{} sharding values: {}", dataSourceRule.getDataSourceNames(), shardingValue.get());

        Collection<String> routedResult = databaseShardingStrategy.doStaticSharding(sqlType, dataSourceRule.getDataSourceNames(), Collections.<ShardingValue<?>>singleton(shardingValueOptional.get()));

        Collection<String> result = databaseShardingStrategy.doStaticSharding(sqlType, dataSourceRule.getDataSourceNames(), Collections.<ShardingValue<?>>singleton(shardingValue.get()));

        Preconditions.checkState(!routedResult.isEmpty(), "no database route info");

        Preconditions.checkState(!result.isEmpty(), "no database route info");

        log.debug("After database sharding only result: {}", routedResult);

        log.debug("After database sharding only result: {}", result);

        return new DatabaseRoutingResult(routedResult);

        return new DatabaseRoutingResult(result);

    }

    }

}
 
}