Update grammar validation settings (#3165)

Signed-off-by: Tomoyuki Morita <[email protected]>
opensearch-project · Nov 21, 2024 · 8b2d01e · 8b2d01e
1 parent 380dd34
commit 8b2d01e
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 3 deletions.
diff --git a/async-query-core/src/main/java/org/opensearch/sql/spark/validator/GrammarElement.java b/async-query-core/src/main/java/org/opensearch/sql/spark/validator/GrammarElement.java
@@ -74,7 +74,9 @@ public enum GrammarElement {
 
   // Functions
   MAP_FUNCTIONS("Map functions"),
+  BITWISE_FUNCTIONS("Bitwise functions"),
   CSV_FUNCTIONS("CSV functions"),
+  GENERATOR_FUNCTIONS("Generator functions"),
   MISC_FUNCTIONS("Misc functions"),
 
   // UDF

diff --git a/...-core/src/main/java/org/opensearch/sql/spark/validator/S3GlueGrammarElementValidator.java b/...-core/src/main/java/org/opensearch/sql/spark/validator/S3GlueGrammarElementValidator.java
@@ -6,6 +6,7 @@
 package org.opensearch.sql.spark.validator;
 
 import static org.opensearch.sql.spark.validator.GrammarElement.ALTER_VIEW;
+import static org.opensearch.sql.spark.validator.GrammarElement.BITWISE_FUNCTIONS;
 import static org.opensearch.sql.spark.validator.GrammarElement.CLUSTER_BY;
 import static org.opensearch.sql.spark.validator.GrammarElement.CREATE_FUNCTION;
 import static org.opensearch.sql.spark.validator.GrammarElement.CREATE_VIEW;
@@ -71,6 +72,7 @@ public class S3GlueGrammarElementValidator extends DenyListGrammarElementValidat
               SET,
               SHOW_FUNCTIONS,
               SHOW_VIEWS,
+              BITWISE_FUNCTIONS,
               MISC_FUNCTIONS,
               UDF)
           .build();

diff --git a/...uery-core/src/main/java/org/opensearch/sql/spark/validator/SQLQueryValidationVisitor.java b/...uery-core/src/main/java/org/opensearch/sql/spark/validator/SQLQueryValidationVisitor.java
@@ -561,12 +561,18 @@ private void validateFunctionAllowed(String function) {
       case MAP:
         validateAllowed(GrammarElement.MAP_FUNCTIONS);
         break;
+      case BITWISE:
+        validateAllowed(GrammarElement.BITWISE_FUNCTIONS);
+        break;
       case CSV:
         validateAllowed(GrammarElement.CSV_FUNCTIONS);
         break;
       case MISC:
         validateAllowed(GrammarElement.MISC_FUNCTIONS);
         break;
+      case GENERATOR:
+        validateAllowed(GrammarElement.GENERATOR_FUNCTIONS);
+        break;
       case UDF:
         validateAllowed(GrammarElement.UDF);
         break;

diff --git a/async-query-core/src/test/java/org/opensearch/sql/spark/validator/SQLQueryValidatorTest.java b/async-query-core/src/test/java/org/opensearch/sql/spark/validator/SQLQueryValidatorTest.java
@@ -309,7 +309,7 @@ void testDenyAllValidator() {
     v.ok(TestElement.JSON_FUNCTIONS);
     v.ok(TestElement.MATHEMATICAL_FUNCTIONS);
     v.ok(TestElement.STRING_FUNCTIONS);
-    v.ok(TestElement.BITWISE_FUNCTIONS);
+    v.ng(TestElement.BITWISE_FUNCTIONS);
     v.ok(TestElement.CONVERSION_FUNCTIONS);
     v.ok(TestElement.CONDITIONAL_FUNCTIONS);
     v.ok(TestElement.PREDICATE_FUNCTIONS);
@@ -321,7 +321,7 @@ void testDenyAllValidator() {
     v.ok(TestElement.WINDOW_FUNCTIONS);
 
     // Generator Functions
-    v.ok(TestElement.GENERATOR_FUNCTIONS);
+    v.ng(TestElement.GENERATOR_FUNCTIONS);
 
     // UDFs
     v.ng(TestElement.SCALAR_USER_DEFINED_FUNCTIONS);
@@ -426,7 +426,7 @@ void testS3glueQueries() {
     v.ok(TestElement.JSON_FUNCTIONS);
     v.ok(TestElement.MATHEMATICAL_FUNCTIONS);
     v.ok(TestElement.STRING_FUNCTIONS);
-    v.ok(TestElement.BITWISE_FUNCTIONS);
+    v.ng(TestElement.BITWISE_FUNCTIONS);
     v.ok(TestElement.CONVERSION_FUNCTIONS);
     v.ok(TestElement.CONDITIONAL_FUNCTIONS);
     v.ok(TestElement.PREDICATE_FUNCTIONS);