feat: fill param marker order (#15)

Author: jjeffcaii

parser.go

@@ -15277,7 +15277,10 @@ yynewstate:
 		}
 	case 1226:
 		{
-			parser.yyVAL.expr = ast.NewParamMarkerExpr(yyS[yypt].offset)
+			paramExpr := ast.NewParamMarkerExpr(yyS[yypt].offset)
+			paramExpr.SetOrder(parser.paramMarkerCursor)
+			parser.paramMarkerCursor++
+			parser.yyVAL.expr = paramExpr
 		}
 	case 1229:
 		{
@@ -17184,7 +17187,10 @@ yynewstate:
 		}
 	case 1612:
 		{
-			parser.yyVAL.item = ast.NewParamMarkerExpr(yyS[yypt].offset)
+			paramExpr := ast.NewParamMarkerExpr(yyS[yypt].offset)
+			paramExpr.SetOrder(parser.paramMarkerCursor)
+			parser.paramMarkerCursor++
+			parser.yyVAL.item = paramExpr
 		}
 	case 1617:
 		{

parser.y

@@ -6754,7 +6754,10 @@ SimpleExpr:
 |	Literal
 |	paramMarker
 	{
-		$$ = ast.NewParamMarkerExpr(yyS[yypt].offset)
+		paramExpr := ast.NewParamMarkerExpr(yyS[yypt].offset)
+		paramExpr.SetOrder(parser.paramMarkerCursor)
+		parser.paramMarkerCursor++
+		$$ = paramExpr
 	}
 |	Variable
 |	SumExpr
@@ -8941,7 +8944,10 @@ LimitOption:
 	}
 |	paramMarker
 	{
-		$$ = ast.NewParamMarkerExpr(yyS[yypt].offset)
+		paramExpr := ast.NewParamMarkerExpr(yyS[yypt].offset)
+		paramExpr.SetOrder(parser.paramMarkerCursor)
+		parser.paramMarkerCursor++
+		$$ = paramExpr
 	}
 
 RowOrRows:

parser_test.go

@@ -6660,3 +6660,19 @@ SELECT 3;
 	assert.Equal(t, []string{"bar", "qux"}, stmts[1].Hints())
 	assert.Len(t, stmts[2].Hints(), 0)
 }
+
+func TestFillParamMarker(t *testing.T) {
+	p := parser.New()
+	stmt, err := p.ParseOneStmt("select * from t where k = ? limit ?,?", "", "")
+	assert.NoError(t, err)
+	sel := stmt.(*ast.SelectStmt)
+
+	where := sel.Where.(*ast.BinaryOperationExpr).R.(*test_driver.ParamMarkerExpr).Order
+	assert.Equal(t, 0, where)
+
+	offset := sel.Limit.Offset.(*test_driver.ParamMarkerExpr).Order
+	assert.Equal(t, 1, offset)
+
+	limit := sel.Limit.Count.(*test_driver.ParamMarkerExpr).Order
+	assert.Equal(t, 2, limit)
+}

yy_parser.go

@@ -88,6 +88,7 @@ type Parser struct {
 
 	explicitCharset       bool
 	strictDoubleFieldType bool
+	paramMarkerCursor     int
 
 	// the following fields are used by yyParse to reduce allocation.
 	cache  []yySymType
@@ -127,6 +128,10 @@ func New() *Parser {
 	return p
 }
 
+func (parser *Parser) ParamMarkerCursor() int {
+	return parser.paramMarkerCursor
+}
+
 func (parser *Parser) SetStrictDoubleTypeCheck(val bool) {
 	parser.strictDoubleFieldType = val
 }
@@ -397,6 +402,7 @@ var (
 func resetParams(p *Parser) {
 	p.charset = mysql.DefaultCharset
 	p.collation = mysql.DefaultCollationName
+	p.paramMarkerCursor = 0
 }
 
 // ParseParam represents the parameter of parsing.