Fix bq any type parse issue. (#6294)

* Fix bq any type parse issue. * Fix bq any type parse issue.
alibaba · Dec 19, 2024 · 324628c · 324628c
1 parent 49f8166
commit 324628c
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/core/src/main/java/com/alibaba/druid/sql/dialect/bigquery/parser/BigQueryExprParser.java b/core/src/main/java/com/alibaba/druid/sql/dialect/bigquery/parser/BigQueryExprParser.java
@@ -333,12 +333,12 @@ public SQLExpr primaryRest(SQLExpr expr) {
         return super.primaryRest(expr);
     }
 
-    public SQLDataType parseDataType() {
+    public SQLDataType parseDataType(boolean restrict) {
         if (lexer.nextIf(Token.ANY)) {
             acceptIdentifier("TYPE");
             return new SQLDataTypeImpl("ANY TYPE");
         }
-        return parseDataType(true);
+        return super.parseDataType(restrict);
     }
 
     protected SQLExpr dotRest(SQLExpr expr) {

diff --git a/core/src/test/java/com/alibaba/druid/bvt/sql/bigquery/BigQueryDataTypeTest.java b/core/src/test/java/com/alibaba/druid/bvt/sql/bigquery/BigQueryDataTypeTest.java
@@ -0,0 +1,18 @@
+package com.alibaba.druid.bvt.sql.bigquery;
+
+import com.alibaba.druid.DbType;
+import com.alibaba.druid.sql.ast.SQLDataType;
+import com.alibaba.druid.sql.parser.SQLExprParser;
+import com.alibaba.druid.sql.parser.SQLParserUtils;
+import org.junit.Assert;
+import org.junit.Test;
+
+public class BigQueryDataTypeTest {
+    @Test
+    public void testDataType() {
+        String sql = "ANY TYPE";
+        SQLExprParser exprParser = SQLParserUtils.createExprParser(sql, DbType.bigquery);
+        SQLDataType dataType = exprParser.parseDataType(false);
+        Assert.assertEquals(sql, dataType.getName());
+    }
+}