Merge pull request #50 from adangel/soql-grouping

Support GROUPING function in SOQL queries
apex-dev-tools · Sep 2, 2024 · 8522020 · 8522020
2 parents fee3a16 + 3ae515f
commit 8522020
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 0 deletions.
diff --git a/antlr/ApexLexer.g4 b/antlr/ApexLexer.g4
@@ -170,6 +170,7 @@ CUSTOM          : 'custom';
 STANDARD        : 'standard';
 DISTANCE        : 'distance';
 GEOLOCATION     : 'geolocation';
+GROUPING        : 'grouping';
 
 // SOQL Date functions
 CALENDAR_MONTH      : 'calendar_month';

diff --git a/antlr/ApexParser.g4 b/antlr/ApexParser.g4
@@ -636,6 +636,7 @@ soqlFunction
     | WEEK_IN_YEAR LPAREN dateFieldName RPAREN
     | FIELDS LPAREN soqlFieldsParameter RPAREN
     | DISTANCE LPAREN locationValue COMMA locationValue COMMA StringLiteral RPAREN
+    | GROUPING LPAREN fieldName RPAREN
     ;
 
  dateFieldName
@@ -962,6 +963,7 @@ id
     | CUSTOM
     | DISTANCE
     | GEOLOCATION
+    | GROUPING
     // SOQL date functions
     | CALENDAR_MONTH
     | CALENDAR_QUARTER
@@ -1160,6 +1162,7 @@ anyId
     | CUSTOM
     | DISTANCE
     | GEOLOCATION
+    | GROUPING
     // SOQL date functions
     | CALENDAR_MONTH
     | CALENDAR_QUARTER

diff --git a/jvm/src/test/java/io/github/apexdevtools/apexparser/SOQLParserTest.java b/jvm/src/test/java/io/github/apexdevtools/apexparser/SOQLParserTest.java
@@ -111,4 +111,23 @@ void testSubQuery() {
         assertNotNull(context);
         assertEquals(0, parserAndCounter.getValue().getNumErrors());
     }
+
+    @Test
+    void testGroupingFunction() {
+        Map.Entry<ApexParser, SyntaxErrorCounter> parserAndCounter = createParser(
+                        "SELECT\n" +
+                        "  OBJ1__c O1,\n" +
+                        "  OBJ2__c O2,\n" +
+                        "  OBJ3__c O3,\n" +
+                        "  SUM(OBJ4__c) O4,\n" +
+                        "  GROUPING(OBJ1__c) O1Group,\n" +
+                        "  GROUPING(OBJ2__c) O2Group,\n" +
+                        "  GROUPING(OBJ3__c) O3Group\n" +
+                        "FROM OBJ4__c\n" +
+                        "GROUP BY ROLLUP(OBJ1__c, OBJ2__c, OBJ3__c)"
+        );
+        ApexParser.QueryContext context = parserAndCounter.getKey().query();
+        assertNotNull(context);
+        assertEquals(0, parserAndCounter.getValue().getNumErrors());
+    }
 }
diff --git a/npm/src/__tests__/SOQLParserTest.ts b/npm/src/__tests__/SOQLParserTest.ts
@@ -118,3 +118,23 @@ test("SubQuery", () => {
   expect(context).toBeInstanceOf(QueryContext);
   expect(errorCounter.getNumErrors()).toEqual(0);
 });
+
+test("Grouping function", () => {
+  const [parser, errorCounter] = createParser(
+   `SELECT
+     OBJ1__c O1,
+     OBJ2__c O2,
+     OBJ3__c O3,
+     SUM(OBJ4__c) O4,
+     GROUPING(OBJ1__c) O1Group,
+     GROUPING(OBJ2__c) O2Group,
+     GROUPING(OBJ3__c) O3Group
+   FROM OBJ4__c
+   GROUP BY ROLLUP(OBJ1__c, OBJ2__c, OBJ3__c)`
+  );
+
+  const context = parser.query();
+
+  expect(context).toBeInstanceOf(QueryContext);
+  expect(errorCounter.getNumErrors()).toEqual(0);
+});