[parser] parser: fix select field text (#30)

pingcap · Oct 9, 2021 · 5fe32fe · 5fe32fe
1 parent bd8cc65
commit 5fe32fe
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/parser/parser.go b/parser/parser.go
diff --git a/parser/parser.y b/parser/parser.y
@@ -4378,7 +4378,7 @@ SelectStmtFromTable:
 		}
 		lastField := st.Fields.Fields[len(st.Fields.Fields)-1]
 		if lastField.Expr != nil && lastField.AsName.O == "" {
-			lastEnd := parser.endOffset(&yyS[yypt-4])
+			lastEnd := parser.endOffset(&yyS[yypt-5])
 			lastField.SetText(parser.src[lastField.Offset:lastEnd])
 		}
 		if $4 != nil {

diff --git a/parser/parser_test.go b/parser/parser_test.go
@@ -2523,3 +2523,11 @@ func (s *testParserSuite) TestWindowFunctions(c *C) {
 	s.enableWindowFunc = true
 	s.RunTest(c, table)
 }
+
+func (s *testParserSuite) TestFieldText(c *C) {
+	parser := New()
+	stmts, err := parser.Parse("select a from t", "", "")
+	c.Assert(err, IsNil)
+	tmp := stmts[0].(*ast.SelectStmt)
+	c.Assert(tmp.Fields.Fields[0].Text(), Equals, "a")
+}