提交 cfabe6bd 编写于 作者: T tristaZero

use FromSchemaSegment instead of removeToken

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