How to use

getDataSource

method

in

org.apache.shardingsphere.core.metadata.ShardingMetaData

public MySQLResponseHandler(final String dataSourceName, final String schema) {
  dataSourceParameter = GLOBAL_REGISTRY.getLogicSchema(schema).getDataSources().get(dataSourceName);
  dataSourceMetaData = GLOBAL_REGISTRY.getLogicSchema(schema).getMetaData().getDataSource().getActualDataSourceMetaData(dataSourceName);
  resultMap = new HashMap<>();
}

public MySQLResponseHandler(final String dataSourceName, final String schema) {
  dataSourceParameter = GLOBAL_REGISTRY.getLogicSchema(schema).getDataSources().get(dataSourceName);
  dataSourceMetaData = GLOBAL_REGISTRY.getLogicSchema(schema).getMetaData().getDataSource().getActualDataSourceMetaData(dataSourceName);
  resultMap = new HashMap<>();
}

/**
 * Rewrite SQL.
 * 
 * @return SQL
 */
public String rewrite() {
  if (sqlTokens.isEmpty()) {
    return originalSQL;
  }
  SQLBuilder result = new SQLBuilder(Collections.emptyList());
  int count = 0;
  for (SQLToken each : sqlTokens) {
    if (0 == count) {
      result.appendLiterals(originalSQL.substring(0, each.getStartIndex()));
    }
    if (each instanceof SchemaToken) {
      appendSchemaPlaceholder(originalSQL, result, (SchemaToken) each, count);
    }
    count++;
  }
  return result.toSQL(masterSlaveRule, metaData.getDataSource());
}

@Override
public SQLRouteResult route(final String logicSQL, final List<Object> parameters, final SQLStatement sqlStatement) {
  Optional<GeneratedKey> generatedKey = sqlStatement instanceof InsertStatement ? getGenerateKey(parameters, (InsertStatement) sqlStatement) : Optional.<GeneratedKey>absent();
  SQLRouteResult result = new SQLRouteResult(sqlStatement, generatedKey.orNull());
  ShardingConditions shardingConditions = OptimizeEngineFactory.newInstance(shardingRule, sqlStatement, parameters, generatedKey.orNull()).optimize();
  if (generatedKey.isPresent()) {
    setGeneratedKeys(result, generatedKey.get());
  }
  if (sqlStatement instanceof SelectStatement && isNeedMergeShardingValues((SelectStatement) sqlStatement)) {
    checkSubqueryShardingValues(sqlStatement, sqlStatement.getConditions(), shardingConditions);
    mergeShardingValues(shardingConditions);
  }
  RoutingResult routingResult = RoutingEngineFactory.newInstance(shardingRule, shardingMetaData.getDataSource(), sqlStatement, shardingConditions).route();
  boolean isSingleRouting = routingResult.isSingleRouting();
  SQLRewriteEngine rewriteEngine = new SQLRewriteEngine(shardingRule, logicSQL, databaseType, sqlStatement, shardingConditions, parameters);
  if (sqlStatement instanceof SelectStatement && null != ((SelectStatement) sqlStatement).getLimit()) {
    processLimit(parameters, (SelectStatement) sqlStatement, isSingleRouting);
  }
  SQLBuilder sqlBuilder = rewriteEngine.rewrite(isSingleRouting);
  for (TableUnit each : routingResult.getTableUnits().getTableUnits()) {
    result.getRouteUnits().add(new RouteUnit(each.getDataSourceName(), rewriteEngine.generateSQL(each, sqlBuilder, shardingMetaData.getDataSource())));
  }
  if (showSQL) {
    SQLLogger.logSQL(logicSQL, sqlStatement, result.getRouteUnits());
  }
  return result;
}

  @Override
  protected SimpleChannelPool newPool(final String dataSourceName) {
    DataSourceMetaData dataSourceMetaData = logicSchema.getMetaData().getDataSource().getActualDataSourceMetaData(dataSourceName);
    return new FixedChannelPool(
        bootstrap.remoteAddress(dataSourceMetaData.getHostName(), dataSourceMetaData.getPort()), 
        new BackendNettyClientChannelPoolHandler(dataSourceName, logicSchema.getName()), maxConnections);
  }
};

  @Override
  protected SimpleChannelPool newPool(final String dataSourceName) {
    DataSourceMetaData dataSourceMetaData = logicSchema.getMetaData().getDataSource().getActualDataSourceMetaData(dataSourceName);
    return new FixedChannelPool(
        bootstrap.remoteAddress(dataSourceMetaData.getHostName(), dataSourceMetaData.getPort()), 
        new BackendNettyClientChannelPoolHandler(dataSourceName, logicSchema.getName()), maxConnections);
  }
};