text: /*
   * Copyright 1999-2018 Alibaba Group Holding Ltd.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package com.alibaba.druid.sql.dialect.hive.parser;
  
  import com.alibaba.druid.sql.ast.SQLExpr;
  import com.alibaba.druid.sql.ast.SQLName;
  import com.alibaba.druid.sql.ast.statement.SQLColumnDefinition;
  import com.alibaba.druid.sql.ast.statement.SQLConstraint;
  import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement;
  import com.alibaba.druid.sql.ast.statement.SQLExprTableSource;
  import com.alibaba.druid.sql.ast.statement.SQLExternalRecordFormat;
  import com.alibaba.druid.sql.ast.statement.SQLSelect;
  import com.alibaba.druid.sql.ast.statement.SQLSelectOrderByItem;
  import com.alibaba.druid.sql.ast.statement.SQLTableElement;
  import com.alibaba.druid.sql.dialect.hive.stmt.HiveCreateTableStatement;
  import com.alibaba.druid.sql.parser.Lexer;
  import com.alibaba.druid.sql.parser.ParserException;
  import com.alibaba.druid.sql.parser.SQLCreateTableParser;
  import com.alibaba.druid.sql.parser.SQLExprParser;
  import com.alibaba.druid.sql.parser.Token;
  import com.alibaba.druid.util.FnvHash;
  import com.alibaba.druid.util.JdbcConstants;
  
  public class HiveCreateTableParser extends SQLCreateTableParser {
      public HiveCreateTableParser(SQLExprParser exprParser) {
          super(exprParser);
      }
  
      public HiveCreateTableParser(Lexer lexer) {
          super(new HiveExprParser(lexer));
      }
  
      public SQLCreateTableStatement parseCreateTable(boolean acceptCreate) {
          HiveCreateTableStatement stmt = newCreateStatement();
  
          if (acceptCreate) {
              if (lexer.hasComment() && lexer.isKeepComments()) {
                  stmt.addBeforeComment(lexer.readAndResetComments());
              }
  
              accept(Token.CREATE);
          }
  
          if (lexer.identifierEquals("GLOBAL")) {
              lexer.nextToken();
  
              if (lexer.identifierEquals("TEMPORARY")) {
                  lexer.nextToken();
                  stmt.setType(SQLCreateTableStatement.Type.GLOBAL_TEMPORARY);
              } else {
                  throw new ParserException("syntax error " + lexer.info());
              }
          } else if (lexer.token() == Token.IDENTIFIER && lexer.stringVal().equalsIgnoreCase("LOCAL")) {
              lexer.nextToken();
              if (lexer.token() == Token.IDENTIFIER && lexer.stringVal().equalsIgnoreCase("TEMPORAY")) {
                  lexer.nextToken();
                  stmt.setType(SQLCreateTableStatement.Type.LOCAL_TEMPORARY);
              } else {
                  throw new ParserException("syntax error. " + lexer.info());
              }
          }
  
          accept(Token.TABLE);
  
          if (lexer.token() == Token.IF || lexer.identifierEquals("IF")) {
              lexer.nextToken();
              accept(Token.NOT);
              accept(Token.EXISTS);
  
              stmt.setIfNotExiists(true);
          }
  
          stmt.setName(this.exprParser.name());
  
          if (lexer.token() == Token.LPAREN) {
              lexer.nextToken();
  
              for (; ; ) {
                  Token token = lexer.token();
                  if (token == Token.IDENTIFIER
                          && lexer.stringVal().equalsIgnoreCase("SUPPLEMENTAL")
                          && JdbcConstants.ORACLE.equals(dbType)) {
                      this.parseCreateTableSupplementalLogingProps(stmt);
                  } else if (token == Token.IDENTIFIER //
                          || token == Token.LITERAL_ALIAS) {
                      SQLColumnDefinition column = this.exprParser.parseColumn();
                      stmt.getTableElementList().add(column);
                  } else if (token == Token.PRIMARY //
                          || token == Token.UNIQUE //
                          || token == Token.CHECK //
                          || token == Token.CONSTRAINT
                          || token == Token.FOREIGN) {
                      SQLConstraint constraint = this.exprParser.parseConstaint();
                      constraint.setParent(stmt);
                      stmt.getTableElementList().add((SQLTableElement) constraint);
                  } else if (token == Token.TABLESPACE) {
                      throw new ParserException("TODO "  + lexer.info());
                  } else {
                      SQLColumnDefinition column = this.exprParser.parseColumn();
                      stmt.getTableElementList().add(column);
                  }
  
                  if (lexer.token() == Token.COMMA) {
                      lexer.nextToken();
  
                      if (lexer.token() == Token.RPAREN) { // compatible for sql server
                          break;
                      }
                      continue;
                  }
  
                  break;
              }
  
              accept(Token.RPAREN);
  
              if (lexer.identifierEquals("INHERITS")) {
                  lexer.nextToken();
                  accept(Token.LPAREN);
                  SQLName inherits = this.exprParser.name();
                  stmt.setInherits(new SQLExprTableSource(inherits));
                  accept(Token.RPAREN);
              }
          }
  
          if (lexer.token() == Token.COMMENT) {
              lexer.nextToken();
              SQLExpr comment = this.exprParser.expr();
              stmt.setComment(comment);
          }
  
          if (lexer.token() == Token.PARTITIONED) {
              lexer.nextToken();
              accept(Token.BY);
              accept(Token.LPAREN);
  
              for (;;) {
                  if (lexer.token() != Token.IDENTIFIER) {
                      throw new ParserException("expect identifier. " + lexer.info());
                  }
  
                  SQLColumnDefinition column = this.exprParser.parseColumn();
                  stmt.addPartitionColumn(column);
  
                  if (lexer.isKeepComments() && lexer.hasComment()) {
                      column.addAfterComment(lexer.readAndResetComments());
                  }
  
                  if (lexer.token() != Token.COMMA) {
                      break;
                  } else {
                      lexer.nextToken();
                      if (lexer.isKeepComments() && lexer.hasComment()) {
                          column.addAfterComment(lexer.readAndResetComments());
                      }
                  }
              }
  
              accept(Token.RPAREN);
          }
  
          if (lexer.identifierEquals(FnvHash.Constants.CLUSTERED)) {
              lexer.nextToken();
              accept(Token.BY);
              accept(Token.LPAREN);
              for (; ; ) {
                  SQLSelectOrderByItem item = this.exprParser.parseSelectOrderByItem();
                  stmt.addClusteredByItem(item);
                  if (lexer.token() == Token.COMMA) {
                      lexer.nextToken();
                      continue;
                  }
                  break;
              }
              accept(Token.RPAREN);
          }
  
          if (lexer.token() == Token.ROW) {
              lexer.nextToken();
              acceptIdentifier("FORMAT");
  
              if (lexer.identifierEquals(FnvHash.Constants.DELIMITED)) {
                  lexer.nextToken();
                  acceptIdentifier("FIELDS");
                  acceptIdentifier("TERMINATED");
                  accept(Token.BY);
                  SQLExternalRecordFormat format = new SQLExternalRecordFormat();
                  format.setTerminatedBy(this.exprParser.expr());
                  stmt.setRowFormat(format);
              } else {
                  throw new ParserException("TODO " + lexer.info());
              }
          }
  
          if (lexer.identifierEquals(FnvHash.Constants.SORTED)) {
              lexer.nextToken();
              accept(Token.BY);
              accept(Token.LPAREN);
              for (; ; ) {
                  SQLSelectOrderByItem item = this.exprParser.parseSelectOrderByItem();
                  stmt.addSortedByItem(item);
                  if (lexer.token() == Token.COMMA) {
                      lexer.nextToken();
                      continue;
                  }
                  break;
              }
              accept(Token.RPAREN);
          }
  
          if (stmt.getClusteredBy().size() > 0 || stmt.getSortedBy().size() > 0) {
              accept(Token.INTO);
              if (lexer.token() == Token.LITERAL_INT) {
                  stmt.setBuckets(lexer.integerValue().intValue());
                  lexer.nextToken();
              } else {
                  throw new ParserException("into buckets must be integer. " + lexer.info());
              }
              acceptIdentifier("BUCKETS");
          }
  
          if (lexer.identifierEquals(FnvHash.Constants.STORED)) {
              lexer.nextToken();
              accept(Token.AS);
              SQLName name = this.exprParser.name();
              stmt.setStoredAs(name);
          }
  
          if (lexer.identifierEquals(FnvHash.Constants.TBLPROPERTIES)) {
              lexer.nextToken();
              accept(Token.LPAREN);
  
              for (;;) {
                  String name = lexer.stringVal();
                  lexer.nextToken();
                  accept(Token.EQ);
                  SQLExpr value = this.exprParser.primary();
                  stmt.getTableOptions().put(name, value);
                  if (lexer.token() == Token.COMMA) {
                      lexer.nextToken();
                      continue;
                  }
                  break;
              }
  
              accept(Token.RPAREN);
          }
  
          if (lexer.token() == Token.AS) {
              lexer.nextToken();
              SQLSelect select = this.createSQLSelectParser().select();
              stmt.setSelect(select);
          }
          return stmt;
      }
  
      protected HiveCreateTableStatement newCreateStatement() {
          return new HiveCreateTableStatement();
      }
  }
  