feat(tsql): clustered index closes #3659

tobymao · tobymao · commit fb4d9080a042 · 2024-06-14T19:00:46.000-07:00
diff --git a/sqlglot/dialects/tsql.py b/sqlglot/dialects/tsql.py
@@ -450,13 +450,15 @@ class Tokenizer(tokens.Tokenizer):
 
         KEYWORDS = {
             **tokens.Tokenizer.KEYWORDS,
+            "CLUSTERED INDEX": TokenType.INDEX,
             "DATETIME2": TokenType.DATETIME,
             "DATETIMEOFFSET": TokenType.TIMESTAMPTZ,
             "DECLARE": TokenType.DECLARE,
             "EXEC": TokenType.COMMAND,
             "FOR SYSTEM_TIME": TokenType.TIMESTAMP_SNAPSHOT,
             "IMAGE": TokenType.IMAGE,
             "MONEY": TokenType.MONEY,
+            "NONCLUSTERED INDEX": TokenType.INDEX,
             "NTEXT": TokenType.TEXT,
             "OPTION": TokenType.OPTION,
             "OUTPUT": TokenType.RETURNING,
diff --git a/tests/dialects/test_tsql.py b/tests/dialects/test_tsql.py
@@ -8,6 +8,9 @@ class TestTSQL(Validator):
     dialect = "tsql"
 
     def test_tsql(self):
+        self.validate_identity(
+            "CREATE CLUSTERED INDEX [IX_OfficeTagDetail_TagDetailID] ON [dbo].[OfficeTagDetail]([TagDetailID] ASC)"
+        )
         self.validate_identity(
             "CREATE INDEX [x] ON [y]([z] ASC) WITH (allow_page_locks=on) ON X([y])"
         )

Original file line number	Diff line number	Diff line change
`@@ -8,6 +8,9 @@ class TestTSQL(Validator):`
`8`	`8`	`dialect = "tsql"`
`9`	`9`
`10`	`10`	`def test_tsql(self):`
	`11`	`+ self.validate_identity(`
	`12`	`+ "CREATE CLUSTERED INDEX [IX_OfficeTagDetail_TagDetailID] ON [dbo].[OfficeTagDetail]([TagDetailID] ASC)"`
	`13`	`+ )`
`11`	`14`	`self.validate_identity(`
`12`	`15`	`"CREATE INDEX [x] ON [y]([z] ASC) WITH (allow_page_locks=on) ON X([y])"`
`13`	`16`	`)`