Add TD select parser

2023-03-04 19:17:16 +00:00 · 2016-03-28 19:36:14 +08:00
parent 8d8451bdff
commit 611f4f8163
1 changed files with 62 additions and 0 deletions
@@ -0,0 +1,62 @@
+package com.alibaba.druid.sql.dialect.teradata.parser;
+
+import com.alibaba.druid.sql.ast.SQLSetQuantifier;
+import com.alibaba.druid.sql.ast.statement.SQLSelectQuery;
+import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
+import com.alibaba.druid.sql.dialect.teradata.ast.stmt.TeradataSelectQueryBlock;
+import com.alibaba.druid.sql.parser.SQLExprParser;
+import com.alibaba.druid.sql.parser.SQLSelectParser;
+import com.alibaba.druid.sql.parser.Token;
+
+public class TeradataSelectParser extends SQLSelectParser{
+
+	public TeradataSelectParser(SQLExprParser exprParser) {
+		super(exprParser);
+	}
+	
+	public TeradataSelectParser(String sql) {
+		this(new TeradataExprParser(sql));
+	}
+	
+	public SQLSelectQuery query() {
+		if (lexer.token() == Token.LPAREN) {
+            lexer.nextToken();
+
+            SQLSelectQuery select = query();
+            accept(Token.RPAREN);
+
+            return queryRest(select);
+        }
+        
+        TeradataSelectQueryBlock queryBlock = new TeradataSelectQueryBlock();
+        
+        if (lexer.token() == Token.SEL 
+        		|| lexer.token() == Token.SELECT) {
+
+            lexer.nextToken();
+
+            if (lexer.token() == Token.COMMENT) {
+                lexer.nextToken();
+            }
+
+            if (lexer.token() == (Token.DISTINCT)) {
+                queryBlock.setDistionOption(SQLSetQuantifier.DISTINCT);
+                lexer.nextToken();
+            } else if (lexer.token() == (Token.ALL)) {
+                queryBlock.setDistionOption(SQLSetQuantifier.ALL);
+                lexer.nextToken();
+            }
+            parseSelectList(queryBlock);
+        }
+        parseFrom(queryBlock);
+
+        parseWhere(queryBlock);
+
+        parseGroupBy(queryBlock);
+        
+        queryBlock.setOrderBy(this.exprParser.parseOrderBy());
+
+        return queryRest(queryBlock);
+	}
+
+}