TSQL: allow leading GO (#2997)

Co-authored-by: Paul Guyot <pguyot@upclear.com> Co-authored-by: Barry Pollard <barry_pollard@hotmail.com>
sqlfluff · Apr 4, 2022 · ae7b629 · ae7b629
1 parent 939113e
commit ae7b629
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 9 deletions.
diff --git a/src/sqlfluff/dialects/dialect_tsql.py b/src/sqlfluff/dialects/dialect_tsql.py
@@ -2572,16 +2572,19 @@ class FileSegment(BaseFileSegment):
 
     # NB: We don't need a match_grammar here because we're
     # going straight into instantiating it directly usually.
-    parse_grammar = Delimited(
-        Ref("BatchSegment"),
-        delimiter=AnyNumberOf(
-            Sequence(
-                Ref("DelimiterGrammar", optional=True), Ref("BatchDelimiterGrammar")
+    parse_grammar = Sequence(
+        AnyNumberOf(Ref("BatchDelimiterGrammar")),
+        Delimited(
+            Ref("BatchSegment"),
+            delimiter=AnyNumberOf(
+                Sequence(
+                    Ref("DelimiterGrammar", optional=True), Ref("BatchDelimiterGrammar")
+                ),
+                min_times=1,
             ),
-            min_times=1,
+            allow_gaps=True,
+            allow_trailing=True,
         ),
-        allow_gaps=True,
-        allow_trailing=True,
     )
 
 

diff --git a/test/fixtures/dialects/tsql/go_delimiters.sql b/test/fixtures/dialects/tsql/go_delimiters.sql
@@ -1,3 +1,5 @@
+-- It's possible to have a file starting with GO
+GO
 -- It's possible to have multiple GO between batches.
 SELECT foo FROM bar GO GO
 SELECT foo FROM bar GO

diff --git a/test/fixtures/dialects/tsql/go_delimiters.yml b/test/fixtures/dialects/tsql/go_delimiters.yml
@@ -3,8 +3,10 @@
 # computed by SQLFluff when running the tests. Please run
 # `python test/generate_parse_fixture_yml.py`  to generate them after adding or
 # altering SQL files.
-_hash: 48ee8f3f556e82816f06c2f0155e2eb39630f07d23a9a0a7cf961d58d58df5eb
+_hash: bd36bce2e40fe28989cdc56078defe2ddae2d528b37e5b717984c9a6e64041af
 file:
+- go_statement:
+    keyword: GO
 - batch:
     statement:
       select_statement: