Examples with SchemaPlaceholder io.shardingjdbc.core.rewrite.placeholder.SchemaPlaceholder used on opensource projects

Example 1 with SchemaPlaceholder

use of io.shardingjdbc.core.rewrite.placeholder.SchemaPlaceholder in project sharding-jdbc by shardingjdbc.

the class SQLBuilderTest method assertSchemaPlaceholderAppendTableWithoutTableToken.

@Test(expected = ShardingJdbcException.class)
public void assertSchemaPlaceholderAppendTableWithoutTableToken() {
    SQLBuilder sqlBuilder = new SQLBuilder();
    sqlBuilder.appendLiterals("SHOW ");
    sqlBuilder.appendLiterals("CREATE TABLE ");
    sqlBuilder.appendPlaceholder(new TablePlaceholder("table_x"));
    sqlBuilder.appendLiterals("ON ");
    sqlBuilder.appendPlaceholder(new SchemaPlaceholder("dx", "table_x"));
    sqlBuilder.toSQL(Collections.<String, String>emptyMap(), createShardingRule());
}

Example 2 with SchemaPlaceholder

use of io.shardingjdbc.core.rewrite.placeholder.SchemaPlaceholder in project sharding-jdbc by shardingjdbc.

the class SQLBuilder method toSQL.

/**
 * Convert to SQL string.
 *
 * @param logicAndActualTableMap logic and actual map
 * @param shardingRule sharding rule
 * @return SQL string
 */
public String toSQL(final Map<String, String> logicAndActualTableMap, final ShardingRule shardingRule) {
    StringBuilder result = new StringBuilder();
    for (Object each : segments) {
        if (!(each instanceof ShardingPlaceholder)) {
            result.append(each);
            continue;
        }
        String logicTableName = ((ShardingPlaceholder) each).getLogicTableName();
        String actualTableName = logicAndActualTableMap.get(logicTableName);
        if (each instanceof TablePlaceholder) {
            result.append(null == actualTableName ? logicTableName : actualTableName);
        } else if (each instanceof SchemaPlaceholder) {
            SchemaPlaceholder schemaPlaceholder = (SchemaPlaceholder) each;
            Optional<TableRule> tableRule = shardingRule.tryFindTableRuleByActualTable(actualTableName);
            if (!tableRule.isPresent() && Strings.isNullOrEmpty(shardingRule.getDefaultDataSourceName())) {
                throw new ShardingJdbcException("Cannot found schema name '%s' in sharding rule.", schemaPlaceholder.getLogicSchemaName());
            }
            // TODO 目前只能找到真实数据源名称. 未来需要在初始化sharding rule时创建connection,并验证连接是否正确,并获取出真实的schema的名字, 然后在这里替换actualDataSourceName为actualSchemaName
            // TODO 目前actualDataSourceName必须actualSchemaName一样,才能保证替换schema的场景不出错, 如: show columns xxx
            result.append(tableRule.get().getActualDatasourceNames().iterator().next());
        } else if (each instanceof IndexPlaceholder) {
            IndexPlaceholder indexPlaceholder = (IndexPlaceholder) each;
            result.append(indexPlaceholder.getLogicIndexName());
            if (!Strings.isNullOrEmpty(actualTableName)) {
                result.append("_");
                result.append(actualTableName);
            }
        } else {
            result.append(each);
        }
    }
    return result.toString();
}

Example 3 with SchemaPlaceholder

use of io.shardingjdbc.core.rewrite.placeholder.SchemaPlaceholder in project sharding-jdbc by shardingjdbc.

the class SQLRewriteEngine method appendSchemaPlaceholder.

private void appendSchemaPlaceholder(final SQLBuilder sqlBuilder, final SchemaToken schemaToken, final int count, final List<SQLToken> sqlTokens) {
    sqlBuilder.appendPlaceholder(new SchemaPlaceholder(schemaToken.getSchemaName().toLowerCase(), schemaToken.getTableName().toLowerCase()));
    int beginPosition = schemaToken.getBeginPosition() + schemaToken.getOriginalLiterals().length();
    appendRest(sqlBuilder, count, sqlTokens, beginPosition);
}

Example 4 with SchemaPlaceholder

use of io.shardingjdbc.core.rewrite.placeholder.SchemaPlaceholder in project sharding-jdbc by shardingjdbc.

the class SQLBuilderTest method assertSchemaPlaceholderAppendTableWithTableToken.

@Test
public void assertSchemaPlaceholderAppendTableWithTableToken() {
    SQLBuilder sqlBuilder = new SQLBuilder();
    sqlBuilder.appendLiterals("SHOW ");
    sqlBuilder.appendLiterals("CREATE TABLE ");
    sqlBuilder.appendPlaceholder(new TablePlaceholder("table_0"));
    sqlBuilder.appendLiterals(" ON ");
    sqlBuilder.appendPlaceholder(new SchemaPlaceholder("ds", "table_0"));
    Map<String, String> tableTokens = new HashMap<>(1, 1);
    tableTokens.put("table_0", "table_1");
    assertThat(sqlBuilder.toSQL(tableTokens, createShardingRule()), is("SHOW CREATE TABLE table_1 ON ds0"));
}