SNOW-2268215: support group by all (#3889)

Author: sfc-gh-mayliu

CHANGELOG.md

@@ -7,6 +7,7 @@
 #### New Features
 
 - Added a new function `service` in `snowflake.snowpark.functions` that allows users to create a callable representing a Snowpark Container Services (SPCS) service.
+- Added a new function `group_by_all()` to the `DataFrame` class.
 - Added `connection_parameters` parameter to `DataFrameReader.dbapi()` (PuPr) method to allow passing keyword arguments to the `create_connection` callable.
 - Added support for `Session.begin_transaction`, `Session.commit` and `Session.rollback`.
 - Added support for the following functions in `functions.py`:

docs/source/snowpark/dataframe.rst

@@ -57,6 +57,7 @@ DataFrame
     DataFrame.flatten
     DataFrame.groupBy
     DataFrame.group_by
+    DataFrame.group_by_all
     DataFrame.group_by_grouping_sets
     DataFrame.intersect
     DataFrame.join

src/snowflake/snowpark/_internal/analyzer/analyzer.py

@@ -91,6 +91,7 @@
 from snowflake.snowpark._internal.analyzer.grouping_set import (
     GroupingSet,
     GroupingSetsExpression,
+    GroupByAll,
 )
 from snowflake.snowpark._internal.analyzer.select_statement import (
     Selectable,
@@ -346,6 +347,8 @@ def analyze(
             )
 
         if isinstance(expr, GroupingSet):
+            if isinstance(expr, GroupByAll):
+                return "ALL"
             return self.grouping_extractor(expr, df_aliased_col_name_to_real_col_name)
 
         if isinstance(expr, WindowExpression):

src/snowflake/snowpark/_internal/analyzer/grouping_set.py

@@ -40,6 +40,10 @@ class Rollup(GroupingSet):
     pass
 
 
+class GroupByAll(GroupingSet):
+    pass
+
+
 class GroupingSetsExpression(Expression):
     def __init__(self, args: List[List[Expression]]) -> None:
         super().__init__()