Parsed ALTER DATABASE query (#45)

* Parsed ALTER DATABASE query

Author: AdalbertMemSQL

sqlglot/dialects/singlestore.py

@@ -444,6 +444,13 @@ class Parser(parser.Parser):
             TokenType.DPIPE: exp.Or,
         }
 
+        ALTERABLES = parser.Parser.ALTERABLES | {TokenType.DATABASE}
+
+        ALTER_PARSERS = {
+            **parser.Parser.ALTER_PARSERS,
+            "WITH": lambda self: self._parse_alter_table_set(),
+        }
+
         def _parse_match_against(self) -> exp.MatchAgainst:
             if self._match_text_seq("TABLE"):
                 expressions = []
@@ -461,6 +468,17 @@ def _parse_match_against(self) -> exp.MatchAgainst:
                 exp.MatchAgainst, this=this, expressions=expressions
             )
 
+        def _parse_alter_table_set(self) -> exp.AlterSet:
+            alter_set = self.expression(exp.AlterSet)
+
+            if self._match_text_seq("SYNC", "REPLICATION"):
+                alter_set.set("expressions", [exp.Var(this=f"SYNC REPLICATION")])
+            elif self._match_text_seq("ASYNC", "REPLICATION"):
+                    alter_set.set("expressions", [exp.Var(
+                        this=f"ASYNC REPLICATION")])
+
+            return alter_set
+
     class Generator(generator.Generator):
         LOCKING_READS_SUPPORTED = True
         EXCEPT_INTERSECT_SUPPORT_ALL_CLAUSE = False
@@ -2959,10 +2977,6 @@ def indexconstraintoption_sql(self,
             self.unsupported("Unsupported index constraint option.")
             return ""
 
-        def alterset_sql(self, expression: exp.AlterSet) -> str:
-            self.unsupported("ALTER SET query is not supported in SingleStore")
-            return super().alterset_sql(expression)
-
         def periodforsystemtimeconstraint_sql(self,
             expression: exp.PeriodForSystemTimeConstraint) -> str:
             self.unsupported(

tests/dialects/test_singlestore.py

@@ -1957,12 +1957,10 @@ def test_expressions_generation(self):
             sql="CREATE TABLE ColumnConstraint (id INT AUTO_INCREMENT PRIMARY KEY)",
             exp_type=exp.ColumnConstraint)
         self.validate_generation(
-            sql="ALTER TABLE orders SET DATA_RETENTION_TIME_IN_DAYS = 1",
-            expected_sql="ALTER TABLE orders SET",
-            error_message="ALTER SET query is not supported in SingleStore",
+            sql="ALTER DATABASE db SET SYNC REPLICATION",
+            expected_sql="ALTER DATABASE db SET SYNC REPLICATION",
             exp_type=exp.AlterSet,
-            from_dialect="snowflake",
-            run=False
+            from_dialect="singlestore",
         )
         self.validate_generation(
             sql="CREATE TABLE ConstraintTable (a INT, CONSTRAINT id PRIMARY KEY (a))",
@@ -6282,3 +6280,53 @@ def test_operators_parsing(self):
             ),
             run=False
         )
+
+    def test_alter_database_parsing(self):
+        self.validate_parsing(
+            "ALTER DATABASE db WITH SYNC REPLICATION",
+            exp.Alter(
+                this=exp.Table(this=exp.Identifier(this="db", kind="SCHEMA")),
+                kind="DATABASE",
+                actions=[
+                    exp.AlterSet(
+                        expressions=[exp.Var(this="SYNC REPLICATION")]
+                    )
+                ]
+            )
+        )
+        self.validate_parsing(
+            "ALTER DATABASE db WITH ASYNC REPLICATION",
+            exp.Alter(
+                this=exp.Table(this=exp.Identifier(this="db", kind="SCHEMA")),
+                kind="DATABASE",
+                actions=[
+                    exp.AlterSet(
+                        expressions=[exp.Var(this="ASYNC REPLICATION")]
+                    )
+                ]
+            )
+        )
+        self.validate_parsing(
+            "ALTER DATABASE db SET SYNC REPLICATION",
+            exp.Alter(
+                this=exp.Table(this=exp.Identifier(this="db", kind="SCHEMA")),
+                kind="DATABASE",
+                actions=[
+                    exp.AlterSet(
+                        expressions=[exp.Var(this="SYNC REPLICATION")]
+                    )
+                ]
+            )
+        )
+        self.validate_parsing(
+            "ALTER DATABASE db SET ASYNC REPLICATION",
+            exp.Alter(
+                this=exp.Table(this=exp.Identifier(this="db", kind="SCHEMA")),
+                kind="DATABASE",
+                actions=[
+                    exp.AlterSet(
+                        expressions=[exp.Var(this="ASYNC REPLICATION")]
+                    )
+                ]
+            )
+        )