fix: Add MAX_BY & MIN_BY to FUNCTION_PARSER (#5021)

* fix: Add MAX_BY & MIN_BY to FUNCTION_PARSER

* PR Feedback 1

Author: VaggelisD

sqlglot/parser.py

@@ -1184,6 +1184,12 @@ def _parse_partitioned_by_bucket_or_truncate(self) -> exp.Expression:
     KEY_VALUE_DEFINITIONS = (exp.Alias, exp.EQ, exp.PropertyEQ, exp.Slice)
 
     FUNCTION_PARSERS = {
+        **{
+            name: lambda self: self._parse_max_min_by(exp.ArgMax) for name in exp.ArgMax.sql_names()
+        },
+        **{
+            name: lambda self: self._parse_max_min_by(exp.ArgMin) for name in exp.ArgMin.sql_names()
+        },
         "CAST": lambda self: self._parse_cast(self.STRICT_CAST),
         "CEIL": lambda self: self._parse_ceil_floor(exp.Ceil),
         "CONVERT": lambda self: self._parse_convert(self.STRICT_CAST),
@@ -8224,3 +8230,16 @@ def _parse_format_name(self) -> exp.Property:
             this=exp.var("FORMAT_NAME"),
             value=self._parse_string() or self._parse_table_parts(),
         )
+
+    def _parse_max_min_by(self, expr_type: t.Type[exp.AggFunc]) -> exp.AggFunc:
+        args: t.List[exp.Expression] = []
+
+        if self._match(TokenType.DISTINCT):
+            args.append(self.expression(exp.Distinct, expressions=[self._parse_assignment()]))
+            self._match(TokenType.COMMA)
+
+        args.extend(self._parse_csv(self._parse_assignment))
+
+        return self.expression(
+            expr_type, this=seq_get(args, 0), expression=seq_get(args, 1), count=seq_get(args, 2)
+        )

tests/dialects/test_snowflake.py

@@ -2578,3 +2578,11 @@ def test_parameter(self):
         self.assertEqual(expr.find(exp.Placeholder), exp.Placeholder(this="1"))
         self.validate_identity("SELECT :1, :2")
         self.validate_identity("SELECT :1 + :2")
+
+    def test_max_by_min_by(self):
+        max_by = self.validate_identity("MAX_BY(DISTINCT selected_col, filtered_col)")
+        min_by = self.validate_identity("MIN_BY(DISTINCT selected_col, filtered_col)")
+
+        for node in (max_by, min_by):
+            self.assertEqual(len(node.this.expressions), 1)
+            self.assertIsInstance(node.expression, exp.Column)

tests/fixtures/identity.sql

@@ -889,3 +889,5 @@ CAST(x AS INT128)
 CAST(x AS UINT128)
 CAST(x AS UINT256)
 SELECT export
+SELECT ARG_MAX(DISTINCT selected_col, filtered_col) FROM table
+SELECT ARG_MIN(DISTINCT selected_col, filtered_col) FROM table