feat(duckdb): Add support for SET VARIABLE (#5360)

Author: VaggelisD

sqlglot/dialects/duckdb.py

@@ -484,6 +484,11 @@ class Parser(parser.Parser):
             TokenType.SHOW: lambda self: self._parse_show(),
         }
 
+        SET_PARSERS = {
+            **parser.Parser.SET_PARSERS,
+            "VARIABLE": lambda self: self._parse_set_item_assignment("VARIABLE"),
+        }
+
         def _parse_lambda(self, alias: bool = False) -> t.Optional[exp.Expression]:
             index = self._index
             if not self._match_text_seq("LAMBDA"):

tests/dialects/test_duckdb.py

@@ -1686,3 +1686,12 @@ def test_extract_date_parts(self):
             with self.subTest(f"Testing DuckDB EXTRACT({part} FROM foo)"):
                 # All of these should remain as is, they don't have synonyms
                 self.validate_identity(f"EXTRACT({part} FROM foo)")
+
+    def test_set_item(self):
+        self.validate_identity("SET memory_limit = '10GB'")
+        self.validate_identity("SET SESSION default_collation = 'nocase'")
+        self.validate_identity("SET GLOBAL sort_order = 'desc'")
+        self.validate_identity("SET VARIABLE my_var = 30")
+        self.validate_identity("SET VARIABLE location_map = (SELECT foo FROM bar)")
+
+        self.validate_identity("SET VARIABLE my_var TO 30", "SET VARIABLE my_var = 30")