parser: add LOW_PRIORITY for LOAD DATA (#51023)

ref #49602
pingcap · Feb 8, 2024 · 2437633 · 2437633
1 parent c5eced1
commit 2437633
Show file tree

Hide file tree

Showing 4 changed files with 5,313 additions and 5,279 deletions.
diff --git a/pkg/parser/ast/dml.go b/pkg/parser/ast/dml.go
@@ -1865,6 +1865,7 @@ const (
 type LoadDataStmt struct {
 	dmlNode
 
+	LowPriority       bool
 	FileLocRef        FileLocRefTp
 	Path              string
 	Format            *string
@@ -1884,6 +1885,9 @@ type LoadDataStmt struct {
 // Restore implements Node interface.
 func (n *LoadDataStmt) Restore(ctx *format.RestoreCtx) error {
 	ctx.WriteKeyWord("LOAD DATA ")
+	if n.LowPriority {
+		ctx.WriteKeyWord("LOW_PRIORITY ")
+	}
 	switch n.FileLocRef {
 	case FileLocServerOrRemote:
 	case FileLocClient:

diff --git a/pkg/parser/ast/dml_test.go b/pkg/parser/ast/dml_test.go
@@ -414,6 +414,10 @@ func TestWindowSpecRestore(t *testing.T) {
 
 func TestLoadDataRestore(t *testing.T) {
 	testCases := []NodeRestoreTestCase{
+		{
+			sourceSQL: "load data low_priority infile '/a.csv' into table `t`",
+			expectSQL: "LOAD DATA LOW_PRIORITY INFILE '/a.csv' INTO TABLE `t`",
+		},
 		{
 			sourceSQL: "load data infile '/a.csv' format 'sql file' into table `t`",
 			expectSQL: "LOAD DATA INFILE '/a.csv' FORMAT 'sql file' INTO TABLE `t`",