Add limit clause to pinot queries (#5337)

* add limit clause to queries

* cleanup: run make copyright things

common/persistence/Pinot/pinotVisibilityStore.go

@@ -932,6 +932,16 @@ func getListWorkflowExecutionsByTypeQuery(tableName string, request *p.InternalL
 
 	//query.addPinotSorter(CloseTime, DescendingOrder)
 	//query.addPinotSorter(RunID, DescendingOrder)
+
+	token, err := pnt.GetNextPageToken(request.NextPageToken)
+	if err != nil {
+		panic(fmt.Sprintf("deserialize next page token error: %s", err))
+	}
+
+	from := token.From
+	pageSize := request.PageSize
+	query.addOffsetAndLimits(from, pageSize)
+
 	return query.String()
 }
 
@@ -958,6 +968,16 @@ func getListWorkflowExecutionsByWorkflowIDQuery(tableName string, request *p.Int
 
 	query.addPinotSorter(CloseTime, DescendingOrder)
 	//query.addPinotSorter(RunID, DescendingOrder)
+
+	token, err := pnt.GetNextPageToken(request.NextPageToken)
+	if err != nil {
+		panic(fmt.Sprintf("deserialize next page token error: %s", err))
+	}
+
+	from := token.From
+	pageSize := request.PageSize
+	query.addOffsetAndLimits(from, pageSize)
+
 	return query.String()
 }
 
@@ -991,6 +1011,16 @@ func getListWorkflowExecutionsByStatusQuery(tableName string, request *p.Interna
 
 	query.addPinotSorter(CloseTime, DescendingOrder)
 	//query.addPinotSorter(RunID, DescendingOrder)
+
+	token, err := pnt.GetNextPageToken(request.NextPageToken)
+	if err != nil {
+		panic(fmt.Sprintf("deserialize next page token error: %s", err))
+	}
+
+	from := token.From
+	pageSize := request.PageSize
+	query.addOffsetAndLimits(from, pageSize)
+
 	return query.String()
 }
 

common/persistence/Pinot/pinotVisibilityStore_test.go

@@ -386,6 +386,7 @@ AND WorkflowType = 'test-wf-type'
 AND CloseTime BETWEEN 1547596871371 AND 2547596873371
 AND CloseStatus >= 0
 Order BY CloseTime DESC
+LIMIT 0, 10
 `, testTableName)
 	expectOpenResult := fmt.Sprintf(`SELECT *
 FROM %s
@@ -395,6 +396,7 @@ AND StartTime BETWEEN 1547596871371 AND 2547596873371
 AND CloseStatus < 0
 AND CloseTime = -1
 Order BY RunID DESC
+LIMIT 0, 10
 `, testTableName)
 	expectNilResult := ""
 
@@ -426,6 +428,7 @@ AND WorkflowID = 'test-wid'
 AND CloseTime BETWEEN 1547596871371 AND 2547596873371
 AND CloseStatus >= 0
 Order BY CloseTime DESC
+LIMIT 0, 10
 `, testTableName)
 	expectOpenResult := fmt.Sprintf(`SELECT *
 FROM %s
@@ -435,6 +438,7 @@ AND StartTime BETWEEN 1547596871371 AND 2547596873371
 AND CloseStatus < 0
 AND CloseTime = -1
 Order BY CloseTime DESC
+LIMIT 0, 10
 `, testTableName)
 	expectNilResult := ""
 
@@ -464,6 +468,7 @@ WHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'
 AND CloseStatus = '0'
 AND CloseTime BETWEEN 1547596872371 AND 2547596872371
 Order BY CloseTime DESC
+LIMIT 0, 10
 `, testTableName)
 	expectNilResult := ""