odps support mysql quote, fixed issue #1292

src/main/java/com/alibaba/druid/sql/PagerUtils.java

@@ -30,6 +30,7 @@ import com.alibaba.druid.sql.ast.expr.SQLBinaryOperator;
 import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
 import com.alibaba.druid.sql.ast.expr.SQLIntegerExpr;
 import com.alibaba.druid.sql.ast.expr.SQLNumberExpr;
+import com.alibaba.druid.sql.ast.expr.SQLNumericLiteralExpr;
 import com.alibaba.druid.sql.ast.expr.SQLPropertyExpr;
 import com.alibaba.druid.sql.ast.statement.SQLSelect;
 import com.alibaba.druid.sql.ast.statement.SQLSelectItem;
@@ -41,6 +42,7 @@ import com.alibaba.druid.sql.ast.statement.SQLUnionQuery;
 import com.alibaba.druid.sql.dialect.db2.ast.stmt.DB2SelectQueryBlock;
 import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlSelectQueryBlock;
 import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlSelectQueryBlock.Limit;
+import com.alibaba.druid.sql.dialect.odps.ast.OdpsSelectQueryBlock;
 import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleSelectQueryBlock;
 import com.alibaba.druid.sql.dialect.postgresql.ast.stmt.PGSelectQueryBlock;
 import com.alibaba.druid.sql.dialect.postgresql.ast.stmt.PGSelectQueryBlock.PGLimit;
@@ -447,4 +449,58 @@ public class PagerUtils {
             clearOrderBy(union.getRight());
         }
     }
+    
+    /**
+     * 
+     * @param sql
+     * @param dbType
+     * @return if not exists limit, return -1;
+     */
+    public static int getLimit(String sql, String dbType) {
+        List<SQLStatement> stmtList = SQLUtils.parseStatements(sql, dbType);
+
+        if (stmtList.size() != 1) {
+            return -1;
+        }
+
+        SQLStatement stmt = stmtList.get(0);
+
+        if (stmt instanceof SQLSelectStatement) {
+            SQLSelectStatement selectStmt = (SQLSelectStatement) stmt;
+            SQLSelectQuery query = selectStmt.getSelect().getQuery();
+            if (query instanceof SQLSelectQueryBlock) {
+                if (query instanceof MySqlSelectQueryBlock) {
+                    MySqlSelectQueryBlock.Limit limit = ((MySqlSelectQueryBlock) query).getLimit();
+
+                    if (limit == null) {
+                        return -1;
+                    }
+
+                    SQLExpr rowCountExpr = limit.getRowCount();
+
+                    if (rowCountExpr instanceof SQLNumericLiteralExpr) {
+                        int rowCount = ((SQLNumericLiteralExpr) rowCountExpr).getNumber().intValue();
+                        return rowCount;
+                    }
+
+                    return Integer.MAX_VALUE;
+                }
+
+                if (query instanceof OdpsSelectQueryBlock) {
+                    SQLExpr rowCountExpr = ((OdpsSelectQueryBlock) query).getLimit();
+
+                    if (rowCountExpr instanceof SQLNumericLiteralExpr) {
+                        int rowCount = ((SQLNumericLiteralExpr) rowCountExpr).getNumber().intValue();
+                        return rowCount;
+                    }
+
+                    return Integer.MAX_VALUE;
+                }
+
+                return -1;
+            }
+        }
+        
+        return -1;
+    }
 }

src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlLexer.java

@@ -17,7 +17,6 @@ package com.alibaba.druid.sql.dialect.mysql.parser;
 
 import static com.alibaba.druid.sql.parser.CharTypes.isFirstIdentifierChar;
 import static com.alibaba.druid.sql.parser.LayoutCharacters.EOI;
-import static com.alibaba.druid.sql.parser.Token.LITERAL_CHARS;
 
 import java.util.HashMap;
 import java.util.Map;

src/main/java/com/alibaba/druid/sql/dialect/odps/parser/OdpsLexer.java

@@ -181,6 +181,34 @@ public class OdpsLexer extends Lexer {
 
     public void scanIdentifier() {
         final char first = ch;
+        
+        if (first == '`') {
+
+            mark = pos;
+            bufPos = 1;
+            char ch;
+            for (;;) {
+                ch = charAt(++pos);
+
+                if (ch == '`') {
+                    bufPos++;
+                    ch = charAt(++pos);
+                    break;
+                } else if (ch == EOI) {
+                    throw new ParserException("illegal identifier");
+                }
+
+                bufPos++;
+                continue;
+            }
+
+            this.ch = charAt(pos);
+
+            stringVal = subString(mark, bufPos);
+            token = Token.IDENTIFIER;
+            
+            return;
+        }
 
         final boolean firstFlag = isFirstIdentifierChar(first);
         if (!firstFlag) {

src/test/java/com/alibaba/druid/bvt/sql/odps/OdpsSelectTest13.java

@@ -0,0 +1,59 @@
+/*
+ * Copyright 1999-2101 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.bvt.sql.odps;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.SQLUtils;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat.Column;
+import com.alibaba.druid.util.JdbcConstants;
+
+import junit.framework.TestCase;
+
+public class OdpsSelectTest13 extends TestCase {
+
+    public void test_select() throws Exception {
+        String sql = "select `name` from `abc`;";//
+        Assert.assertEquals("SELECT `name`"
+                + "\nFROM `abc`;", SQLUtils.formatOdps(sql));
+        Assert.assertEquals("select `name`"
+                + "\nfrom `abc`;", SQLUtils.formatOdps(sql, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
+        
+        List<SQLStatement> statementList = SQLUtils.parseStatements(sql, JdbcConstants.ODPS);
+        SQLStatement stmt = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+        
+        SchemaStatVisitor visitor = SQLUtils.createSchemaStatVisitor(JdbcConstants.ODPS);
+        stmt.accept(visitor);
+        
+//        System.out.println("Tables : " + visitor.getTables());
+//      System.out.println("fields : " + visitor.getColumns());
+//      System.out.println("coditions : " + visitor.getConditions());
+//      System.out.println("orderBy : " + visitor.getOrderByColumns());
+        
+        Assert.assertEquals(1, visitor.getTables().size());
+        Assert.assertEquals(1, visitor.getColumns().size());
+        Assert.assertEquals(0, visitor.getConditions().size());
+        
+        Assert.assertTrue(visitor.getColumns().contains(new Column("abc", "name")));
+    }
+    
+}