View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *     http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  
18  package org.apache.shardingsphere.distsql.parser.core.utility;
19  
20  import org.apache.shardingsphere.distsql.parser.autogen.UtilityDistSQLStatementBaseVisitor;
21  import org.apache.shardingsphere.distsql.parser.autogen.UtilityDistSQLStatementParser.FormatSQLContext;
22  import org.apache.shardingsphere.distsql.parser.autogen.UtilityDistSQLStatementParser.ParseSQLContext;
23  import org.apache.shardingsphere.distsql.parser.autogen.UtilityDistSQLStatementParser.PreviewSQLContext;
24  import org.apache.shardingsphere.distsql.statement.rul.sql.FormatStatement;
25  import org.apache.shardingsphere.distsql.statement.rul.sql.ParseStatement;
26  import org.apache.shardingsphere.distsql.statement.rul.sql.PreviewStatement;
27  import org.apache.shardingsphere.sql.parser.api.ASTNode;
28  import org.apache.shardingsphere.sql.parser.api.visitor.SQLVisitor;
29  
30  /**
31   * SQL statement visitor for utility DistSQL.
32   */
33  public final class UtilityDistSQLStatementVisitor extends UtilityDistSQLStatementBaseVisitor<ASTNode> implements SQLVisitor<ASTNode> {
34      
35      @Override
36      public ASTNode visitPreviewSQL(final PreviewSQLContext ctx) {
37          return new PreviewStatement(ctx.sql().getText().trim());
38      }
39      
40      @Override
41      public ASTNode visitParseSQL(final ParseSQLContext ctx) {
42          return new ParseStatement(ctx.sql().getText().trim());
43      }
44      
45      @Override
46      public ASTNode visitFormatSQL(final FormatSQLContext ctx) {
47          return new FormatStatement(ctx.sql().getText().trim());
48      }
49  }