strongduanmu commented on code in PR #26038:
URL: https://github.com/apache/shardingsphere/pull/26038#discussion_r1223862524
##########
parser/sql/dialect/mysql/src/main/java/org/apache/shardingsphere/sql/parser/mysql/visitor/statement/MySQLStatementVisitor.java:
##########
@@ -768,9 +770,25 @@ public ASTNode visitTableStatement(final
TableStatementContext ctx) {
@Override
public ASTNode visitWindowClause(final WindowClauseContext ctx) {
+ if (null != ctx.windowItem()) {
+ return new WindowSegment(ctx.getStart().getStartIndex(),
ctx.getStop().getStopIndex(), getWindowItem(ctx.windowItem(0)),
+
getWindowSpecification(ctx.windowItem(0).windowSpecification()));
+ }
return new WindowSegment(ctx.getStart().getStartIndex(),
ctx.getStop().getStopIndex());
}
+ private IdentifierValue getWindowItem(final WindowItemContext ctx) {
+ return new IdentifierValue(ctx.identifier().getText());
+ }
+
+ private Collection<ExpressionSegment> getWindowSpecification(final
WindowSpecificationContext ctx) {
+ Collection<ExpressionSegment> segments = new LinkedList<>();
+ for (ExprContext each : ctx.expr()) {
+ segments.add((ExpressionSegment) visit(each));
+ }
+ return segments;
Review Comment:
Please rename segments to result.
##########
parser/sql/dialect/postgresql/src/main/java/org/apache/shardingsphere/sql/parser/postgresql/visitor/statement/PostgreSQLStatementVisitor.java:
##########
@@ -980,9 +982,22 @@ public ASTNode visitHavingClause(final HavingClauseContext
ctx) {
@Override
public ASTNode visitWindowClause(final WindowClauseContext ctx) {
+ if (null != ctx.windowDefinitionList()) {
+ return new WindowSegment(ctx.getStart().getStartIndex(),
ctx.getStop().getStopIndex(),
getWindowItem(ctx.windowDefinitionList().windowDefinition()),
+
getWindowSpecification(ctx.windowDefinitionList().windowDefinition().windowSpecification()));
+ }
return new WindowSegment(ctx.getStart().getStartIndex(),
ctx.getStop().getStopIndex());
}
+ private IdentifierValue getWindowItem(final WindowDefinitionContext ctx) {
+ return new IdentifierValue(ctx.colId().identifier().getText());
+ }
+
+ private Collection<ExpressionSegment> getWindowSpecification(final
WindowSpecificationContext ctx) {
Review Comment:
Hi @kanha-gupta, can you add sql parse test case for new visit logic?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
