Merge pull request pingcap#19 from rebelice/dev_limit

support limit
crazycs520 · Jan 7, 2022 · 7db01e9 · 7db01e9
2 parents 81e897c + 7dcfa1d
commit 7db01e9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/planner/core/common_plans.go b/planner/core/common_plans.go
@@ -1619,6 +1619,7 @@ type RestoreData struct {
 	Agg     string
 	GroupBy string
 	Where   []string
+	Limit   string
 }
 
 func (d *RestoreData) String() string {
@@ -1650,6 +1651,10 @@ func (d *RestoreData) String() string {
 		fmt.Fprint(&buffer, " group by ")
 		fmt.Fprint(&buffer, d.GroupBy)
 	}
+	if len(d.Limit) != 0 {
+		fmt.Fprint(&buffer, " limit ")
+		fmt.Fprint(&buffer, d.Limit)
+	}
 	return buffer.String()
 }
 
@@ -1748,6 +1753,11 @@ func BuildAWSQueryInfo(v *PhysicalTableReader, id int64) *RestoreData {
 				}
 				info.GroupBy = buffer.String()
 			}
+		case *PhysicalLimit:
+			if x.Offset != 0 {
+				logutil.BgLogger().Warn("S3 don't support OFFSET for LIMIT")
+			}
+			info.Limit = fmt.Sprintf("%v", x.Count)
 		}
 	}
 	return &info