提交 cfabe6bd 编写于 作者: T tristaZero

use FromSchemaSegment instead of removeToken

上级 0da410ae
......@@ -30,6 +30,7 @@ import org.apache.shardingsphere.core.parse.sql.context.condition.Column;
import org.apache.shardingsphere.core.parse.sql.context.condition.Condition;
import org.apache.shardingsphere.core.parse.sql.context.table.Table;
import org.apache.shardingsphere.core.parse.sql.segment.common.TableSegment;
import org.apache.shardingsphere.core.parse.sql.segment.dal.FromSchemaSegment;
import org.apache.shardingsphere.core.parse.sql.segment.ddl.index.IndexSegment;
import org.apache.shardingsphere.core.parse.sql.segment.dml.InsertValuesSegment;
import org.apache.shardingsphere.core.parse.sql.segment.dml.SelectItemsSegment;
......@@ -55,7 +56,6 @@ import org.apache.shardingsphere.core.rewrite.SQLRewriteEngine;
import org.apache.shardingsphere.core.rewrite.builder.ParameterBuilder;
import org.apache.shardingsphere.core.rewrite.builder.SQLBuilder;
import org.apache.shardingsphere.core.rewrite.token.pojo.OrderByToken;
import org.apache.shardingsphere.core.rewrite.token.pojo.RemoveToken;
import org.apache.shardingsphere.core.rewrite.token.pojo.SelectItemsToken;
import org.apache.shardingsphere.core.rewrite.token.pojo.TableToken;
import org.apache.shardingsphere.core.route.SQLRouteResult;
......@@ -587,7 +587,7 @@ public final class ShardingSQLRewriterTest {
@Test
public void assertTableTokenWithoutBackQuoteForShow() {
showTablesStatement.addSQLToken(new TableToken(18, 24, "table_x", QuoteCharacter.NONE));
showTablesStatement.getSqlSegments().add(new TableSegment(18, 24, "table_x"));
routeResult = new SQLRouteResult(showTablesStatement);
RoutingResult routingResult = new RoutingResult();
routingResult.getRoutingUnits().add(new RoutingUnit("ds"));
......@@ -599,8 +599,8 @@ public final class ShardingSQLRewriterTest {
@Test
public void assertTableTokenWithoutBackQuoteFromSchemaForShow() {
showTablesStatement.getSqlSegments().add(new FromSchemaSegment(25, 43));
showTablesStatement.addSQLToken(new TableToken(18, 24, "table_x", QuoteCharacter.NONE));
showTablesStatement.addSQLToken(new RemoveToken(25, 43));
routeResult = new SQLRouteResult(showTablesStatement);
RoutingResult routingResult = new RoutingResult();
routingResult.getRoutingUnits().add(new RoutingUnit("ds"));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册