fix(redshift): support OBJECT_TRANSFORM function (#7601)

* added parser support for OBJECT_TRANSFORM

* add OBJECT_TRANSFORM tests

* return formatting tweak

Author: fivetran-kwoodbeck

sqlglot-integration-tests

@@ -148,6 +148,10 @@ class Nullif(Expression, Func):
     arg_types = {"this": True, "expression": True}
 
 
+class ObjectTransform(Expression, Func):
+    arg_types = {"this": True, "keep": False, "set_": False}
+
+
 class Nvl2(Expression, Func):
     arg_types = {"this": True, "true": True, "false": False}
 

sqlglot/expressions/functions.py

@@ -270,6 +270,14 @@ class RedshiftGenerator(PostgresGenerator):
         "without",
     }
 
+    def objecttransform_sql(self, expression: exp.ObjectTransform) -> str:
+        this = self.sql(expression, "this")
+        keep = self.expressions(expression, key="keep", flat=True)
+        set_ = self.expressions(expression, key="set_", flat=True)
+        keep_sql = " KEEP " + keep if keep else ""
+        set_sql = " SET " + set_ if set_ else ""
+        return f"OBJECT_TRANSFORM({this}{keep_sql}{set_sql})"
+
     def approxquantile_sql(self, expression: exp.ApproxQuantile) -> str:
         return "APPROXIMATE " + self.sql(
             exp.WithinGroup(

sqlglot/generators/redshift.py

@@ -66,6 +66,11 @@ class RedshiftParser(PostgresParser):
         "SYSDATE": lambda self: self.expression(exp.CurrentTimestamp(sysdate=True)),
     }
 
+    FUNCTION_PARSERS = {
+        **PostgresParser.FUNCTION_PARSERS,
+        "OBJECT_TRANSFORM": lambda self: self._parse_object_transform(),
+    }
+
     SUPPORTS_IMPLICIT_UNNEST = True
 
     def _parse_table(
@@ -96,6 +101,16 @@ def _parse_convert(self, strict: bool, safe: bool | None = None) -> exp.Expr | N
         this = self._parse_bitwise()
         return self.expression(exp.TryCast(this=this, to=to, safe=safe))
 
+    def _parse_object_transform(self) -> exp.ObjectTransform:
+        this = self._parse_column()
+        keep: list[exp.Expr] = []
+        set_: list[exp.Expr] = []
+        if self._match(TokenType.KEEP):
+            keep = self._parse_csv(self._parse_primary)
+        if self._match(TokenType.SET):
+            set_ = self._parse_csv(self._parse_expression)
+        return self.expression(exp.ObjectTransform(this=this, keep=keep, set_=set_))
+
     def _parse_approximate_count(self) -> exp.ApproxDistinct | exp.ApproxQuantile | None:
         index = self._index - 1
         func = self._parse_function()