jingshanglu commented on a change in pull request #8116:
URL: https://github.com/apache/shardingsphere/pull/8116#discussion_r521062526
##########
File path:
shardingsphere-sql-parser/shardingsphere-sql-parser-dialect/shardingsphere-sql-parser-mysql/src/main/java/org/apache/shardingsphere/sql/parser/mysql/visitor/format/impl/MySQLFormatSQLVisitor.java
##########
@@ -19,11 +19,423 @@
import lombok.AccessLevel;
import lombok.Getter;
+import org.antlr.v4.runtime.tree.ParseTree;
+import org.antlr.v4.runtime.tree.RuleNode;
+import org.antlr.v4.runtime.tree.TerminalNode;
import org.apache.shardingsphere.sql.parser.autogen.MySQLStatementBaseVisitor;
+import org.apache.shardingsphere.sql.parser.autogen.MySQLStatementParser;
/**
* MySQL Format SQL visitor for MySQL.
*/
@Getter(AccessLevel.PROTECTED)
public abstract class MySQLFormatSQLVisitor extends
MySQLStatementBaseVisitor<String> {
+ private StringBuilder result = new StringBuilder();
+
+ private final boolean uperCase = true;
+
+ private int indentCount;
+
+ private int lines;
+
+ private int projectionsCountOfLine = 3;
+
+ @Override
+ public String visitSelect(final MySQLStatementParser.SelectContext ctx) {
+ if (null != ctx.queryExpression()) {
+ visit(ctx.queryExpression());
+ if (null != ctx.lockClauseList()) {
+ result.append(" ");
+ visit(ctx.lockClauseList());
+ }
+ } else if (null != ctx.queryExpressionParens()) {
+ visit(ctx.queryExpressionParens());
+ } else {
+ visit(ctx.selectWithInto());
+ }
+ print(";");
+ return result.toString();
+ }
+
+ @Override
+ public String visitQueryExpression(final
MySQLStatementParser.QueryExpressionContext ctx) {
+ if (null != ctx.withClause()) {
+ visit(ctx.withClause());
+ print(" ");
+ }
+ if (null != ctx.queryExpressionBody()) {
+ visit(ctx.queryExpressionBody());
+ } else {
+ visit(ctx.queryExpressionParens());
+ }
+ if (null != ctx.orderByClause()) {
+ print(" ");
+ visit(ctx.orderByClause());
+ }
+ if (null != ctx.limitClause()) {
+ print(" ");
+ visit(ctx.limitClause());
+ }
+ return result.toString();
+ }
+
+ @Override
+ public String visitQueryExpressionParens(final
MySQLStatementParser.QueryExpressionParensContext ctx) {
+ println();
+ this.indentCount++;
+ print("(");
+ println();
+ if (null != ctx.queryExpressionParens()) {
+ visit(ctx.queryExpressionParens());
+ } else {
+ visit(ctx.queryExpression());
+ if (null != ctx.lockClauseList()) {
+ result.append(" ");
+ visit(ctx.lockClauseList());
+ }
+ }
+ this.indentCount--;
+ println();
+ print(")");
+ return result.toString();
+ }
+
+ @Override
+ public String visitQueryExpressionBody(final
MySQLStatementParser.QueryExpressionBodyContext ctx) {
+ if (1 == ctx.getChildCount()) {
+ visit(ctx.queryPrimary());
+ } else {
+ visit(ctx.queryExpressionParens(0));
+ result.append("\nUNION\n");
+ if (null != ctx.unionOption()) {
+ visit(ctx.unionOption());
+ result.append(" ");
+ }
+ if (null != ctx.queryPrimary()) {
+ visit(ctx.queryPrimary());
+ } else {
+ visit(ctx.queryExpressionParens(1));
+ }
+ }
+ return result.toString();
+ }
+
+ @Override
+ public String visitQuerySpecification(final
MySQLStatementParser.QuerySpecificationContext ctx) {
+ print("SELECT ");
+ int selectSpecCount = ctx.selectSpecification().size();
+ for (int i = 0; i < selectSpecCount; i++) {
+ visit(ctx.selectSpecification(i));
+ print(" ");
+ }
+ visit(ctx.projections());
+ if (null != ctx.fromClause()) {
+ println();
+ visit(ctx.fromClause());
+ }
+ if (null != ctx.whereClause()) {
+ println();
+ visit(ctx.whereClause());
+ }
+ if (null != ctx.groupByClause()) {
+ println();
+ visit(ctx.groupByClause());
+ }
+ if (null != ctx.havingClause()) {
+ println();
+ visit(ctx.havingClause());
+ }
+ if (null != ctx.windowClause()) {
+ println();
+ visit(ctx.windowClause());
+ }
+ return result.toString();
+ }
+
+ @Override
+ public String visitExplicitTable(final
MySQLStatementParser.ExplicitTableContext ctx) {
+ print("TABLE ");
+ visit(ctx.tableName());
+ return result.toString();
+ }
+
+ @Override
+ public String visitTableName(final MySQLStatementParser.TableNameContext
ctx) {
+ if (null != ctx.owner()) {
+ print(ctx.owner().getText());
+ print(".");
+ }
+ print(ctx.name().getText());
+ return result.toString();
+ }
+
+ @Override
+ public String visitTableValueConstructor(final
MySQLStatementParser.TableValueConstructorContext ctx) {
+ print("VALUES ");
+ visit(ctx.rowConstructorList());
+ return result.toString();
+ }
+
+ @Override
+ public String visitRowConstructorList(final
MySQLStatementParser.RowConstructorListContext ctx) {
+ int rowCount = ctx.assignmentValues().size();
+ for (int i = 0; i < rowCount; i++) {
+ if (i != 0 && i != rowCount) {
+ print(", ROW");
+ visit(ctx.assignmentValues(i));
+
+ } else {
+ print("ROW");
+ visit(ctx.assignmentValues(i));
+ }
+ }
+ return result.toString();
+ }
+
+ @Override
+ public String visitAssignmentValues(final
MySQLStatementParser.AssignmentValuesContext ctx) {
+ print("(");
+ int assignCount = ctx.assignmentValue().size();
+ for (int i = 0; i < assignCount; i++) {
+ if (i != 0) {
+ print(", ");
+ visit(ctx.assignmentValue(i));
+ } else {
+ visit(ctx.assignmentValue(i));
+ }
+ }
+ print(")");
+ return result.toString();
+ }
+
+ @Override
+ public String visitWhereClause(final
MySQLStatementParser.WhereClauseContext ctx) {
+ visit(ctx.WHERE());
+ print(" ");
+ this.indentCount++;
+ println();
+ visit(ctx.expr());
+ this.indentCount--;
+ return result.toString();
+ }
+
+ @Override
+ public String visitExpr(final MySQLStatementParser.ExprContext ctx) {
+ if (null != ctx.logicalOperator()) {
+ MySQLStatementParser.ExprContext left = ctx.expr(0);
+ visit(left);
+ println();
+ MySQLStatementParser.ExprContext right = ctx.expr(1);
+ print(ctx.logicalOperator().getText());
+ visit(right);
+ } else if (null != ctx.notOperator()) {
+ print(ctx.notOperator().getText());
+ visit(ctx.expr(0));
+ } else {
+ visitChildren(ctx);
+ }
+ return result.toString();
+ }
+
+ @Override
+ public String visitAlias(final MySQLStatementParser.AliasContext ctx) {
+ print(ctx.getText());
+ return result.toString();
+ }
+
+ @Override
+ public String visitProjections(final
MySQLStatementParser.ProjectionsContext ctx) {
+ this.indentCount++;
+ if (null != ctx.unqualifiedShorthand()) {
+ visit(ctx.unqualifiedShorthand());
+ result.append(" ");
+ }
+ int projectionCount = ctx.projection().size();
+ int lineItemCount = 0;
+ for (int i = 0; i < projectionCount; i++) {
+ if (0 != i) {
+ result.append(", ");
+ if (lineItemCount >= projectionsCountOfLine) {
+ lineItemCount = 0;
+ println();
+ }
+ }
+ visit(ctx.projection(i));
+ lineItemCount++;
+ }
+ this.indentCount--;
+ return result.toString();
+ }
+
+ @Override
+ public String visitProjection(final MySQLStatementParser.ProjectionContext
ctx) {
+ if (null != ctx.expr()) {
+ visit(ctx.expr());
+ print(" ");
+ }
+ if (null != ctx.AS()) {
+ print("AS ");
+ }
+ if (null != ctx.alias()) {
+ visit(ctx.alias());
+ }
+ if (null != ctx.qualifiedShorthand()) {
+ visit(ctx.qualifiedShorthand());
+ }
+ return result.toString();
+ }
+
+ @Override
+ public String visitTerminal(final TerminalNode node) {
+ if (isUperCase()) {
+ print(node.getText().toUpperCase());
+ } else {
+ print(node.getText().toLowerCase());
+ }
+ return result.toString();
+ }
+
+ @Override
+ public String visitIdentifier(final MySQLStatementParser.IdentifierContext
ctx) {
+ print(ctx.getText());
+ return result.toString();
+ }
+
+ @Override
+ public String visitLiterals(final MySQLStatementParser.LiteralsContext
ctx) {
+ print("?");
+ return result.toString();
+ }
+
+ @Override
+ public String visitStringLiterals(final
MySQLStatementParser.StringLiteralsContext ctx) {
+ print("?");
+ return result.toString();
+ }
+
+ @Override
+ public String visitNumberLiterals(final
MySQLStatementParser.NumberLiteralsContext ctx) {
+ print("?");
+ return result.toString();
+ }
+
+ @Override
+ public String visitWithClause(final MySQLStatementParser.WithClauseContext
ctx) {
+ print("WITH ");
+ if (null != ctx.RECURSIVE()) {
+ visit(ctx.RECURSIVE());
+ result.append(" ");
+ }
+ for (int i = 0; i < ctx.cteClause().size(); i++) {
+ if (i != 0 && i < ctx.cteClause().size() - 1) {
+ result.append(", ");
+ }
+ visit(ctx.cteClause(i));
+ }
+ if (null != ctx.parent) {
+ result.append("\n");
+ }
+ return result.toString();
+ }
+
+ @Override
+ public String visitCteClause(final MySQLStatementParser.CteClauseContext
ctx) {
+ visit(ctx.ignoredIdentifier());
+ result.append(" ");
+ if (null != ctx.columnNames()) {
+ visit(ctx.columnNames());
+ result.append(" ");
+ }
+ result.append("AS ");
+ visit(ctx.subquery());
+ return result.toString();
+ }
+
+ @Override
+ public String visitIgnoredIdentifier(final
MySQLStatementParser.IgnoredIdentifierContext ctx) {
+ visit(ctx.identifier(0));
+ if (null != ctx.DOT_()) {
+ visit(ctx.DOT_());
+ visit(ctx.identifier(1));
+ }
+ return result.toString();
+ }
+
+ @Override
+ public String visitColumnNames(final
MySQLStatementParser.ColumnNamesContext ctx) {
+ if (null != ctx.LP_()) {
+ visit(ctx.LP_());
+ }
+ int columnCount = ctx.columnName().size();
+ for (int i = 0; i < columnCount; i++) {
+ if (i != 0 && i < columnCount - 1) {
+ result.append(", ");
+ } else {
+ visit(ctx.columnName(i));
+ }
+ }
+ if (null != ctx.RP_()) {
+ visit(ctx.RP_());
+ }
+ return result.toString();
+ }
+
+ @Override
+ public String visitColumnName(final MySQLStatementParser.ColumnNameContext
ctx) {
+ if (null != ctx.owner()) {
+ visit(ctx.owner());
+ visit(ctx.DOT_(0));
+ }
+ visit(ctx.name());
+ return result.toString();
+ }
+
+ @Override
+ public String visitChildren(final RuleNode node) {
+ String result = defaultResult();
+
+ int n = node.getChildCount();
+ for (int i = 0; i < n; i++) {
+ if (i != 0) {
+ print(" ");
+ }
+ if (!shouldVisitNextChild(node, result)) {
+ break;
+ }
+
+ ParseTree c = node.getChild(i);
+ String childResult = c.accept(this);
+ result = aggregateResult(result, childResult);
+ }
+ return result;
+ }
+
+ private void print(final char value) {
+ if (this.result == null) {
Review comment:
OK
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
[email protected]
