parser: reset lexer.identifierDot correctly when reusing parsers (#…

…45903) (#45910) close #45898
pingcap · Aug 8, 2023 · c620def · c620def
1 parent e21efe5
commit c620def
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/parser/lexer.go b/parser/lexer.go
@@ -101,6 +101,7 @@ func (s *Scanner) reset(sql string) {
 	s.stmtStartPos = 0
 	s.inBangComment = false
 	s.lastKeyword = 0
+	s.identifierDot = false
 }
 
 func (s *Scanner) stmtText() string {

diff --git a/parser/parser_test.go b/parser/parser_test.go
@@ -7269,6 +7269,18 @@ func TestTTLTableOption(t *testing.T) {
 	RunTest(t, table, false)
 }
 
+func TestIssue45898(t *testing.T) {
+	p := parser.New()
+	p.ParseSQL("a.")
+	stmts, _, err := p.ParseSQL("select count(1) from t")
+	require.NoError(t, err)
+	var sb strings.Builder
+	restoreCtx := NewRestoreCtx(DefaultRestoreFlags, &sb)
+	sb.Reset()
+	stmts[0].Restore(restoreCtx)
+	require.Equal(t, "SELECT COUNT(1) FROM `t`", sb.String())
+}
+
 func TestMultiStmt(t *testing.T) {
 	p := parser.New()
 	stmts, _, err := p.Parse("SELECT 'foo'; SELECT 'foo;bar','baz'; select 'foo' , 'bar' , 'baz' ;select 1", "", "")