Procházet zdrojové kódy

set max results param for athena pagination

Sean Holcomb před 2 roky
rodič
revize
c1f1ba7541
1 změnil soubory, kde provedl 2 přidání a 0 odebrání
  1. 2 0
      pkg/cloud/aws/athenaquerier.go

+ 2 - 0
pkg/cloud/aws/athenaquerier.go

@@ -126,8 +126,10 @@ func (aq *AthenaQuerier) queryAthenaPaginated(ctx context.Context, query string,
 	if err != nil {
 		return fmt.Errorf("QueryAthenaPaginated: query execution error: %s", err.Error())
 	}
+	var maxResults int32 = 10000
 	queryResultsInput := &athena.GetQueryResultsInput{
 		QueryExecutionId: startQueryExecutionOutput.QueryExecutionId,
+		MaxResults:       &maxResults,
 	}
 	getQueryResultsPaginator := athena.NewGetQueryResultsPaginator(cli, queryResultsInput)
 	for getQueryResultsPaginator.HasMorePages() {